iwla/tools/extract_doc.py
2014-12-19 11:35:00 +01:00

30 lines
661 B
Python
Executable File

#!/usr/bin/env python
import sys
filename = sys.argv[1]
if filename.endswith('__init__.py'):
sys.exit(0)
package_name = filename.replace('/', '.').replace('.py', '')
sys.stdout.write('**%s**' % (package_name))
sys.stdout.write('\n\n')
# sys.stdout.write('-' * len(package_name))
# sys.stdout.write('\n\n')
sys.stderr.write('\tExtract doc from %s\n' % (filename))
with open(filename) as infile:
copy = False
for line in infile:
if line.strip() in ['"""', "'''"]:
if not copy:
copy = True
else:
break
elif copy:
sys.stdout.write(line)
sys.stdout.write('\n\n')