diff --git a/seminar/management/commands/load_org_permissions.py b/seminar/management/commands/load_org_permissions.py index 58086601..c1e4a106 100644 --- a/seminar/management/commands/load_org_permissions.py +++ b/seminar/management/commands/load_org_permissions.py @@ -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: