|
@ -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): |
|
|
|
|
|
try: |
|
|
orgroup = Group.objects.get(name='org') |
|
|
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: |
|
|