topaz tools 1.0 (I think)

2010-01-17 12:10:35 +00:00
parent 1fc40376cf
commit 0a437510f6
9 changed files with 2931 additions and 0 deletions
--- a/Topaz_Tools/lib/genhtml.py
+++ b/Topaz_Tools/lib/genhtml.py
@@ -0,0 +1,125 @@
+#! /usr/bin/python
+# vim:ts=4:sw=4:softtabstop=4:smarttab:expandtab
+
+import os, sys, getopt
+
+# local routines
+import convert2xml
+import flatxml2html
+import decode_meta
+import stylexml2css
+
+
+def usage():
+    print 'Usage: '
+    print ' '
+    print '   genhtml.py unencryptedBookDir'
+    print '  '
+
+
+
+def main(argv):
+    bookDir = ''
+
+    if len(argv) == 0:
+        argv = sys.argv
+    else :
+        argv = argv.split()
+
+    try:
+        opts, args = getopt.getopt(argv[1:], "h:")
+
+    except getopt.GetoptError, err:
+        print str(err)
+        usage()
+        sys.exit(2)
+    
+    if len(opts) == 0 and len(args) == 0 :
+        usage()
+        sys.exit(2) 
+       
+    for o, a in opts:
+        if o =="-h":
+            usage()
+            sys.exit(0)
+
+    bookDir = args[0]
+
+    if not os.path.exists(bookDir) :
+        print "Can not find directory with unencrypted book"
+        sys.exit(-1)
+
+    dictFile = os.path.join(bookDir,'dict0000.dat')
+
+    if not os.path.exists(dictFile) :
+        print "Can not find dict0000.dat file"
+        sys.exit(-1)
+
+    pageDir = os.path.join(bookDir,'page')
+    if not os.path.exists(pageDir) :
+        print "Can not find page directory in unencrypted book"
+        sys.exit(-1)
+
+    imgDir = os.path.join(bookDir,'img')
+    if not os.path.exists(imgDir) :
+        print "Can not find image directory in unencrypted book"
+        sys.exit(-1)
+
+    otherFile = os.path.join(bookDir,'other0000.dat')
+    if not os.path.exists(otherFile) :
+        print "Can not find other0000.dat in unencrypted book"
+        sys.exit(-1)
+
+    metaFile = os.path.join(bookDir,'metadata0000.dat')
+    if not os.path.exists(metaFile) :
+        print "Can not find metadata0000.dat in unencrypted book"
+        sys.exit(-1)
+
+
+    htmlFileName = "book.html"
+    htmlstr = '<html>\n'
+
+    filenames = os.listdir(pageDir)
+    filenames = sorted(filenames)
+
+    print 'Processing ... '
+
+    htmlstr += '<head>\n'
+
+    print '     ', 'metadata0000.dat'
+    fname = os.path.join(bookDir,'metadata0000.dat')
+    xname = os.path.join(bookDir, 'metadata.txt')
+    metastr = decode_meta.getMetaData(fname)
+    file(xname, 'wb').write(metastr)
+    meta_array = decode_meta.getMetaArray(fname)
+    htmlstr += '<meta name="Author" content="' + meta_array['Authors'] + '" />\n'
+    htmlstr += '<meta name="Title" content="' + meta_array['Title'] + '" />\n'
+
+    print '     ', 'other0000.dat'
+    fname = os.path.join(bookDir,'other0000.dat')
+    xname = os.path.join(bookDir, 'style.css')
+    xmlstr = convert2xml.main('convert2xml.py --flat-xml ' + dictFile + ' ' + fname)
+    cssstr = '<style>\n'
+    cssstr += stylexml2css.convert2CSS(xmlstr)
+    cssstr += '</style>\n'
+    file(xname, 'wb').write(cssstr)
+    htmlstr += cssstr
+    htmlstr += '</head>\n<body>\n'
+
+    for filename in filenames:
+        print '     ', filename
+        fname = os.path.join(pageDir,filename)
+        flat_xml = convert2xml.main('convert2xml.py --flat-xml ' + dictFile + ' ' + fname) 
+        htmlstr += flatxml2html.convert2HTML(flat_xml, fname)
+
+    htmlstr += '</body>\n</html>\n'
+
+    file(os.path.join(bookDir, htmlFileName), 'wb').write(htmlstr)
+    print 'Processing Complete'
+
+    return 0
+
+if __name__ == '__main__':
+    sys.exit(main(''))
+
+