yml2c
changeset 64 4e4782144f4f
parent 50 963c1b542610
child 65 d659b8c2ed22
child 68 26120af11afd
--- a/yml2c	Thu Mar 19 10:47:36 2020 +0100
+++ b/yml2c	Thu Mar 19 15:20:28 2020 +0100
@@ -28,70 +28,76 @@
         msg = codecs.encode(msg, sys.stderr.encoding)
     sys.stderr.write(msg)
 
-optParser = OptionParser()
-optParser.add_option("-C", "--old-syntax", action="store_true", dest="old_syntax",
-        help="syntax of YML 2 version 1.x (compatibility mode)", default=False)
-optParser.add_option("-D", "--emit-linenumbers", action="store_true", dest="emitlinenumbers",
-        help="emit line numbers into the resulting XML for debugging purposes", default=False)
-optParser.add_option("-E", "--encoding", dest="encoding", metavar="ENCODING", default=locale.getdefaultlocale()[1],
-        help="encoding of input files (default to locale)")
-optParser.add_option("-I", "--include", dest="includePathText", metavar="INCLUDE_PATH",
-        help="precede YML_PATH by a colon separated INCLUDE_PATH to search for include files")
-optParser.add_option("-m", "--omit-empty-parm-tags", action="store_true", dest="omitemptyparm",
-        help="does nothing (only there for compatibility reasons)", default=False)
-optParser.add_option("-n", "--normalization", dest="normalization", metavar="NORMALIZATION", default="NFC",
-        help="Unicode normalization (none, NFD, NFKD, NFC, NFKC, FCD, default is NFC)")
-optParser.add_option("-o", "--output", dest="outputFile", metavar="FILE",
-        help="place output in file FILE")
-optParser.add_option("-p", "--parse-only", action="store_true", dest="parseonly",
-        help="parse only, then output pyAST as text to stdout", default=False)
-optParser.add_option("-V", "--version", action="callback", callback=printInfo, help="show version info")
-(options, args) = optParser.parse_args()
+def main():
+    optParser = OptionParser()
+    optParser.add_option("-C", "--old-syntax", action="store_true", dest="old_syntax",
+            help="syntax of YML 2 version 1.x (compatibility mode)", default=False)
+    optParser.add_option("-D", "--emit-linenumbers", action="store_true", dest="emitlinenumbers",
+            help="emit line numbers into the resulting XML for debugging purposes", default=False)
+    optParser.add_option("-E", "--encoding", dest="encoding", metavar="ENCODING", default=locale.getdefaultlocale()[1],
+            help="encoding of input files (default to locale)")
+    optParser.add_option("-I", "--include", dest="includePathText", metavar="INCLUDE_PATH",
+            help="precede YML_PATH by a colon separated INCLUDE_PATH to search for include files")
+    optParser.add_option("-m", "--omit-empty-parm-tags", action="store_true", dest="omitemptyparm",
+            help="does nothing (only there for compatibility reasons)", default=False)
+    optParser.add_option("-n", "--normalization", dest="normalization", metavar="NORMALIZATION", default="NFC",
+            help="Unicode normalization (none, NFD, NFKD, NFC, NFKC, FCD, default is NFC)")
+    optParser.add_option("-o", "--output", dest="outputFile", metavar="FILE",
+            help="place output in file FILE")
+    optParser.add_option("-p", "--parse-only", action="store_true", dest="parseonly",
+            help="parse only, then output pyAST as text to stdout", default=False)
+    optParser.add_option("-V", "--version", action="callback", callback=printInfo, help="show version info")
+    (options, args) = optParser.parse_args()
 
-if options.old_syntax:
-    oldSyntax()
+    if options.old_syntax:
+        oldSyntax()
 
-if options.emitlinenumbers:
-    backend.emitlinenumbers = True
+    if options.emitlinenumbers:
+        backend.emitlinenumbers = True
 
-backend.encoding = options.encoding
+    backend.encoding = options.encoding
 
-try:
-    if options.includePathText:
-        backend.includePath = options.includePathText.split(':')
+    try:
+        if options.includePathText:
+            backend.includePath = options.includePathText.split(':')
 
-    dirs = os.environ.get('YML_PATH', '.').split(':') + YML_DEFAULT_PATH
-    backend.includePath.extend(dirs)
+        dirs = os.environ.get('YML_PATH', '.').split(':') + YML_DEFAULT_PATH
+        backend.includePath.extend(dirs)
 
-    files = fileinput.input(args, mode="rU", openhook=fileinput.hook_encoded(options.encoding))
+        files = fileinput.input(args, mode="rU", openhook=fileinput.hook_encoded(options.encoding))
 
-    ymlC = ymlCStyle()
-    result = parse(ymlC, files, True, comment, packrat=True)
+        ymlC = ymlCStyle()
+        result = parse(ymlC, files, True, comment, packrat=True)
 
-    if options.parseonly:
-        print(result)
-    else:
-        result = backend.finish(result)
-        if options.normalization != "none":
-            result = unicodedata.normalize(options.normalization, result)
+        if options.parseonly:
+            print(result)
+        else:
+            result = backend.finish(result)
+            if options.normalization != "none":
+                result = unicodedata.normalize(options.normalization, result)
 
-        if options.outputFile and options.outputFile != "-":
-            outfile = open(options.outputFile, "wb")
-            outfile.write(codecs.encode(result, options.encoding))
-            outfile.close()
-        else:
-            sys.stdout.buffer.write(codecs.encode(result, options.encoding))
-            print()
+            if options.outputFile and options.outputFile != "-":
+                outfile = open(options.outputFile, "wb")
+                outfile.write(codecs.encode(result, options.encoding))
+                outfile.close()
+            else:
+                sys.stdout.buffer.write(codecs.encode(result, options.encoding))
+                print()
 
-except KeyboardInterrupt:
-    w("\n")
-    sys.exit(1)
-except KeyError as msg:
-    w("not found: " + u(msg) + "\n")
-    sys.exit(4)
-except LookupError as msg:
-    w("not found: " + u(msg) + "\n")
-    sys.exit(4)
-except Exception as msg:
-    w(msg)
-    sys.exit(5)
+    except KeyboardInterrupt:
+        w("\n")
+        sys.exit(1)
+    except KeyError as msg:
+        w("not found: " + u(msg) + "\n")
+        sys.exit(4)
+    except LookupError as msg:
+        w("not found: " + u(msg) + "\n")
+        sys.exit(4)
+    except Exception as msg:
+        w(msg)
+        sys.exit(5)
+
+
+if __name__ == "__main__":
+    sys.exit(main())
+