summaryrefslogtreecommitdiff
path: root/yaksh
diff options
context:
space:
mode:
Diffstat (limited to 'yaksh')
-rw-r--r--yaksh/management/commands/create_moderator.py28
1 files changed, 18 insertions, 10 deletions
diff --git a/yaksh/management/commands/create_moderator.py b/yaksh/management/commands/create_moderator.py
index 96276b5..3ec012e 100644
--- a/yaksh/management/commands/create_moderator.py
+++ b/yaksh/management/commands/create_moderator.py
@@ -24,22 +24,30 @@ class Command(BaseCommand):
group_name = 'moderator'
group = create_group(group_name, app_label)
if group and isinstance(group, Group):
- self.stdout.write('Moderator group added successfully')
+ self.stdout.write(self.style.SUCCESS(
+ '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)
+ 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(
+ self.style.WARNING(
+ '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)
- )
+ user.profile.is_moderator = True
+ user.profile.save()
+ self.stdout.write(
+ self.style.SUCCESS(
+ 'Successfully added User "{0}"'
+ ' to Moderator group'.format(uname)
+ )
+ )