33 lines
871 B
Python
33 lines
871 B
Python
"""
|
||
@Remark: 自定义的Response返回内容
|
||
"""
|
||
|
||
from rest_framework import status as HttpStatus
|
||
from rest_framework.response import Response as RestResponse
|
||
|
||
from utils.ecode import ECode
|
||
|
||
|
||
# 1.返回单一数据
|
||
class Response(RestResponse):
|
||
def __init__(
|
||
self,
|
||
data=None,
|
||
code=HttpStatus.HTTP_200_OK,
|
||
message=None,
|
||
status=None,
|
||
template_name=None,
|
||
headers=None,
|
||
exception=False,
|
||
content_type="application/json",
|
||
):
|
||
# 如果未提供 message，则通过 code 从 ECode 中获取
|
||
message = message or ECode.desc(code)
|
||
data = {
|
||
"code": code,
|
||
"data": data,
|
||
"success": code == HttpStatus.HTTP_200_OK,
|
||
"message": message,
|
||
}
|
||
super().__init__(data, status, template_name, headers, exception, content_type)
|