import functools from fastapi import FastAPI from fastapi.openapi.utils import get_openapi def custom_openapi(): if oapi.openapi_schema: return oapi.openapi_schema openapi_schema = get_openapi( title="Custom title", version="2.5.0", description="This is a very custom OpenAPI schema", routes=oapi.routes, ) openapi_schema["info"]["x-logo"] = { "url": "https://fastapi.tiangolo.com/img/logo-margin/logo-teal.png" } oapi.openapi_schema = openapi_schema return oapi.openapi_schema oapi = FastAPI() oapi.openapi = custom_openapi def apiroute(route=None, **kw): routing = kw.copy() def apidecorator(f): if route: if isinstance(route, list): routes = route else: routes = [route] routing["routes"] = routes @functools.wraps(f) def response_wrap(*args, **kw): response = f(*args, **kw) return response oapi.add_api_route(routes[0], f, include_in_schema=True) response_wrap.routing = routing response_wrap.original_func = f return response_wrap return apidecorator #