summaryrefslogtreecommitdiff
path: root/yaksh/management/commands/add_group.py
diff options
context:
space:
mode:
authorankitjavalkar2015-09-14 12:28:41 +0530
committerankitjavalkar2015-09-15 11:33:59 +0530
commit7a60f782995eed0352f3172f445bb35ae8ee6aa9 (patch)
tree91d6d12b0a494dd9777fd1120e92db5763bf009a /yaksh/management/commands/add_group.py
parent32cde26f9807a29eb1fae0833e353212d43d7802 (diff)
downloadonline_test-7a60f782995eed0352f3172f445bb35ae8ee6aa9.tar.gz
online_test-7a60f782995eed0352f3172f445bb35ae8ee6aa9.tar.bz2
online_test-7a60f782995eed0352f3172f445bb35ae8ee6aa9.zip
Remove testapp dir, remove reference to testapp in paths, files
Diffstat (limited to 'yaksh/management/commands/add_group.py')
-rw-r--r--yaksh/management/commands/add_group.py33
1 files changed, 33 insertions, 0 deletions
diff --git a/yaksh/management/commands/add_group.py b/yaksh/management/commands/add_group.py
new file mode 100644
index 0000000..03ef103
--- /dev/null
+++ b/yaksh/management/commands/add_group.py
@@ -0,0 +1,33 @@
+'''
+ This command adds moderator group with permissions to add, change and delete
+ the objects in the exam app.
+ We can modify this command to add more groups by providing arguments.
+ Arguments like group-name, app-name can be passed.
+'''
+
+# 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 = '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)
+ # Get list of permissions for the models
+ permission_list = Permission.objects.filter(content_type=content_types)
+ for permission in permission_list:
+ group.permissions.add(permission)
+ group.save()
+
+ self.stdout.write('Moderator group added successfully')