33 lines
930 B
Python
33 lines
930 B
Python
# 依赖安装：pip install orjson
|
||
from fastapi.responses import ORJSONResponse as Response
|
||
from fastapi import status as http_status
|
||
from utils import status as http
|
||
|
||
|
||
class SuccessResponse(Response):
|
||
"""
|
||
成功响应
|
||
"""
|
||
def __init__(self, data=None, msg="success", code=http.HTTP_SUCCESS, status=http_status.HTTP_200_OK, **kwargs):
|
||
self.data = {
|
||
"code": code,
|
||
"message": msg,
|
||
"data": data
|
||
}
|
||
self.data.update(kwargs)
|
||
super().__init__(content=self.data, status_code=status)
|
||
|
||
|
||
class ErrorResponse(Response):
|
||
"""
|
||
失败响应
|
||
"""
|
||
def __init__(self, msg=None, code=http.HTTP_ERROR, status=http_status.HTTP_200_OK, **kwargs):
|
||
self.data = {
|
||
"code": code,
|
||
"message": msg,
|
||
"data": []
|
||
}
|
||
self.data.update(kwargs)
|
||
super().__init__(content=self.data, status_code=status)
|