From 510a8d60168405111aae2fe291ba77e3d5456869 Mon Sep 17 00:00:00 2001
From: ankitjavalkar
Date: Mon, 2 Jan 2017 15:44:31 +0530
Subject: Bug Fix: Add permissions for all models related to yaksh - Earlier
 only permissions related to Answer models were being added to the group,
 fixed so that all model permissions are added

---
 yaksh/management/commands/add_group.py | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

(limited to 'yaksh')

diff --git a/yaksh/management/commands/add_group.py b/yaksh/management/commands/add_group.py
index 03ef103..624ff3c 100644
--- a/yaksh/management/commands/add_group.py
+++ b/yaksh/management/commands/add_group.py
@@ -15,7 +15,7 @@ class Command(BaseCommand):
     help = 'Adds the moderator group'
 
     def handle(self, *args, **options):
-        app = 'yaksh'
+        app_label = 'yaksh'
         group = Group(name='moderator')
         try:
             group.save()
@@ -23,11 +23,10 @@ class Command(BaseCommand):
             raise CommandError("The group already exits")
         else:
             # Get the models for the given app
-            content_types = ContentType.objects.filter(app_label=app)
+            content_types = ContentType.objects.filter(app_label=app_label)
             # 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()
+            permission_list = Permission.objects.filter(content_type__in=content_types)
+            group.permissions.add(*permission_list)
+            group.save()
 
         self.stdout.write('Moderator group added successfully')
-- 
cgit