Update search vectors script: output total updated

1 files changed, 4 insertions, 0 deletions

diff --git a/ishtar_common/management/commands/update_search_vectors.py b/ishtar_common/management/commands/update_search_vectors.py
index f30f40f4e..59166fba7 100644
--- a/ishtar_common/management/commands/update_search_vectors.py
+++ b/ishtar_common/management/commands/update_search_vectors.py
@@ -21,6 +21,7 @@ class Command(BaseCommand):
 
     def handle(self, *args, **options):
         quiet = options['quiet']
+        total = 0
         for model in django.apps.apps.get_models():
             if options['model'] and model.__name__ != options['model']:
                 continue
@@ -32,6 +33,7 @@ class Command(BaseCommand):
                 if not quiet:
                     self.stdout.write("\n* update {}".format(model))
                 total = model.objects.count()
+                idx = 0
                 for idx, item in enumerate(model.objects.all()):
                     if not quiet:
                         sys.stdout.write("\r{}/{} ".format(idx + 1, total))
@@ -40,5 +42,7 @@ class Command(BaseCommand):
                         item.update_search_vector()
                     except:
                         pass
+                total += idx
         if not quiet:
             self.stdout.write("\n")
+        self.stdout.write("{} items updated\n".format(total))