47 lines
1.6 KiB
Python
Executable File
47 lines
1.6 KiB
Python
Executable File
#
|
|
# Copyright (c) 2022 Airbyte, Inc., all rights reserved.
|
|
#
|
|
|
|
import sys
|
|
from os import path
|
|
from typing import Any, Dict
|
|
|
|
from sphinx.cmd.quickstart import QuickstartRenderer
|
|
from sphinx.ext.apidoc import get_parser, main, recurse_tree, write_file
|
|
from sphinx.locale import __
|
|
from sphinx.util import ensuredir
|
|
|
|
|
|
def write_master_file(templatedir: str, master_name: str, values: Dict, opts: Any):
|
|
template = QuickstartRenderer(templatedir=templatedir)
|
|
opts.destdir = opts.destdir[: opts.destdir.rfind("/")]
|
|
write_file(master_name, template.render(f"{templatedir}/master_doc.rst_t", values), opts)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = get_parser()
|
|
parser.add_argument("--master", metavar="MASTER", default="index", help=__("master document name"))
|
|
args = parser.parse_args(sys.argv[1:])
|
|
|
|
rootpath = path.abspath(args.module_path)
|
|
|
|
# normalize opts
|
|
if args.header is None:
|
|
args.header = rootpath.split(path.sep)[-1]
|
|
if args.suffix.startswith("."):
|
|
args.suffix = args.suffix[1:]
|
|
if not path.isdir(rootpath):
|
|
print(__(f"{rootpath} is not a directory."), file=sys.stderr)
|
|
sys.exit(1)
|
|
if not args.dryrun:
|
|
ensuredir(args.destdir)
|
|
excludes = [path.abspath(exclude) for exclude in args.exclude_pattern]
|
|
modules = recurse_tree(rootpath, excludes, args, args.templatedir)
|
|
|
|
template_values = {
|
|
"top_modules": [{"path": f"api/{module}", "caption": module.split(".")[1].title()} for module in modules if module.count(".") == 1],
|
|
"maxdepth": args.maxdepth,
|
|
}
|
|
write_master_file(templatedir=args.templatedir, master_name=args.master, values=template_values, opts=args)
|
|
main()
|