Changeset 7516 for main

Timestamp:

26 Jan 2012, 09:38:22 (13 years ago)

Author:

uli

Message:

• Make col names a configuration setting.
• Remove doubled gender converter.

File:

• main/waeup.sirp/trunk/tools/fix_import_file.py (modified) (4 diffs)

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

@@ -53 +53 @@
     'current_session': 'session',
     }
+
+# Mapping input file colnames --> output file colnames
+COLNAME_MAPPING = {
+    'jamb_reg_no': 'reg_no',
+    'birthday': 'date_of_birth',
+    }
 ##
 ## END OF CONFIG
@@ -61 +67 @@
 import sys
 
-def convert_fieldnames(reader):
-    header = {}
-    for x in reader.fieldnames:
-        if x == 'jamb_reg_no':
-            header['jamb_reg_no'] = 'reg_no'
-        elif x == 'birthday':
-            header['birthday'] = 'date_of_birth'
-        else:
-            header[x] = x
+def convert_fieldnames(fieldnames):
+    """Replace input fieldnames by fieldnames of COLNAME_MAPPING.
+    """
+    header = dict([(name, name) for name in fieldnames])
+    for in_name, out_name in COLNAME_MAPPING.items():
+        if in_name not in header:
+            continue
+        header[in_name] = out_name
     return header
 
@@ -89 +94 @@
         else:
             return 9999
-
-    @classmethod
-    def gender(self, value):
-        """ 'True'/'False' --> 'female'/'male'
-        """
-        if value == 'True':
-            value = 'female'
-        elif value == 'False':
-            value = 'male'
-        else:
-            value = ''
-        return value
 
     @classmethod
@@ -166 +159 @@
         for x, y in enumerate(reader.fieldnames):
             print x, y
-        header = convert_fieldnames(reader)
+        header = convert_fieldnames(reader.fieldnames)
         writer.writerow(header)
     for key, value in row.items():