diff options
Diffstat (limited to 'yaksh/management/commands/create_mod_group.py')
-rw-r--r-- | yaksh/management/commands/create_mod_group.py | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/yaksh/management/commands/create_mod_group.py b/yaksh/management/commands/create_mod_group.py new file mode 100644 index 0000000..0bf08df --- /dev/null +++ b/yaksh/management/commands/create_mod_group.py @@ -0,0 +1,30 @@ +''' + This command creates a moderator group with permissions to add, change and delete + the objects in the exam app. +''' + +# django imports +from django.core.management.base import BaseCommand, CommandError +from django.contrib.auth.models import Group, Permission +from django.contrib.contenttypes.models import ContentType +from django.db.utils import IntegrityError + +class Command(BaseCommand): + help = 'Adds the moderator group' + + def handle(self, *args, **options): + app_label = 'yaksh' + group = Group(name='moderator') + try: + group.save() + except IntegrityError: + raise CommandError("The group already exits") + else: + # Get the models for the given app + content_types = ContentType.objects.filter(app_label=app_label) + # Get list of permissions for the models + permission_list = Permission.objects.filter(content_type__in=content_types) + group.permissions.add(*permission_list) + group.save() + + self.stdout.write('Moderator group added successfully') |