Bugfix - když skupina 'org' neexistuje, přidej ji.

This commit is contained in:
Tomas 'Jethro' Pokorny 2021-03-16 22:20:41 +01:00
parent 97ae472d2e
commit 4284cbd35c

View file

@ -4,6 +4,7 @@ from django.core.management.base import BaseCommand
from django.contrib.sessions.models import Session
from django.contrib.auth.models import Group, Permission
from django.contrib.contenttypes.models import ContentType
from django.core.exceptions import ObjectDoesNotExist
import json
import argparse
@ -16,7 +17,11 @@ class Command(BaseCommand):
def handle(self, *args, **options):
orgroup = Group.objects.get(name='org')
try:
orgroup = Group.objects.get(name='org')
except ObjectDoesNotExist:
orgroup = Group(name='org')
orgroup.save()
permissions = json.load(options['file'][0])
orgroup.permissions.clear()
for jp in permissions: