✨ management command - ishtar_update_permissions: add force option

1 files changed, 7 insertions, 3 deletions

diff --git a/ishtar_common/management/commands/ishtar_update_permissions.py b/ishtar_common/management/commands/ishtar_update_permissions.py
index d8d8944bf..4907beba6 100644
--- a/ishtar_common/management/commands/ishtar_update_permissions.py
+++ b/ishtar_common/management/commands/ishtar_update_permissions.py
@@ -12,14 +12,14 @@ from ishtar_common import models
 from ishtar_common.utils import BColors
 
 
-def update_permissions(quiet=False):
+def update_permissions(quiet=False, force=False):
     q = models.IshtarUser.objects.all()
     nb = q.count()
     updated = 0
     for idx, user in enumerate(q.all()):
         if not quiet:
             sys.stdout.write(f"\r[{percent(idx, nb)}] {idx + 1}/{nb}")
-        if user.need_permission_refresh():
+        if force or user.need_permission_refresh():
             user.generate_permission()
             updated += 1
     if not quiet:
@@ -55,6 +55,10 @@ class Command(BaseCommand):
         parser.add_argument(
             "--quiet", dest="quiet", action="store_true", help="Quiet output"
         )
+        parser.add_argument(
+            "--force", dest="force", action="store_true",
+            help="Force regeneration for all accounts"
+        )
 
     def handle(self, *args, **options):
         settings.USE_BACKGROUND_TASK = False
@@ -62,5 +66,5 @@ class Command(BaseCommand):
         if not quiet:
             msg = BColors.format("HEADER", f"[{get_time()}] Updating permissions\n")
             sys.stdout.write(msg)
-        update_permissions(quiet)
+        update_permissions(quiet, force=options["force"])
         sys.exit(1)