Changeset 9362 for main/waeup.kofa/trunk/tools/fix_import_file.py

Timestamp:

19 Oct 2012, 22:59:41 (12 years ago)

Author:

uli

Message:

Update tools to allow new ids in imports. Support for new import col 'old_id'.

File:

• main/waeup.kofa/trunk/tools/fix_import_file.py (modified) (5 diffs)

main/waeup.kofa/trunk/tools/fix_import_file.py

@@ -36 +36 @@
 import csv
 import datetime
+import os
 import re
 import sys
@@ -159 +160 @@
     'session_id': 'p_session',
     'type': 'r_company',
+    'old_id': 'old_id',
     }
 
@@ -177 +179 @@
     }
 
+# Mapping of special cases, where new id is not deductible from old id
+# Set to `None`, if no such special cases should be considered.
+ID_MAP_CSV = None
+ID_MAP_CSV = "id_mapping.csv"
+
 ##
 ## END OF CONFIG
@@ -183 +190 @@
 # Look for the first sequence of numbers
 RE_PHONE = re.compile('[^\d]*(\d*)[^\d]*')
+
+def get_id_mapping():
+    """Returns a dict mapping from old (SRP) ids to new ids.
+
+    The dict is read from ID_MAP_CSV file. If this var is set to
+    ``None`` an empty dict is returned. The ID_MAP_CSV contains only
+    the student ids of those students, for which the standard method
+    (new_id=CHAR+old_id) does not work.
+    """
+    if ID_MAP_CSV is None:
+        return {}
+    if not os.path.isfile(ID_MAP_CSV):
+        raise IOError(
+            "No such file for mapping old to new ids: %s" % ID_MAP_CSV)
+    result = dict()
+    reader = csv.DictReader(open(ID_MAP_CSV, 'rb'))
+    for row in reader:
+        result[row['student_id']] = row['new_id']
+    return result
+
 
 def convert_fieldnames(fieldnames):
@@ -201 +228 @@
     """
 
-    @classmethod
-    def student_id(self, value, row):
+    old_new_id_map = get_id_mapping()
+
+    @classmethod
+    def student_id(cls, value, row):
         """ 'A123456' --> 'EA123456'
         """
+        value = cls.old_new_id_map.get(value, value)
         if len(value) == 7:
             return 'M' + value