From: Fredrik Unger <fred@tree.se>
Date: Mon, 7 Mar 2011 20:16:26 +0000 (+0100)
Subject: Adding argparse to arguments --style and --output.
X-Git-Tag: v1.0~34
X-Git-Url: https://source.tree.se/git?p=treecutter.git;a=commitdiff_plain;h=852b687ddb2f0289c0fe794652a7ddcaeb7daf6b

Adding argparse to arguments --style and --output.
---

diff --git a/src/tree-cutter.py b/src/tree-cutter.py
index a4c24f9..f1790b8 100755
--- a/src/tree-cutter.py
+++ b/src/tree-cutter.py
@@ -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