"""
@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)