57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
from typing import Any, Optional
|
|
|
|
from fastapi.encoders import jsonable_encoder
|
|
from fastapi.responses import JSONResponse
|
|
|
|
|
|
class Success(JSONResponse):
|
|
def __init__(
|
|
self,
|
|
code: int = 200,
|
|
msg: Optional[str] = "OK",
|
|
data: Optional[Any] = None,
|
|
**kwargs,
|
|
):
|
|
content = {"code": code, "msg": msg, "data": data}
|
|
content.update(kwargs)
|
|
encoded_content = jsonable_encoder(content)
|
|
super().__init__(content=encoded_content, status_code=code)
|
|
|
|
|
|
class Fail(JSONResponse):
|
|
def __init__(
|
|
self,
|
|
code: int = 400,
|
|
msg: Optional[str] = None,
|
|
data: Optional[Any] = None,
|
|
**kwargs,
|
|
):
|
|
content = {"code": code, "msg": msg, "data": data}
|
|
content.update(kwargs)
|
|
encoded_content = jsonable_encoder(content)
|
|
super().__init__(content=encoded_content, status_code=code)
|
|
|
|
|
|
class SuccessExtra(JSONResponse):
|
|
def __init__(
|
|
self,
|
|
code: int = 200,
|
|
msg: Optional[str] = None,
|
|
data: Optional[Any] = None,
|
|
total: int = 0,
|
|
page: int = 1,
|
|
page_size: int = 20,
|
|
**kwargs,
|
|
):
|
|
content = {
|
|
"code": code,
|
|
"msg": msg,
|
|
"data": data,
|
|
"total": total,
|
|
"page": page,
|
|
"page_size": page_size,
|
|
}
|
|
content.update(kwargs)
|
|
encoded_content = jsonable_encoder(content)
|
|
super().__init__(content=encoded_content, status_code=code)
|