Spaces:
Running
Running
File size: 1,154 Bytes
50553ea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
from functools import wraps
from typing import Generic, Optional, TypeVar
from fastapi import HTTPException
from pydantic import BaseModel
from starlette.responses import JSONResponse
from trauma.core.config import settings
T = TypeVar('T')
class ErrorTraumaResponse(BaseModel):
message: str
class TraumaResponseWrapper(BaseModel, Generic[T]):
data: Optional[T] = None
successful: bool = True
error: Optional[ErrorTraumaResponse] = None
def response(self, status_code: int):
return JSONResponse(
status_code=status_code,
content={
"data": self.data,
"successful": self.successful,
"error": self.error.dict() if self.error else None
}
)
def exception_wrapper(http_error: int, error_message: str):
def decorator(func):
@wraps(func)
async def wrapper(*args, **kwargs):
try:
return await func(*args, **kwargs)
except Exception as e:
raise HTTPException(status_code=http_error, detail=error_message) from e
return wrapper
return decorator
|