diff options
Diffstat (limited to 'yaksh/management')
-rw-r--r-- | yaksh/management/commands/create_moderator.py | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/yaksh/management/commands/create_moderator.py b/yaksh/management/commands/create_moderator.py index 96276b5..b6bbc50 100644 --- a/yaksh/management/commands/create_moderator.py +++ b/yaksh/management/commands/create_moderator.py @@ -24,7 +24,8 @@ 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']: @@ -35,11 +36,13 @@ class Command(BaseCommand): uname) ) if user in group.user_set.all(): - self.stdout.write('User "{0}" is ' + 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}"' + user.profile.is_moderator = True + user.profile.save() + self.stdout.write(self.style.SUCCESS( + 'Successfully added User "{0}"' ' to Moderator group'.format(uname) - ) + )) |