Changeset 7514 for main/waeup.sirp/trunk

Timestamp:

26 Jan 2012, 07:29:26 (13 years ago)

Author:

Henrik Bettermann

Message:

Add more converters and also change header fields in output file.

File:

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

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

@@ -49 +49 @@
     'birthday': 'date',
     'request_date': 'datetime',
+    'marit_stat': 'marit_stat',
+    'entry_session': 'session',
+    'current_session': 'session',
     }
 ##
@@ -58 +61 @@
 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
+    return header
+
 class Converters():
     """Converters to turn old-style values into new ones.
     """
     @classmethod
+    def session(self, value):
+        """ '08' --> '2008'
+        """
+        try:
+            number = int(value)
+        except ValueError:
+            return 9999
+        if number < 14:
+            return number + 2000
+        elif number in range(2000,2015):
+            return number
+        else:
+            return 9999
+
+    @classmethod
     def gender(self, value):
-        """ 'True'/'False' --> 'female'/'male'/''
+        """ 'True'/'False' --> 'female'/'male'
+        """
+        if value == 'True':
+            value = 'female'
+        elif value == 'False':
+            value = 'male'
+        else:
+            value = ''
+        return value
+
+    @classmethod
+    def marit_stat(self, value):
+        """ 'True'/'False' --> 'married'/'unmarried'
+        """
+        if value == 'True':
+            value = 'married'
+        elif value == 'False':
+            value = 'unmarried'
+        else:
+            value = ''
+        return value
+
+    @classmethod
+    def gender(self, value):
+        """ 'True'/'False' --> 'female'/'male'
         """
         if value == 'True':
@@ -83 +136 @@
         else:
             value = value.replace('/', '-')
+            # We add the hash symbol to avoid automatic date transformation
+            # in Excel and Calc for further processing
+            value += '#'
         return value
 
@@ -110 +166 @@
         for x, y in enumerate(reader.fieldnames):
             print x, y
-        header = dict([(x, x) for x in reader.fieldnames])
+        header = convert_fieldnames(reader)
         writer.writerow(header)
     for key, value in row.items():