mirror of https://github.com/tqdm/tqdm.git
73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
"""
|
|
Auto-generate README.rst from .meta/.readme.rst and docstrings.
|
|
"""
|
|
import sys
|
|
from pathlib import Path
|
|
from textwrap import dedent
|
|
|
|
sys.path.insert(0, str(Path(__file__).parent.parent))
|
|
import tqdm # NOQA
|
|
import tqdm.cli # NOQA
|
|
|
|
HEAD_ARGS = """
|
|
Parameters
|
|
----------
|
|
"""
|
|
HEAD_RETS = """
|
|
Returns
|
|
-------
|
|
"""
|
|
HEAD_CLI = """
|
|
Extra CLI Options
|
|
-----------------
|
|
name : type, optional
|
|
TODO: find out why this is needed.
|
|
"""
|
|
|
|
|
|
def doc2rst(doc, arglist=True, raw=False):
|
|
"""
|
|
arglist : bool, whether to create argument lists
|
|
raw : bool, ignores arglist and indents by 2 spaces
|
|
"""
|
|
doc = doc.replace('`', '``')
|
|
if raw:
|
|
doc = doc.replace('\n ', '\n ')
|
|
else:
|
|
doc = dedent(doc)
|
|
if arglist:
|
|
doc = '\n'.join(i if not i or i[0] == ' ' else '* ' + i + ' '
|
|
for i in doc.split('\n'))
|
|
return doc
|
|
|
|
|
|
src_dir = Path(__file__).parent.resolve()
|
|
README_rst = (src_dir / '.readme.rst').read_text("utf-8")
|
|
class_doc, init_doc = tqdm.tqdm.__doc__.split('\n\n', 1)
|
|
DOC_tqdm = doc2rst(class_doc + '\n', False).replace('\n', '\n ')
|
|
DOC_tqdm_init = doc2rst('\n' + init_doc)
|
|
DOC_tqdm_init_args = DOC_tqdm_init.partition(doc2rst(HEAD_ARGS))[-1].replace(
|
|
'\n ', '\n ').replace('\n ', '\n ')
|
|
DOC_tqdm_init_args, _, DOC_tqdm_init_rets = DOC_tqdm_init_args.partition(doc2rst(HEAD_RETS))
|
|
DOC_cli = doc2rst(tqdm.cli.CLI_EXTRA_DOC).partition(doc2rst(HEAD_CLI))[-1]
|
|
DOC_tqdm_tqdm = {}
|
|
for i in dir(tqdm.tqdm):
|
|
doc = getattr(tqdm.tqdm, i).__doc__
|
|
if doc:
|
|
DOC_tqdm_tqdm[i] = doc2rst(doc, raw=True)
|
|
|
|
# special cases
|
|
DOC_tqdm_init_args = DOC_tqdm_init_args.replace(' *,', ' ``*``,')
|
|
DOC_tqdm_init_args = DOC_tqdm_init_args.partition('* gui : bool, optional')[0]
|
|
|
|
README_rst = (
|
|
README_rst.replace('{DOC_tqdm}', DOC_tqdm)
|
|
.replace('{DOC_tqdm.tqdm.__init__.Parameters}', DOC_tqdm_init_args)
|
|
.replace('{DOC_tqdm.cli.CLI_EXTRA_DOC}', DOC_cli)
|
|
.replace('{DOC_tqdm.tqdm.__init__.Returns}', DOC_tqdm_init_rets))
|
|
for k, v in DOC_tqdm_tqdm.items():
|
|
README_rst = README_rst.replace('{DOC_tqdm.tqdm.%s}' % k, v)
|
|
|
|
if __name__ == "__main__":
|
|
(src_dir.parent / 'README.rst').write_text(README_rst, encoding='utf-8')
|