153 lines
5.0 KiB
Python
Executable File
153 lines
5.0 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""Extract Html files from a Microsoft Compiled Html Help file (.chm)
|
|
Needs at least python 2.2 for HTMLParser (chmlib needs 2.2 too)"""
|
|
|
|
import sys
|
|
import os
|
|
import rclexecm
|
|
try:
|
|
from chm import chm,chmlib
|
|
except:
|
|
print "RECFILTERROR HELPERNOTFOUND python:chm"
|
|
sys.exit(1);
|
|
|
|
try:
|
|
from HTMLParser import HTMLParser
|
|
except:
|
|
print "RECFILTERROR HELPERNOTFOUND python:HTMLParser"
|
|
sys.exit(1);
|
|
|
|
class ChmTopicsParser(HTMLParser):
|
|
"""Parse the chm's Topic file which is basically
|
|
a listing of internal nodes (html files mostly). Build a list of
|
|
all nodes (self.contents), which will then be used to walk and index
|
|
the chm.
|
|
|
|
Most nodes in the Topic file look like the following:
|
|
<LI> <OBJECT type="text/sitemap">
|
|
<param name="Name" value="Global Module Index">
|
|
<param name="Local" value="modindex.html">
|
|
</OBJECT>
|
|
|
|
Maybe we should filter out non "text/sitemap" Objects, and maybe there are
|
|
things of interest whose name is not Local, but for now, we just take
|
|
all values for parameters named "Local" (with some filtering/massaging),
|
|
until proven wrong
|
|
"""
|
|
def __init__(self):
|
|
HTMLParser.__init__(self)
|
|
self.contents = []
|
|
|
|
def handle_starttag(self, tag, attrs):
|
|
#print >> sys.stderr, "Encountered the beginning of a %s tag" % tag
|
|
# If this is a param tag with name Local, we're interested in
|
|
# the value which lists a file ref. Discard those with #
|
|
# in them (references inside files)
|
|
# Sometimes it seems that refs are like Vendor:filename::path,
|
|
# we only keep the path, and only if the file matches
|
|
|
|
if tag != 'param':
|
|
return
|
|
|
|
name = ''
|
|
value = ''
|
|
for (nm,val) in attrs:
|
|
if nm == 'name':
|
|
name = val
|
|
if nm == 'value':
|
|
value = val
|
|
|
|
if name != 'Local' or value == '':
|
|
return
|
|
|
|
localpath = ""
|
|
ll = value.split(":")
|
|
if len(ll) == 1:
|
|
localpath = value
|
|
elif len(ll) == 4 and ll[-1] and ll[-3]:
|
|
#print >>sys.stderr, "File: %s" % ll[-3]
|
|
if ll[-3] == self.fname:
|
|
localpath = ll[-1]
|
|
else:
|
|
#print >> sys.stderr, "SKIPPING %s" % ll[-3]
|
|
pass
|
|
|
|
if len(localpath) != 0 and localpath.find("#") == -1:
|
|
if localpath[0] != '/':
|
|
localpath = "/" + localpath
|
|
self.contents.append(localpath)
|
|
|
|
def reset(self):
|
|
self.contents = []
|
|
self.fname = ""
|
|
HTMLParser.reset(self)
|
|
|
|
def setname(self, name):
|
|
self.fname = name
|
|
|
|
class rclCHM:
|
|
"""RclExecM slave worker for extracting all files from an Msoft chm
|
|
file. We first extract the list of internal nodes, and them return them
|
|
one by one. The ipath is the node path"""
|
|
|
|
def __init__(self, em):
|
|
self.chm = chm.CHMFile()
|
|
self.tp = ChmTopicsParser()
|
|
self.currentindex = 0
|
|
self.em = em
|
|
|
|
def extractone(self, path):
|
|
"""Extract one path-named internal file from the chm file"""
|
|
|
|
#self.em.rclog("extractone: [%s]"%(path))
|
|
iseof = rclexecm.RclExecM.noteof
|
|
if self.currentindex >= len(self.tp.contents) -1:
|
|
iseof = rclexecm.RclExecM.eofnext
|
|
|
|
res, ui = self.chm.ResolveObject(path)
|
|
#self.em.rclog("extract: ResolveO: %d [%s]" % (res, ui))
|
|
if res != chmlib.CHM_RESOLVE_SUCCESS:
|
|
return (False, "", path, iseof)
|
|
# RetrieveObject() returns len,value
|
|
res, doc = self.chm.RetrieveObject(ui)
|
|
#self.em.rclog("extract: RetrieveObject: %d [%s]" % (res, doc))
|
|
if res > 0:
|
|
self.em.setmimetype("text/html")
|
|
return (True, doc, path, iseof)
|
|
return (False, "", path, iseof)
|
|
|
|
def openfile(self, params):
|
|
"""Open the chm file and build the contents list by extracting and
|
|
parsing the Topics object"""
|
|
|
|
self.currentindex = 0
|
|
self.tp.reset()
|
|
filename = params["filename:"]
|
|
self.chm.LoadCHM(filename)
|
|
self.chm.GetArchiveInfo()
|
|
self.topics = self.chm.GetTopicsTree()
|
|
if self.topics == None:
|
|
return False
|
|
#self.em.rclog(self.topics)
|
|
# Parse Topics file and extract list of internal nodes
|
|
self.tp.setname(os.path.basename(filename))
|
|
self.tp.feed(self.topics)
|
|
self.tp.close()
|
|
#self.em.rclog("Contents size %d" % len(self.tp.contents))
|
|
return True
|
|
|
|
def getipath(self, params):
|
|
return self.extractone(params["ipath:"])
|
|
|
|
def getnext(self, params):
|
|
if self.currentindex >= len(self.tp.contents):
|
|
return (False, "", "", rclexecm.RclExecM.eofnow)
|
|
else:
|
|
ret= self.extractone(self.tp.contents[self.currentindex])
|
|
self.currentindex += 1
|
|
return ret
|
|
|
|
proto = rclexecm.RclExecM()
|
|
extract = rclCHM(proto)
|
|
rclexecm.main(proto, extract)
|