resolve {ENV_VAR} placeholders in spec before serving
- app.py: add _resolve_env and _resolve_spec_env_vars to replace
{ENV_VAR} patterns (e.g. {AUTH_SERVER}) with os.environ values
so the frontend receives resolved URLs via /openapi.json
This commit is contained in:
@@ -32,12 +32,35 @@ Notes:
|
|||||||
- Alter FastAPI dependency injection semantics.
|
- Alter FastAPI dependency injection semantics.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
import os
|
||||||
|
import re
|
||||||
|
|
||||||
from fastapi import FastAPI
|
from fastapi import FastAPI
|
||||||
|
|
||||||
from .binder import bind_routes
|
from .binder import bind_routes
|
||||||
from .loader import load_openapi
|
from .loader import load_openapi
|
||||||
from .security import make_security_dependencies, parse_security_schemes
|
from .security import make_security_dependencies, parse_security_schemes
|
||||||
|
|
||||||
|
_env_pattern = re.compile(r"\{(\w+)\}")
|
||||||
|
|
||||||
|
|
||||||
|
def _resolve_env(value: str) -> str:
|
||||||
|
"""Replace {ENV_VAR} placeholders with values from os.environ."""
|
||||||
|
def _replace(m: re.Match) -> str:
|
||||||
|
return os.environ.get(m.group(1), m.group(0))
|
||||||
|
return _env_pattern.sub(_replace, value)
|
||||||
|
|
||||||
|
|
||||||
|
def _resolve_spec_env_vars(obj):
|
||||||
|
"""Recursively resolve {ENV_VAR} in all string values of the spec."""
|
||||||
|
if isinstance(obj, str):
|
||||||
|
return _resolve_env(obj)
|
||||||
|
if isinstance(obj, dict):
|
||||||
|
return {k: _resolve_spec_env_vars(v) for k, v in obj.items()}
|
||||||
|
if isinstance(obj, list):
|
||||||
|
return [_resolve_spec_env_vars(v) for v in obj]
|
||||||
|
return obj
|
||||||
|
|
||||||
|
|
||||||
class OpenAPIFirstApp(FastAPI):
|
class OpenAPIFirstApp(FastAPI):
|
||||||
"""
|
"""
|
||||||
@@ -107,7 +130,7 @@ class OpenAPIFirstApp(FastAPI):
|
|||||||
super().__init__(**fastapi_kwargs)
|
super().__init__(**fastapi_kwargs)
|
||||||
|
|
||||||
# Load and validate OpenAPI specification
|
# Load and validate OpenAPI specification
|
||||||
self._openapi_spec = load_openapi(openapi_path)
|
self._openapi_spec = _resolve_spec_env_vars(load_openapi(openapi_path))
|
||||||
|
|
||||||
# Parse security schemes and build per-route dependencies
|
# Parse security schemes and build per-route dependencies
|
||||||
security_schemes = parse_security_schemes(self._openapi_spec)
|
security_schemes = parse_security_schemes(self._openapi_spec)
|
||||||
|
|||||||
Reference in New Issue
Block a user