Adding argparse to arguments --style and --output.
authorFredrik Unger <fred@tree.se>
Mon, 7 Mar 2011 20:16:26 +0000 (21:16 +0100)
committerFredrik Unger <fred@tree.se>
Mon, 7 Mar 2011 20:16:26 +0000 (21:16 +0100)
src/tree-cutter.py

index a4c24f9ce91baea8eb0ef31076b0d6dfb063b203..f1790b8652424094b8c388136070493866cdf4ae 100755 (executable)
@@ -7,15 +7,21 @@ import re
 import tempfile
 import errno
 import time
+import argparse
 from amara import bindery
 from amara.xslt import transform
 from Cheetah.Template import Template
 
-dist = os.path.dirname(os.getcwd())
-style = "default"
-style_xslt = dist+"/style/"+style+"/docbook.xsl"
-style_tmpl = dist+"/style/"+style+"/index.html.tmpl"
-outputdir = dist+"/htdocs/"
+parser = argparse.ArgumentParser(description='Process docbook article tree.')
+parser.add_argument('--style', nargs='?',
+                    default=os.path.dirname(os.getcwd())+'/style/default/')
+parser.add_argument('--output', nargs='?',
+                    default=os.path.dirname(os.getcwd())+'/htdocs/')
+args = parser.parse_args()
+
+style_xslt = args.style+"docbook.xsl"
+style_tmpl = args.style+"index.en.html.tmpl"
+outputdir = args.output
 
 valid_scripts = ['.py','.pl']
 MAXLEVEL = 10000