import htmllib, formatter, string, os, pprint, types, sys, glob
api_name = sys.argv[1] #'wxpyapi'
api_path = sys.argv[2] #'./docs/api/'
header_hhx = '''
'''
entry_hhx = '''
'''
file_hhp = '''[OPTIONS]
Compatibility=1.1
Full-text search=Yes
Contents file=%(n)s.hhc
Compiled file=%(n)s.chm
Default Window=wxHelp
Default topic=index.html
Index file=%(n)s.hhk
Title=wxPython API Documentation
[WINDOWS]
wxHelp=,"%(n)s.hhc","%(n)s.hhk","index.html",,,,,,0x2420,,0x380e,,,,,0,,,
[FILES]
'''
def read_segment(fn, start, end):
data = open(fn).read()
begin = data.find(start)+len(start)
return data[begin:data.find(end, begin)]
#-------------------------------------------------------------------------------
def build_project():
hhp = file_hhp % {'n' : api_name}
def doglob(mask):
st = ''
for name in glob.glob(os.path.join(api_path, mask)):
name = os.path.split(name)[-1]
st += name + '\n'
return st
hhp += doglob("*.html")
hhp += doglob("*.css")
open(os.path.join(api_path, api_name+'.hhp'), 'w').write(hhp)
#-------------------------------------------------------------------------------
def build_contents():
def traverse(l, r):
for i in l:
if type(i) is types.ListType:
r.append(''+os.linesep)
traverse(i, r)
r.append('
'+os.linesep)
elif type(i) is types.TupleType:
r.append(entry_hhx%i)
else:
raise Exception, 'Unhandled type: %s'%type(i)
data = read_segment(os.path.join(api_path, 'trees.html'),
'',
'')
p = APIContentsParser(formatter.NullFormatter(formatter.NullWriter()))
p.feed(data)
class_hierarchy = []
traverse(p.current, class_hierarchy)
data = read_segment(os.path.join(api_path, 'wx-module.html'),
'',
'')
p = APIContentsParser(formatter.NullFormatter(formatter.NullWriter()))
p.feed(data)
submodules = []
traverse(p.current, submodules)
hhc = header_hhx+\
''+os.linesep+entry_hhx%('wx-module.html', 'Submodules')+\
''.join(submodules)+'
'+os.linesep+\
''+os.linesep+entry_hhx%('trees.html', 'Class Hierarchy')+\
''.join(class_hierarchy)+'
'+os.linesep
open(os.path.join(api_path, api_name+'.hhc'), 'w').write(hhc)
class APIContentsParser(htmllib.HTMLParser):
def __init__(self, formatter, verbose=0):
htmllib.HTMLParser.__init__(self, formatter, verbose)
self.contents = []
self.history = []
self.current = self.contents
self.cur_href = None
self.li_a_cnt = 0
def start_li(self, attrs):
self.li_a_cnt = 0
def start_a(self, attrs):
self.li_a_cnt += 1
if self.li_a_cnt == 1:
if attrs[0][0] == 'href':
self.cur_href = attrs[0][1]
def end_a(self):
self.cur_href = None
def start_code(self, attrs):
self.save_bgn()
def end_code(self):
text = string.strip(self.save_end())
if self.cur_href and text:
self.current.append( (self.cur_href, text) )
def start_ul(self, attrs):
self.history.append(self.current)
self.current = []
def end_ul(self):
c = self.history.pop()
c.append(self.current)
self.current = c
#-------------------------------------------------------------------------------
def build_keywords():
data = read_segment(os.path.join(api_path, 'indices.html'),
'',
'')
p = APIIndicesParser(formatter.NullFormatter(formatter.NullWriter()))
p.feed(data)
hhk = header_hhx+ ''+os.linesep+\
''.join([entry_hhx%(u, k) for u, k in p.indices])+os.linesep+'
'
open(os.path.join(api_path, api_name+'.hhk'), 'w').write(hhk)
class APIIndicesParser(htmllib.HTMLParser):
def __init__(self, formatter, verbose=0):
htmllib.HTMLParser.__init__(self, formatter, verbose)
self.indices = []
self.cur_href = None
self.tr_a_cnt = 0
def start_tr(self, attrs):
self.tr_a_cnt = 0
def end_tr(self):
self.tr_a_cnt = 0
def start_a(self, attrs):
self.tr_a_cnt += 1
if self.tr_a_cnt == 1:
if attrs[0][0] == 'href':
self.cur_href = attrs[0][1]
def end_a(self):
self.cur_href = None
def start_code(self, attrs):
self.save_bgn()
def end_code(self):
text = string.strip(self.save_end())
if self.cur_href and text:
self.indices.append( (self.cur_href, text) )
#-------------------------------------------------------------------------------
if __name__ == '__main__':
print 'Building project...'
build_project()
print 'Building contents...'
build_contents()
print 'Building keywords...'
build_keywords()