File size: 1,357 Bytes
0914710 fcb8c81 0914710 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
# https://www.doctave.com/blog/python-export-fastapi-openapi-spec#step-2-create-an-export-script
import argparse
import json
import logging
import sys
import yaml
from uvicorn.importer import import_from_string
from samgis_lisa_on_cuda import PROJECT_ROOT_FOLDER
parser = argparse.ArgumentParser(prog="extract-openapi-fastapi.py")
parser.add_argument("app", help='App import string. Eg. "main:app"', default="main:app")
parser.add_argument("--app-dir", help="Directory containing the app", default=None)
if __name__ == "__main__":
# python scripts/extract-openapi.py fastapi_wrapper:app --app-dir wrappers --out docs/specs/openapi_new.yaml
args = parser.parse_args()
if args.app_dir is not None:
logging.info(f"adding {args.app_dir} to sys.path")
sys.path.insert(0, args.app_dir)
logging.info(f"importing app from {args.app}")
app = import_from_string(args.app)
openapi = app.openapi()
version = openapi.get("openapi", "unknown version")
logging.info(f"writing openapi spec v{version}...")
output_dir_path = PROJECT_ROOT_FOLDER / "docs" / "specs"
with open(output_dir_path / "output.json", "w") as f:
json.dump(openapi, f)
with open(output_dir_path / "output.yaml", "w") as f:
yaml.dump(openapi, f, sort_keys=False)
logging.info(f"spec written to {args.out} #")
|