base.py 489 B

123456789101112131415
  1. from typing import Generic, TypeVar, Optional, Any
  2. from pydantic import BaseModel
  3. T = TypeVar('T')
  4. class Response(BaseModel, Generic[T]):
  5. code: int = 200
  6. message: str = "Success"
  7. data: Optional[T] = None
  8. def success(data: T = None, message: str = "Success") -> Response[T]:
  9. return Response(code=200, message=message, data=data)
  10. def error(code: int = 500, message: str = "Error", data: Any = None) -> Response:
  11. return Response(code=code, message=message, data=data)