Changeset beaf731 for asadb/util

Timestamp:

Dec 26, 2011, 5:28:39 AM (15 years ago)

Author:

Alex Dehnert <adehnert@…>

Branches:

master, space-access, stable, stage

Children:

41146ff

Parents:

1ffd96e

git-author:

Alex Dehnert <adehnert@…> (12/26/11 05:28:39)

git-committer:

Alex Dehnert <adehnert@…> (12/26/11 05:28:39)

Message:

Clean up util/import_db.py slightly

File:

• asadb/util/import_db.py (modified) (2 diffs)

asadb/util/import_db.py

@@ -33 +33 @@
     else: return int(number)
 
+def import_group(d):
+    g = groups.models.Group()
+    g.pk                = d['ASA_STUDENT_GROUP_KEY']
+    g.name              = d['STUDENT_GROUP_NAME']
+    g.abbreviation      = d['STUDENT_GROUP_ACRONYM']
+    g.description       = d['STUDENT_GROUP_DESCRIPTION']
+    cat_name = d['GROUP_ACTIVITY_CATEGORY']
+    try:
+        g.activity_category = groups.models.ActivityCategory.objects.get(name=cat_name)
+    except groups.models.ActivityCategory.DoesNotExist:
+        print ">> Unknown category '%s' on group '%s'" % (cat_name, g.name, )
+        pass
+    class_name = d['GROUP_CLASS']
+    status_name = d['GROUP_STATUS']
+    funding_name = d['GROUP_FUNDING_TYPE']
+    if class_name == 'Standard':
+        if status_name == 'Provisional':
+            class_name = 'Unfunded'
+            status_name = 'Active'
+        elif status_name == 'Active':
+            class_name = 'MIT-funded'
+        elif status_name == 'Suspended' or status_name == 'Derecognized':
+            class_name = 'MIT-funded'
+    g.group_class = groups.models.GroupClass.objects.get(name=class_name)
+    g.group_status = groups.models.GroupStatus.objects.get(name=status_name)
+    if funding_name == 'none':
+        g.group_funding = None
+    else:
+        g.group_funding = groups.models.GroupFunding.objects.get(name=funding_name)
+    g.website_url       = d['WEBSITE_URL']
+    g.constitution_url  = d['CONSTITUTION_WEB_URL']
+    g.meeting_times     = d['MEETING_TIMES']
+    g.advisor_name      = d['ADVISOR']
+    g.num_undergrads    = convert_to_int(d['NUM_OF_UNDERGRADUATE'])
+    g.num_grads         = convert_to_int(d['NUM_OF_GRADUATE'])
+    g.num_community     = convert_to_int(d['NUM_OF_COMMUNITY'])
+    g.num_other         = convert_to_int(d['NUM_OF_OTHERS'])
+    g.group_email       = canonicalize_email(d['STUDENT_GROUP_EMAIL'])
+    g.officer_email     = canonicalize_email(d['OFFICER_EMAIL'])
+    try:
+        g.main_account_id   = convert_to_int(d['MAIN_ACCOUNT_ID'])
+    except ValueError:
+        if d['MAIN_ACCOUNT_ID'] == "contact LWard":
+            print "Ignoring account ID contact LWard..."
+            g.main_account_id = None
+        else:
+            raise
+    g.funding_account_id= convert_to_int(d['FUNDING_ACCOUNT_ID'])
+    g.athena_locker     = d['ATHENA_LOCKER']
+    g.recognition_date  = db_parse_date(d['RECOGNITION_DATE'])
+    g.update_date       = db_parse_date(d['UPDATE_DATE'])
+    g.updater           = d['UPDATER']
+    g.save()
+
 if __name__ == '__main__':
     indb = sys.stdin
@@ -40 +94 @@
         d = dictize_line(header, line)
         print d
-        g = groups.models.Group()
-        g.pk                = d['ASA_STUDENT_GROUP_KEY']
-        g.name              = d['STUDENT_GROUP_NAME']
-        g.abbreviation      = d['STUDENT_GROUP_ACRONYM']
-        g.description       = d['STUDENT_GROUP_DESCRIPTION']
-        cat_name = d['GROUP_ACTIVITY_CATEGORY']
-        try:
-            g.activity_category = groups.models.ActivityCategory.objects.get(name=cat_name)
-        except groups.models.ActivityCategory.DoesNotExist:
-            print ">> Unknown category '%s' on group '%s'" % (cat_name, g.name, )
-            pass
-        class_name = d['GROUP_CLASS']
-        status_name = d['GROUP_STATUS']
-        funding_name = d['GROUP_FUNDING_TYPE']
-        if class_name == 'Standard':
-            if status_name == 'Provisional':
-                class_name = 'Unfunded'
-                status_name = 'Active'
-            elif status_name == 'Active':
-                class_name = 'MIT-funded'
-            elif status_name == 'Suspended' or status_name == 'Derecognized':
-                class_name = 'MIT-funded'
-        g.group_class = groups.models.GroupClass.objects.get(name=class_name)
-        g.group_status = groups.models.GroupStatus.objects.get(name=status_name)
-        if funding_name == 'none':
-            g.group_funding = None
-        else:
-            g.group_funding = groups.models.GroupFunding.objects.get(name=funding_name)
-        g.website_url       = d['WEBSITE_URL']
-        g.constitution_url  = d['CONSTITUTION_WEB_URL']
-        g.meeting_times     = d['MEETING_TIMES']
-        g.advisor_name      = d['ADVISOR']
-        g.num_undergrads    = convert_to_int(d['NUM_OF_UNDERGRADUATE'])
-        g.num_grads         = convert_to_int(d['NUM_OF_GRADUATE'])
-        g.num_community     = convert_to_int(d['NUM_OF_COMMUNITY'])
-        g.num_other         = convert_to_int(d['NUM_OF_OTHERS'])
-        g.group_email       = canonicalize_email(d['STUDENT_GROUP_EMAIL'])
-        g.officer_email     = canonicalize_email(d['OFFICER_EMAIL'])
-        try:
-            g.main_account_id   = convert_to_int(d['MAIN_ACCOUNT_ID'])
-        except ValueError:
-            if d['MAIN_ACCOUNT_ID'] == "contact LWard":
-                print "Ignoring account ID contact LWard..."
-                g.main_account_id = None
-            else:
-                raise
-        g.funding_account_id= convert_to_int(d['FUNDING_ACCOUNT_ID'])
-        g.athena_locker     = d['ATHENA_LOCKER']
-        g.recognition_date  = db_parse_date(d['RECOGNITION_DATE'])
-        g.update_date       = db_parse_date(d['UPDATE_DATE'])
-        g.updater           = d['UPDATER']
-        g.save()
+        import_group(d)