treecutter/main.py

   1 #!/usr/bin/python
   2 import os
   3 import time
   4 import argparse
   5 from treecutter.directory import Directory
   6 from treecutter.sitemap import Sitemap
   7
   8 parser = argparse.ArgumentParser(description='Process docbook article tree.')
   9 parser.add_argument('--style', nargs='?',
  10                     default=os.path.dirname(os.getcwd())+'/style/default/')
  11 parser.add_argument('--output', nargs='?',
  12                     default=os.path.dirname(os.getcwd())+'/htdocs/')
  13 args = parser.parse_args()
  14
  15 style_xslt = args.style+"docbook.xsl"
  16 outputdir = args.output
  17
  18 tmptarget = tempfile.mkdtemp()+'/'
  19
  20 MAXLEVEL = 10000
  21
  22 ts = time.time()
  23 dir_ = Directory()
  24 sitemap = Sitemap()
  25
  26 dir_.scan()
  27 sitemap.read_map()
  28
  29 missing = dir_.set() - sitemap.set()
  30 removed = sitemap.set() - dir_.set()
  31 for page in removed:
  32     print page+' pages missing!!'
  33 for page in missing:
  34     print 'adding missing page '+page
  35     sitemap.add_link(page)
  36 if len(missing)+len(removed) != 0:
  37     print 'writing new sitemap - please adjust if needed'
  38     sitemap.write_map()
  39 sitemap.graph()
  40
  41 sitemap.process()
  42
  43 t1 = time.time()
  44 sitemap.publish()
  45 t2 = time.time()
  46 print "Publish  [%5.2f s]" % (round(t2-t1,2))
  47 print "Total    [%5.2f s]" % (round(t2-ts,2))