Command regenerate: add quiet option

1 files changed, 11 insertions, 4 deletions

diff --git a/ishtar_common/management/commands/regenerate_search_vector_cached_label.py b/ishtar_common/management/commands/regenerate_search_vector_cached_label.py
index 2b3d0454e..542e5f915 100644
--- a/ishtar_common/management/commands/regenerate_search_vector_cached_label.py
+++ b/ishtar_common/management/commands/regenerate_search_vector_cached_label.py
@@ -37,8 +37,12 @@ class Command(BaseCommand):
         parser.add_argument('app_name', nargs='?', default=None,
                             choices=APPS)
         parser.add_argument('model_name', nargs='?', default=None)
+        parser.add_argument(
+            '--quiet', dest='quiet', action='store_true',
+            help='Quiet output')
 
     def handle(self, *args, **options):
+        quiet = options['quiet']
         limit = options['app_name']
         model_name = options['model_name']
         if model_name:
@@ -46,7 +50,8 @@ class Command(BaseCommand):
         for app in APPS:
             if limit and app != limit:
                 continue
-            print(u"* app: {}".format(app))
+            if not quiet:
+                print(u"* app: {}".format(app))
             for model in apps.get_app_config(app).get_models():
                 if model_name and model.__name__.lower() != model_name:
                     continue
@@ -63,7 +68,9 @@ class Command(BaseCommand):
                     object.skip_history_when_saving = True
                     object._no_move = True
                     cmsg = u"\r{} {}/{}".format(msg, idx + 1, ln)
-                    sys.stdout.write(cmsg)
-                    sys.stdout.flush()
+                    if not quiet:
+                        sys.stdout.write(cmsg)
+                        sys.stdout.flush()
                     object.save()
-                sys.stdout.write("\n")
+                if not quiet:
+                    sys.stdout.write("\n")