iframe-proxy | Sunbelt Computer Software

30 lines (21 loc) · 769 Bytes

from typing import Optional

from fastapi import FastAPI, status

from fastapi.responses import JSONResponse

from google.cloud.sqlcommenter.fastapi import (

SQLCommenterMiddleware, get_fastapi_info,

)

from starlette.exceptions import HTTPException as StarletteHTTPException

app = FastAPI(title="SQLCommenter")

app.add_middleware(SQLCommenterMiddleware)

@app.get("/fastapi-info")

def fastapi_info():

return get_fastapi_info()

@app.get("/items/{item_id}")

def read_item(item_id: int, q: Optional[str] = None):

return get_fastapi_info()

@app.exception_handler(StarletteHTTPException)

async def custom_http_exception_handler(request, exc):

return JSONResponse(

status_code=status.HTTP_404_NOT_FOUND,

content=get_fastapi_info(),

)

Sunbelt Computer Software