Merge pull request #857 from dotKom/new-member-lists-sync

New member lists sync

onlineweb4/settings/base.py

@@ -174,6 +174,7 @@
     'apps.mommy',
     'apps.profiles',
     'apps.genfors',
+    'scripts',
 
      # Wiki
     'wiki',

scripts/management/commands/new_membership_lists.py

@@ -0,0 +1,74 @@
+# -*- coding: utf-8 -*-
+
+import pytz
+import datetime
+
+from django.db.utils import IntegrityError
+from django.core.management.base import BaseCommand, CommandError
+from django.utils import timezone
+
+from apps.authentication.models import AllowedUsername
+
+
+class Command(BaseCommand):
+
+    def handle(self, *args, **kwargs):
+        now = timezone.now()
+        now = datetime.datetime(now.year, 9, 15, 0, 0)
+        # subtract a year if we're adding people with this script in the spring
+        if now.month < 6:
+            now = now - datetime.timedelta(days=365)
+        pytz.timezone(timezone.get_default_timezone_name()).localize(now)
+
+        expiration_date = now
+        note = ''
+
+        if len(args) != 1:
+            self.stdout.write("Error: You need to specify a filename as the first argument.")
+            return
+        filename = args[0]
+        f = open(filename, 'r')
+
+        new_count = 0
+        update_count = 0
+
+        for line in f:
+            line = line.strip()
+            # skip empty lines
+            if not line:
+                continue
+            # set the correct expiry date according to FOS.
+            if line == 'MIT' or line == 'mit':
+                expiration_date = now + datetime.timedelta(days=365*2)
+                note = 'Master %d' % now.year
+                continue
+            if line == 'BIT' or line == 'bit':
+                expiration_date = now + datetime.timedelta(days=365*3)
+                note = 'Bachelor %d' % now.year
+                continue
+
+
+            try:
+                entry = AllowedUsername(
+                                        username = line, 
+                                        registered = now,
+                                        note = note,
+                                        description = "Added by script.",
+                                        expiration_date = expiration_date
+                                        )
+                entry.save()
+
+                new_count = new_count + 1
+            except IntegrityError:
+                au = AllowedUsername.objects.get(username = line)
+                au.expiration_date = expiration_date
+                au.save()
+
+                update_count = update_count + 1
+
+
+        if new_count > 0:
+            self.stdout.write("%d new memberships added" % new_count)
+        if update_count > 0:
+            self.stdout.write("%d memberships updated" % update_count)
+