Removed EDK API extract, added Assign Groups command

app/tools/management/commands/assigngroup.py

@@ -0,0 +1,49 @@
+#!/usr/bin/env python
+
+import unicodedata
+import re
+from optparse import make_option
+from django.core.management.base import BaseCommand, CommandError
+from eve_api.models import EVEPlayerCharacter
+from eve_api.app_defines import API_STATUS_OK
+
+class Command(BaseCommand):
+    help = ("Assigns a group to corporation or alliance.")
+
+    option_list = BaseCommand.option_list + (
+        make_option('-a', '--alli', action='store', dest='alliance', help='Alliance ID'),
+        make_option('-c', '--corp', action='store', dest='corporation', help='Corp ID'),
+        make_option('-g', '--group', action='store', dest='group', help='Group ID to assign to the corp/alliance')
+    )
+
+    requires_model_validation = False
+
+    def handle(self, **options):
+
+        if not options.get('alliance', None) and not options.get('corporation', None):
+            raise CommandError("Please provide either a corporation or alliance")
+
+        if options.get('alliance', None) and options.get('corporation', None):
+            raise CommandError("Use either alliance or corporation, not both")
+
+        if not options.get('group', None):
+            raise CommandError("You need to specify a domain")
+
+        alliance = options.get('alliance', None)
+        corporation = options.get('corporation', None)
+        group = Group.objects.get(id=options.get('group'))
+
+        args = {'eveaccount__api_status': API_STATUS_OK}
+
+        if alliance:
+            args['eveaccount__characters__corporation__alliance__id'] = alliance
+        elif corporation:
+            args['eveaccount__characters__corporation__id'] = corporation
+
+        users = User.objects.filter(**args).distinct()
+        print "%s user(s) to update." % users.count()
+        for user in users:
+            user.groups.add(group)
+        print "Done."
+
+