Create_moderator command sets is_moderator to True

- Fixes #513.

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)
-                                      )
+                                      ))