mp_driver_server/main.py

24 lines
766 B
Python
Raw Normal View History

2024-06-26 16:44:12 +03:00
from importlib import reload
from fastapi import FastAPI, Request
from fastapi.responses import JSONResponse
from fastapi.exceptions import RequestValidationError
from core.transport.rest import router
from core.transport.graphql import graphql_router
from core.errors.base import MPDriverException
app = FastAPI()
app.include_router(router)
app.include_router(graphql_router, prefix="/graphql")
@app.exception_handler(MPDriverException)
async def mpdriver_exception(request: Request, exc: MPDriverException):
return exc.response()
@app.exception_handler(RequestValidationError)
async def validation_exception_handler(request: Request, exc: RequestValidationError):
error = exc.errors()[0]
raise MPDriverException(422, error['type'], error['msg'])