portier/source/core/management/commands/createdefaultgroup.py

26 lines
806 B
Python

from django.core.management.base import BaseCommand
from django.contrib.auth.models import Group
from django.contrib.auth.models import Permission
from django.conf import settings
PERMISSIONS = [
'add_stream',
'add_restream',
'add_pull',
]
class Command(BaseCommand):
help = 'Creates default user group with default permissions available to all users'
def handle(self, *args, **options):
default_group, _ = Group.objects.get_or_create(name=settings.DEFAULT_GROUP)
for permission in PERMISSIONS:
try:
perm = Permission.objects.get(codename=permission)
except Permission.DoesNotExist:
print("permission '%s' does not exist" % permission)
continue
default_group.permissions.add(perm)