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.sessions.models import Session
from django.contrib.auth.models import Group, Permission from django.contrib.auth.models import Group, Permission
from django.contrib.contenttypes.models import ContentType from django.contrib.contenttypes.models import ContentType
from django.core.exceptions import ObjectDoesNotExist
import json import json
import argparse import argparse
@ -16,7 +17,11 @@ class Command(BaseCommand):
def handle(self, *args, **options): 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]) permissions = json.load(options['file'][0])
orgroup.permissions.clear() orgroup.permissions.clear()
for jp in permissions: for jp in permissions: