Changeset 7573 for main/waeup.sirp/trunk/tools

Timestamp:

2 Feb 2012, 22:49:54 (13 years ago)

Author:

uli

Message:

As some constants (input, output-path) are not really constants any
more, we can treat them like usual vars and remove them from config
section.

File:

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

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

@@ -29 +29 @@
 into the specified output file.
 """
+import csv
+import datetime
+import sys
 
 ##
 ## CONFIGURATION SECTION
 ##
-# file with input data
-INPUT_FILE = '%s' % sys.argv[1]
-
-# file written with modified output
-OUTPUT_FILE = '%s_edited.csv' % sys.argv[1].split('.')[0]
-
 # keys are fieldnames in input file, values are methods of class
 # Converter (see below)
@@ -85 +82 @@
 ##
 
-import csv
-import datetime
-import sys
-
 def convert_fieldnames(fieldnames):
     """Replace input fieldnames by fieldnames of COLNAME_MAPPING.
@@ -186 +179 @@
 
 def main():
-    reader = csv.DictReader(open(INPUT_FILE, 'rb'))
+    input_file = '%s' % sys.argv[1]
+    output_file = '%s_edited.csv' % sys.argv[1].split('.')[0]
+    reader = csv.DictReader(open(input_file, 'rb'))
     writer = None
 
     for num, row in enumerate(reader):
         if num == 0:
-            writer = csv.DictWriter(open(OUTPUT_FILE, 'wb'), reader.fieldnames)
+            writer = csv.DictWriter(open(output_file, 'wb'), reader.fieldnames)
             print "FIELDS: "
             for x, y in enumerate(reader.fieldnames):
@@ -197 +192 @@
             header = convert_fieldnames(reader.fieldnames)
             writer.writerow(header)
+        print row['phone']
         for key, value in row.items():
             if not key in OPTIONS.keys():
@@ -208 +204 @@
         writer.writerow(row)
 
-    print "Output written to %s" % OUTPUT_FILE
+    print "Output written to %s" % output_file