I-school-BE/utils.py at main · ais-internship-cnu/I-school-BE · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from fastapi import Request

from typing import Optional, Any

from pydantic import BaseModel
from starlette.responses import JSONResponse


class Response(BaseModel):
    success: bool
    data: Optional[Any]
    error: Optional[str]


class UvicornException(Exception):
    def __init__(self, status_code: int, message: str):
        self.status_code = status_code
        self.message = message


async def http_exception_handler(request: Request, exception: UvicornException):
    return JSONResponse(status_code=exception.status_code,
                        content=Response(success=False, data=None, error=exception.message).model_dump())