entrouvert
/
debian-django-watson
17
0
Fork 0
Code Issues Pull Requests Releases Activity

Adds batch_size functionality to buildwatson

This commit is contained in:
Dustin Broderick 2017-11-28 14:22:55 -07:00
parent ab65d162ee
commit 8fd2573747
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
watson/management/commands/buildwatson.py
View File

@ -58,7 +58,7 @@ def rebuild_index_for_model(model_, engine_slug_, verbosity_, batch_size_=100):
engine_slug=force_text(engine_slug_),
)
)
_bulk_save_search_entries(iter_search_entries())
_bulk_save_search_entries(iter_search_entries(), batch_size=batch_size_)
return local_refreshed_model_count[0]
@ -131,7 +131,7 @@ class Command(BaseCommand):
if verbosity >= 3:
print("Using search engine \"%s\"" % engine_slug)
for model in models:
refreshed_model_count += rebuild_index_for_model(model, engine_slug, verbosity)
refreshed_model_count += rebuild_index_for_model(model, engine_slug, verbosity, batch_size_=batch_size)
else: # full rebuild (for one or all search engines)
if engine_selected:
@ -147,7 +147,7 @@ class Command(BaseCommand):
registered_models = search_engine.get_registered_models()
# Rebuild the index for all registered models.
for model in registered_models:
refreshed_model_count += rebuild_index_for_model(model, engine_slug, verbosity)
refreshed_model_count += rebuild_index_for_model(model, engine_slug, verbosity, batch_size_=batch_size)
# Clean out any search entries that exist for stale content types.
# Only do it during full rebuild