Changeset 7572 for main/waeup.sirp/trunk/tools/fix_import_file.py

Timestamp:

2 Feb 2012, 22:40:21 (13 years ago)

Author:

uli

Message:

Wrap main programme in a function. Also put config part back to top of
file.

File:

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

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

@@ -30 +30 @@
 """
 
-import sys
-
-
-if len(sys.argv) != 2:
-    print 'Usage: python fix_import_file.py <filename>'
-    sys.exit(1)
-
 ##
 ## CONFIGURATION SECTION
@@ -192 +185 @@
 
 
-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)
-        print "FIELDS: "
-        for x, y in enumerate(reader.fieldnames):
-            print x, y
-        header = convert_fieldnames(reader.fieldnames)
-        writer.writerow(header)
-    for key, value in row.items():
-        if not key in OPTIONS.keys():
-            continue
-        conv_name = OPTIONS[key]
-        converter = getattr(Converters, conv_name, None)
-        if converter is None:
-            print "WARNING: cannot find converter %s" % conv_name
-            continue
-        row[key] = converter(row[key])
-    writer.writerow(row)
-
-print "Output written to %s" % OUTPUT_FILE
+def main():
+    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)
+            print "FIELDS: "
+            for x, y in enumerate(reader.fieldnames):
+                print x, y
+            header = convert_fieldnames(reader.fieldnames)
+            writer.writerow(header)
+        for key, value in row.items():
+            if not key in OPTIONS.keys():
+                continue
+            conv_name = OPTIONS[key]
+            converter = getattr(Converters, conv_name, None)
+            if converter is None:
+                print "WARNING: cannot find converter %s" % conv_name
+                continue
+            row[key] = converter(row[key])
+        writer.writerow(row)
+
+    print "Output written to %s" % OUTPUT_FILE
+
+
+if __name__ == '__main__':
+    if len(sys.argv) != 2:
+        print 'Usage: %s <filename>' % __file__
+        sys.exit(1)
+    main()