From 89ff92664ff06a974e37c15ab663394271ac4a10 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Thu, 7 Nov 2024 14:59:01 +0100 Subject: ✨ update permission script - admin: delete "owns" groups when non relevent MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../commands/ishtar_update_permissions.py | 66 ++++++++++++++++++++++ 1 file changed, 66 insertions(+) create mode 100644 ishtar_common/management/commands/ishtar_update_permissions.py (limited to 'ishtar_common/management/commands/ishtar_update_permissions.py') diff --git a/ishtar_common/management/commands/ishtar_update_permissions.py b/ishtar_common/management/commands/ishtar_update_permissions.py new file mode 100644 index 000000000..d8d8944bf --- /dev/null +++ b/ishtar_common/management/commands/ishtar_update_permissions.py @@ -0,0 +1,66 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- + +from argparse import RawTextHelpFormatter +import datetime +import sys + +from django.conf import settings +from django.core.management.base import BaseCommand + +from ishtar_common import models +from ishtar_common.utils import BColors + + +def update_permissions(quiet=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(): + user.generate_permission() + updated += 1 + if not quiet: + msg = BColors.format("OKGREEN", f"\r{updated} user permissions updated") + sys.stdout.write(msg) + sys.stdout.write("\n") + + +def percent(current, total): + return f"{(current + 1) / total * 100:.1f}".rjust(4, "0") + "%" + + +def get_time(): + return datetime.datetime.now().isoformat().split(".")[0] + + +class Command(BaseCommand): + help = "Update permissions" + + def parser_error(self, message=""): + sys.stderr.write(f"{message}\n") + self.parser.print_help() + sys.exit(2) + + def create_parser(self, *args, **kwargs): + parser = super(Command, self).create_parser(*args, **kwargs) + parser.formatter_class = RawTextHelpFormatter + self.parser = parser + parser.error = self.parser_error + return parser + + def add_arguments(self, parser): + parser.add_argument( + "--quiet", dest="quiet", action="store_true", help="Quiet output" + ) + + def handle(self, *args, **options): + settings.USE_BACKGROUND_TASK = False + quiet = options["quiet"] + if not quiet: + msg = BColors.format("HEADER", f"[{get_time()}] Updating permissions\n") + sys.stdout.write(msg) + update_permissions(quiet) + sys.exit(1) -- cgit v1.2.3