From 62229427b651a52a2fc4d24d2a0bf764553277c0 Mon Sep 17 00:00:00 2001
From: ankitjavalkar
Date: Wed, 14 Mar 2018 12:32:37 +0530
Subject: Add one command to create moderator group and add users to the group

---
 yaksh/management/commands/create_moderator.py | 48 +++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 yaksh/management/commands/create_moderator.py

(limited to 'yaksh/management/commands/create_moderator.py')

diff --git a/yaksh/management/commands/create_moderator.py b/yaksh/management/commands/create_moderator.py
new file mode 100644
index 0000000..3bbe462
--- /dev/null
+++ b/yaksh/management/commands/create_moderator.py
@@ -0,0 +1,48 @@
+'''
+   This command creates a moderator group and adds users to the 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 User, Group, Permission
+from django.contrib.contenttypes.models import ContentType
+from django.db.utils import IntegrityError
+
+# Yaksh imports
+from yaksh.models import Profile
+
+class Command(BaseCommand):
+    help = 'Adds users to the moderator group'
+
+    def add_arguments(self, parser):
+        # Positional arguments
+        parser.add_argument('usernames', nargs='*', type=str)
+
+    def handle(self, *args, **options):
+        app_label = 'yaksh'
+
+        try:
+            group = Group.objects.get(name='moderator')
+        except Group.DoesNotExist:
+            group = Group(name='moderator')
+            group.save()
+            # 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')
+
+        if options['usernames']:
+            for uname in options['usernames']:
+                try:
+                    user = User.objects.get(username=uname)
+                except User.DoesNotExist:
+                    raise CommandError('User "{0}" does not exist'.format(uname))
+                if user in group.user_set.all():
+                    self.stdout.write('User "{0}" is already a Moderator'.format(uname))
+                else:
+                    group.user_set.add(user)
+                    self.stdout.write('Successfully added User "{0}" to Moderator group'.format(uname))
-- 
cgit 


From 97b657edc2a323f832c81f0e34ce5761bd21f7e9 Mon Sep 17 00:00:00 2001
From: adityacp
Date: Thu, 24 May 2018 12:07:39 +0530
Subject: Pep8 changes

---
 yaksh/management/commands/create_moderator.py | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

(limited to 'yaksh/management/commands/create_moderator.py')

diff --git a/yaksh/management/commands/create_moderator.py b/yaksh/management/commands/create_moderator.py
index 3bbe462..86489d5 100644
--- a/yaksh/management/commands/create_moderator.py
+++ b/yaksh/management/commands/create_moderator.py
@@ -1,5 +1,6 @@
 '''
-   This command creates a moderator group and adds users to the moderator group with permissions to add, change and delete
+   This command creates a moderator group and adds users to the moderator group
+   with permissions to add, change and delete
    the objects in the exam app.
 '''
 
@@ -7,10 +8,7 @@
 from django.core.management.base import BaseCommand, CommandError
 from django.contrib.auth.models import User, Group, Permission
 from django.contrib.contenttypes.models import ContentType
-from django.db.utils import IntegrityError
 
-# Yaksh imports
-from yaksh.models import Profile
 
 class Command(BaseCommand):
     help = 'Adds users to the moderator group'
@@ -30,7 +28,8 @@ class Command(BaseCommand):
             # 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)
+            permission_list = Permission.objects.filter(
+                content_type__in=content_types)
             group.permissions.add(*permission_list)
             group.save()
             self.stdout.write('Moderator group added successfully')
@@ -40,9 +39,15 @@ class Command(BaseCommand):
                 try:
                     user = User.objects.get(username=uname)
                 except User.DoesNotExist:
-                    raise CommandError('User "{0}" does not exist'.format(uname))
+                    raise CommandError('User "{0}" does not exist'.format(
+                        uname)
+                    )
                 if user in group.user_set.all():
-                    self.stdout.write('User "{0}" is already a Moderator'.format(uname))
+                    self.stdout.write('User "{0}" is '
+                                      'already a Moderator'.format(uname)
+                                      )
                 else:
                     group.user_set.add(user)
-                    self.stdout.write('Successfully added User "{0}" to Moderator group'.format(uname))
+                    self.stdout.write('Successfully added User "{0}"'
+                                      ' to Moderator group'.format(uname)
+                                      )
-- 
cgit