From 5923675c35a21d442860b58e04b8f3c5ba749dda Mon Sep 17 00:00:00 2001
From: Max Rothman <max@edx.org>
Date: Tue, 1 Sep 2015 11:36:49 -0400
Subject: [PATCH] Add management command for granting superuser access

---
 .../management/commands/set_superuser.py      | 46 +++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 common/djangoapps/student/management/commands/set_superuser.py

diff --git a/common/djangoapps/student/management/commands/set_superuser.py b/common/djangoapps/student/management/commands/set_superuser.py
new file mode 100644
index 0000000000..068742bc8c
--- /dev/null
+++ b/common/djangoapps/student/management/commands/set_superuser.py
@@ -0,0 +1,46 @@
+"""Management command to grant or revoke superuser access for one or more users"""
+
+from optparse import make_option
+from django.contrib.auth.models import User
+from django.core.management.base import BaseCommand, CommandError
+
+
+class Command(BaseCommand):
+    """Management command to grant or revoke superuser access for one or more users"""
+    option_list = BaseCommand.option_list + (
+        make_option('--unset',
+                    action='store_true',
+                    dest='unset',
+                    default=False,
+                    help='Set is_superuser to False instead of True'),
+    )
+
+    args = '<user|email> [user|email ...]>'
+    help = """
+    This command will set is_superuser to true for one or more users.
+    Lookup by username or email address, assumes usernames
+    do not look like email addresses.
+    """
+
+    def handle(self, *args, **options):
+        if len(args) < 1:
+            raise CommandError('Usage is set_superuser {0}'.format(self.args))
+
+        for user in args:
+            try:
+                if '@' in user:
+                    userobj = User.objects.get(email=user)
+                else:
+                    userobj = User.objects.get(username=user)
+
+                if options['unset']:
+                    userobj.is_superuser = False
+                else:
+                    userobj.is_superuser = True
+
+                userobj.save()
+
+            except Exception as err:  # pylint: disable=broad-except
+                print "Error modifying user with identifier {}: {}: {}".format(user, type(err).__name__, err.message)
+
+        print 'Success!'