changeset 13:2213edf59af4

Further improvements to ease testing.
author Tom Fredrik Blenning Klaussen <bfg@blenning.no>
date Wed, 14 Nov 2012 22:18:11 +0100
parents 07e34df56b2a
children d98bd27f77b2
files codeOptimizer.py
diffstat 1 files changed, 8 insertions(+), 7 deletions(-) [+]
line wrap: on
line diff
--- a/codeOptimizer.py	Wed Nov 14 22:08:58 2012 +0100
+++ b/codeOptimizer.py	Wed Nov 14 22:18:11 2012 +0100
@@ -62,21 +62,22 @@
                 print str(r) + ' : ' + file.lines[r].rstrip()
 
 
-def codeOptimizerFromFilename(conffile, infiles):
+def codeOptimizerFromFilename(conffile, infiles = []):
     try:
         options = Config(conffile)
     except:
-        print sys.argv[1] + ' is not a valid xml file'
+        print conffile + ' is not a valid xml file'
         sys.exit(1)
 
     codeOptimizer(options, infiles)
 
 
 
-if len(sys.argv) < 2:
-    usage(sys.argv[0])
-    sys.exit(1)
+if __name__ == '__main__':
+    if len(sys.argv) < 2:
+        usage(sys.argv[0])
+        sys.exit(1)
 
-infiles = sys.argv[2:]
+    infiles = sys.argv[2:]
 
-codeOptimizerFromFilename(sys.argv[1], infiles)
+    codeOptimizerFromFilename(sys.argv[1], infiles)