Refactoring, using const, removing global variables, passing some new arguments.

[treecutter.git] / treecutter / main.py

diff --git a/treecutter/main.py b/treecutter/main.py
old mode 100755 (executable)
new mode 100644 (file)
index 3c30824..ba09eaf
--- a/treecutter/main.py
+++ b/treecutter/main.py
@@ -1,21 +1,9 @@
 #!/usr/bin/python
 import os
-import fnmatch
-import subprocess
-import amara
-import re
-import tempfile
-import errno
 import time
 import argparse
-import shutil
-import pygraphviz as pgv
-import glob
-import gettext
-import shutil
-from amara import bindery
-from amara.xslt import transform
-from Cheetah.Template import Template
+from treecutter.directory import Directory
+from treecutter.sitemap import Sitemap
 
 parser = argparse.ArgumentParser(description='Process docbook article tree.')
 parser.add_argument('--style', nargs='?',
@@ -24,13 +12,6 @@ parser.add_argument('--output', nargs='?',
                     default=os.path.dirname(os.getcwd())+'/htdocs/')
 args = parser.parse_args()
 
-style_xslt = args.style+"docbook.xsl"
-outputdir = args.output
-
-tmptarget = tempfile.mkdtemp()+'/'
-
-MAXLEVEL = 10000
-
 ts = time.time()
 dir_ = Directory()
 sitemap = Sitemap()
@@ -50,10 +31,10 @@ if len(missing)+len(removed) != 0:
     sitemap.write_map()
 sitemap.graph()
 
-sitemap.process()
+sitemap.process(args.style)
 
 t1 = time.time()
-sitemap.publish()
+sitemap.publish(args.output,args.style)
 t2 = time.time()
 print "Publish  [%5.2f s]" % (round(t2-t1,2))
 print "Total    [%5.2f s]" % (round(t2-ts,2))