diff options
Diffstat (limited to 'yaksh/management/commands/create_moderator.py')
| -rw-r--r-- | yaksh/management/commands/create_moderator.py | 21 |
1 files changed, 13 insertions, 8 deletions
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) + ) |