Files
website-v2/users/management/commands/create_groups.py
2021-11-21 12:54:59 +01:00

21 lines
764 B
Python
Executable File

from django.contrib.auth.models import Group, Permission
from django.core.management import BaseCommand
from django.contrib.contenttypes.models import ContentType
from versions.models import Version
class Command(BaseCommand):
def __init__(self, *args, **kwargs):
super(Command, self).__init__(*args, **kwargs)
help = "Create default groups"
def handle(self, *args, **options):
Group.objects.get(name="version_manager").delete()
# Code to add permission to group ???
# ct = ContentType.objects.get_for_model(Version)
# Now what - Say I want to add 'Can add version' permission to new_group?
Permission.objects.get(codename="can_add_version").delete()
# new_group.permissions.add(permission)