Merge lp:~stevanr/linaro-license-protection/add-setsuperuser-command into lp:~linaro-automation/linaro-license-protection/trunk

[Stevan Radaković]

Add setsuperuser custom command for license_protected_downloads application.

[Данило Шеган (danilo)]

I hate that we are already starting to duplicate this code. Let's thing tomorrow how we can start producing a shared django base for Linaro.

review: Approve

=== added directory 'license_protected_downloads/management'
=== added file 'license_protected_downloads/management/__init__.py'
=== added directory 'license_protected_downloads/management/commands'
=== added file 'license_protected_downloads/management/commands/__init__.py'
=== added file 'license_protected_downloads/management/commands/setsuperuser.py'
--- license_protected_downloads/management/commands/setsuperuser.py	1970-01-01 00:00:00 +0000
+++ license_protected_downloads/management/commands/setsuperuser.py	2012-08-22 21:52:18 +0000
@@ -0,0 +1,23 @@
+from django.core.management.base import BaseCommand, CommandError
+from django.contrib.auth.models import User
+
+
+class Command(BaseCommand):
+    args = '<username username ...>'
+    help = 'Sets specified user as superuser of the ' + \
+      'license_protected_downloads admin app'
+
+    def handle(self, *args, **options):
+        for username in args:
+            self.find_and_update_user(username)
+            self.stdout.write('Successfully updated user "%s" \n' % username)
+
+    def find_and_update_user(self, username):
+        try:
+            user = User.objects.get(username=username)
+        except User.DoesNotExist:
+            raise CommandError('User "%s" does not exist' % username)
+
+        user.is_staff = True
+        user.is_superuser = True
+        user.save()
 
=== modified file 'license_protected_downloads/tests/__init__.py'
--- license_protected_downloads/tests/__init__.py	2012-07-11 13:08:55 +0000
+++ license_protected_downloads/tests/__init__.py	2012-08-22 21:52:18 +0000
@@ -4,6 +4,8 @@
 from license_protected_downloads.tests.test_pyflakes import TestPyflakes
 from license_protected_downloads.tests.test_views import ViewTests
 from license_protected_downloads.tests.test_openid_auth import TestOpenIDAuth
+from license_protected_downloads.tests.test_custom_commands \
+ import SetsuperuserCommandTest
 
 #starts the test suite
 __test__ = {
@@ -13,4 +15,5 @@
     'TestPep8': TestPep8,
     'TestPyflakes': TestPyflakes,
     'TestOpenIDAuth': TestOpenIDAuth,
+    'SetsuperuserCommandTest': SetsuperuserCommandTest,
 }
 
=== added file 'license_protected_downloads/tests/test_custom_commands.py'
--- license_protected_downloads/tests/test_custom_commands.py	1970-01-01 00:00:00 +0000
+++ license_protected_downloads/tests/test_custom_commands.py	2012-08-22 21:52:18 +0000
@@ -0,0 +1,26 @@
+#!/usr/bin/env python
+
+import license_protected_downloads.management.commands.setsuperuser \
+   as setsuperuser
+from django.contrib.auth.models import User
+from django.core.management.base import CommandError
+from django.test import TestCase
+
+
+class SetsuperuserCommandTest(TestCase):
+
+    def setUp(self):
+        self.command = setsuperuser.Command()
+
+    def test_find_and_update_user_non_existing(self):
+        self.assertRaises(CommandError,
+                          self.command.find_and_update_user,
+                          ("non_existing_user"))
+
+    def test_find_and_update_user(self):
+        user = User(username="existing_user")
+        user.save()
+        self.command.find_and_update_user("existing_user")
+        user = User.objects.get(username="existing_user")
+        self.assertEquals(user.is_staff, True)
+        self.assertEquals(user.is_superuser, True)