mirror of
https://github.com/cemu-project/idapython.git
synced 2024-11-24 10:09:20 +01:00
117 lines
3.0 KiB
Python
117 lines
3.0 KiB
Python
import os
|
|
import sys
|
|
import shutil
|
|
from glob import glob
|
|
|
|
# --------------------------------------------------------------------------
|
|
DOC_DIR = 'hr-html'
|
|
PYWRAPS_FN = 'idaapi.py'
|
|
|
|
# --------------------------------------------------------------------------
|
|
def add_footer(lines):
|
|
S1 = 'Generated by Epydoc'
|
|
S2 = '</table>'
|
|
p = lines.find(S1)
|
|
|
|
if p == -1:
|
|
return None
|
|
|
|
p = lines.find(S2, p)
|
|
if p == -1:
|
|
return None
|
|
|
|
p += len(S2)
|
|
|
|
return lines[0:p] + '\n<!--#include virtual="/footer.shtml" -->' + lines[p:]
|
|
|
|
# --------------------------------------------------------------------------
|
|
def define_idaapi_resolver():
|
|
"""
|
|
Whenever a module named \"idaapi_<something>\" is
|
|
spotted, turn it into \"idaapi\".
|
|
"""
|
|
import epydoc.apidoc
|
|
dn = epydoc.apidoc.DottedName.__init__
|
|
def resolver(piece):
|
|
if piece is not None and isinstance(piece, basestring) and piece.startswith("idaapi_"):
|
|
return "idaapi"
|
|
else:
|
|
return piece
|
|
def wrapper(self, *pieces, **options):
|
|
return dn(self, *map(resolver, pieces), **options);
|
|
epydoc.apidoc.DottedName.__init__ = wrapper
|
|
|
|
# --------------------------------------------------------------------------
|
|
def gen_docs():
|
|
import epydoc.cli
|
|
import swigdocs
|
|
|
|
define_idaapi_resolver()
|
|
|
|
swigdocs.gen_docs(outfn = 'pywraps.py')
|
|
# append obj/x86_win_vc_32/idaapi.py to it
|
|
# os.system(r'copy /b idaapi.py+..\obj\x86_win_vc_32\idaapi.py idaapi.py')
|
|
|
|
# delete all output files
|
|
for fn in glob('hr-html/*'):
|
|
os.unlink(fn)
|
|
|
|
epydoc.cli.optparse.sys.argv = [ 'epydoc',
|
|
'--config', '../hrdoc.cfg',
|
|
'--simple-term'
|
|
]
|
|
|
|
# Generate the documentation
|
|
epydoc.cli.cli()
|
|
|
|
# --------------------------------------------------------------------------
|
|
def patch_docs():
|
|
shutil.copy('../../hrdoc.css', 'epydoc.css')
|
|
os.system('chmod +w epydoc.css')
|
|
|
|
for fn in glob('*.html'):
|
|
f = open(fn, 'r')
|
|
lines = f.read()
|
|
f.close()
|
|
|
|
r = add_footer(lines)
|
|
if not r:
|
|
print "-",
|
|
continue
|
|
|
|
f = open(fn, 'w')
|
|
f.write(r)
|
|
f.close()
|
|
print "+",
|
|
|
|
print "\nDocumentation patched!"
|
|
|
|
# --------------------------------------------------------------------------
|
|
def main():
|
|
# Save old directory and adjust import path
|
|
curdir = os.getcwd() + os.sep
|
|
sys.path.append(curdir + 'python')
|
|
sys.path.append(curdir + 'tools')
|
|
sys.path.append(curdir + 'docs')
|
|
|
|
old_dir = os.getcwd()
|
|
|
|
try:
|
|
print "Generating documentation....."
|
|
|
|
os.chdir('docs')
|
|
gen_docs()
|
|
|
|
os.chdir(DOC_DIR)
|
|
patch_docs()
|
|
|
|
print "Documentation generated!"
|
|
|
|
finally:
|
|
os.chdir(old_dir)
|
|
|
|
# --------------------------------------------------------------------------
|
|
if __name__ == '__main__':
|
|
main()
|
|
Exit(0)
|