summaryrefslogtreecommitdiff
path: root/ishtar_common/management/commands/ishtar_update_permissions.py
blob: d8d8944bfc26b4aba5c0dcfec15067ff9bb757e7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
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)