Add a management command to add users to moderator group

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')