diff options
Diffstat (limited to 'ishtar_common')
-rw-r--r-- | ishtar_common/management/commands/update_search_vectors.py | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/ishtar_common/management/commands/update_search_vectors.py b/ishtar_common/management/commands/update_search_vectors.py index 9fa93277c..f30f40f4e 100644 --- a/ishtar_common/management/commands/update_search_vectors.py +++ b/ishtar_common/management/commands/update_search_vectors.py @@ -15,8 +15,12 @@ class Command(BaseCommand): parser.add_argument( '--model', type=str, default='', dest='model', help='Specific model to update') + parser.add_argument( + '--quiet', dest='quiet', action='store_true', + help='Quiet output') def handle(self, *args, **options): + quiet = options['quiet'] for model in django.apps.apps.get_models(): if options['model'] and model.__name__ != options['model']: continue @@ -25,13 +29,16 @@ class Command(BaseCommand): getattr(model, "INT_SEARCH_VECTORS", None) or getattr(model, "M2M_SEARCH_VECTORS", None) or getattr(model, "PARENT_SEARCH_VECTORS", None)): - self.stdout.write("\n* update {}".format(model)) + if not quiet: + self.stdout.write("\n* update {}".format(model)) total = model.objects.count() for idx, item in enumerate(model.objects.all()): - sys.stdout.write("\r{}/{} ".format(idx + 1, total)) - sys.stdout.flush() + if not quiet: + sys.stdout.write("\r{}/{} ".format(idx + 1, total)) + sys.stdout.flush() try: item.update_search_vector() except: pass - self.stdout.write("\n") + if not quiet: + self.stdout.write("\n") |