This commit is contained in:
parent
33b43ae146
commit
aa14e7ddc7
37
pkgs/clan-cli/bin/gen-openapi
Executable file
37
pkgs/clan-cli/bin/gen-openapi
Executable file
@ -0,0 +1,37 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
import argparse
|
||||||
|
import json
|
||||||
|
import sys
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from uvicorn.importer import import_from_string
|
||||||
|
|
||||||
|
|
||||||
|
def main() -> None:
|
||||||
|
parser = argparse.ArgumentParser(prog="gen-openapi")
|
||||||
|
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)
|
||||||
|
parser.add_argument(
|
||||||
|
"--out", help="Output file ending in .json", default="openapi.json"
|
||||||
|
)
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
if args.app_dir is not None:
|
||||||
|
print(f"adding {args.app_dir} to sys.path")
|
||||||
|
sys.path.insert(0, args.app_dir)
|
||||||
|
|
||||||
|
print(f"importing app from {args.app}")
|
||||||
|
app = import_from_string(args.app)
|
||||||
|
openapi = app.openapi()
|
||||||
|
version = openapi.get("openapi", "unknown version")
|
||||||
|
|
||||||
|
print(f"writing openapi spec v{version}")
|
||||||
|
out = Path(args.out)
|
||||||
|
out.parent.mkdir(parents=True, exist_ok=True)
|
||||||
|
out.write_text(json.dumps(openapi, indent=2))
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
@ -1,8 +1,19 @@
|
|||||||
from fastapi import FastAPI
|
from fastapi import FastAPI
|
||||||
|
from fastapi.routing import APIRoute
|
||||||
|
|
||||||
from .routers import health, machines, root
|
from .routers import health, machines, root
|
||||||
|
|
||||||
app = FastAPI()
|
|
||||||
app.include_router(health.router)
|
def setup_app() -> FastAPI:
|
||||||
app.include_router(machines.router)
|
app = FastAPI()
|
||||||
app.include_router(root.router)
|
app.include_router(health.router)
|
||||||
|
app.include_router(machines.router)
|
||||||
|
app.include_router(root.router)
|
||||||
|
|
||||||
|
for route in app.routes:
|
||||||
|
if isinstance(route, APIRoute):
|
||||||
|
route.operation_id = route.name # in this case, 'read_items'
|
||||||
|
return app
|
||||||
|
|
||||||
|
|
||||||
|
app = setup_app()
|
||||||
|
@ -3,6 +3,6 @@ from fastapi import APIRouter
|
|||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
|
|
||||||
|
|
||||||
@router.get("/health")
|
@router.get("/health", include_in_schema=False)
|
||||||
async def health() -> str:
|
async def health() -> str:
|
||||||
return "OK"
|
return "OK"
|
||||||
|
@ -73,6 +73,13 @@ python3.pkgs.buildPythonPackage {
|
|||||||
NIX_STATE_DIR=$TMPDIR/nix ${checkPython}/bin/python -m pytest -s ./tests
|
NIX_STATE_DIR=$TMPDIR/nix ${checkPython}/bin/python -m pytest -s ./tests
|
||||||
touch $out
|
touch $out
|
||||||
'';
|
'';
|
||||||
|
passthru.clan-openapi = runCommand "clan-openapi" { } ''
|
||||||
|
cp -r ${source} ./src
|
||||||
|
chmod +w -R ./src
|
||||||
|
cd ./src
|
||||||
|
${checkPython}/bin/python ./bin/gen-openapi --out $out/openapi.json --app-dir . clan_cli.webui.app:app
|
||||||
|
touch $out
|
||||||
|
'';
|
||||||
|
|
||||||
passthru.devDependencies = [
|
passthru.devDependencies = [
|
||||||
setuptools
|
setuptools
|
||||||
|
@ -9,6 +9,7 @@
|
|||||||
inherit self;
|
inherit self;
|
||||||
zerotierone = self'.packages.zerotierone;
|
zerotierone = self'.packages.zerotierone;
|
||||||
};
|
};
|
||||||
|
clan-openapi = self'.packages.clan-cli.clan-openapi;
|
||||||
default = self'.packages.clan-cli;
|
default = self'.packages.clan-cli;
|
||||||
|
|
||||||
## Optional dependencies for clan cli, we re-expose them here to make sure they all build.
|
## Optional dependencies for clan cli, we re-expose them here to make sure they all build.
|
||||||
|
Loading…
Reference in New Issue
Block a user