Merge pull request #72 from rtfd/cmark-main

Add main() function to cmark.py

Author: nikolas

CommonMark/cmark.py

@@ -3,43 +3,51 @@
 import argparse
 import sys
 import CommonMark
-parser = argparse.ArgumentParser(
-    description="Process Markdown according to the CommonMark specification.")
-if sys.version_info < (3, 0):
-    reload(sys)  # noqa
-    sys.setdefaultencoding('utf-8')
-parser.add_argument(
-    'infile',
-    nargs="?",
-    type=argparse.FileType('r'),
-    default=sys.stdin,
-    help="Input Markdown file to parse, defaults to STDIN")
-parser.add_argument(
-    '-o',
-    nargs="?",
-    type=argparse.FileType('w'),
-    default=sys.stdout,
-    help="Output HTML/JSON file, defaults to STDOUT")
-parser.add_argument('-a', action="store_true", help="Print formatted AST")
-parser.add_argument('-aj', action="store_true", help="Output JSON AST")
-args = parser.parse_args()
-parser = CommonMark.Parser()
-f = args.infile
-o = args.o
-lines = []
-for line in f:
-    lines.append(line)
-data = "".join(lines)
-ast = parser.parse(data)
-if not args.a and not args.aj:
-    renderer = CommonMark.HtmlRenderer()
-    o.write(renderer.render(ast))
-    exit()
-if args.a:
-    # print ast
-    CommonMark.dumpAST(ast)
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Process Markdown according to "
+        "the CommonMark specification.")
+    if sys.version_info < (3, 0):
+        reload(sys)  # noqa
+        sys.setdefaultencoding('utf-8')
+    parser.add_argument(
+        'infile',
+        nargs="?",
+        type=argparse.FileType('r'),
+        default=sys.stdin,
+        help="Input Markdown file to parse, defaults to STDIN")
+    parser.add_argument(
+        '-o',
+        nargs="?",
+        type=argparse.FileType('w'),
+        default=sys.stdout,
+        help="Output HTML/JSON file, defaults to STDOUT")
+    parser.add_argument('-a', action="store_true", help="Print formatted AST")
+    parser.add_argument('-aj', action="store_true", help="Output JSON AST")
+    args = parser.parse_args()
+    parser = CommonMark.Parser()
+    f = args.infile
+    o = args.o
+    lines = []
+    for line in f:
+        lines.append(line)
+    data = "".join(lines)
+    ast = parser.parse(data)
+    if not args.a and not args.aj:
+        renderer = CommonMark.HtmlRenderer()
+        o.write(renderer.render(ast))
+        exit()
+    if args.a:
+        # print ast
+        CommonMark.dumpAST(ast)
+        exit()
+
+    # o.write(ast.to_JSON())
+    o.write(CommonMark.ASTtoJSON(ast))
     exit()
 
-# o.write(ast.to_JSON())
-o.write(CommonMark.ASTtoJSON(ast))
-exit()
+
+if __name__ == '__main__':
+    main()

setup.py

@@ -33,7 +33,7 @@ def run(self):
     keywords=["markup", "markdown", "commonmark"],
     entry_points={
         'console_scripts': [
-            'cmark = CommonMark.cmark:cmark',
+            'cmark = CommonMark.cmark:main',
         ]
     },
     cmdclass={'test': Test},