15二次封装Response类[Python基础]

  • Z时代
  • 2024-01-10
  • 分类：综合

python

二次封装Response类

from rest_framework.response import Response
class APIResponse(Response):
    def __init__(self, data_status=0, data_msg="ok", results=None, http_status=None, headers=None, exception=False, **kwargs):
        """
        :param data_status: 状态码
        :param msg: 提示信息
        :param results: 附加信息,如序列化得到的数据
        :param headers:
        :param status: HTTP状态码
        :param kwargs: 其他信息
        """
        # data的初始状态：状态码与状态信息
        data = {
            "status": data_status,
            "msg": data_msg,
        }
        # data的响应数据体
        # results可能是False、0等数据，这些数据某些情况下也会作为合法数据返回
        if results is not None:
            data["results"] = results
        # data响应的其他内容
        # if kwargs is not None:
        #     for k, v in kwargs.items():
        #         setattr(data, k, v)
        data.update(kwargs)
        super().__init__(data=data, status=http_status, headers=headers, exception=exception)   #重写父类Response的__init__方法
# 使用：
# APIResponse() 代表就返回 {"status": 0, "msg": "ok"}
# APIResponse(result="结果") 代表返回 {"status": 0, "msg": "ok", "result": "结果"}
# APIResponse(status=1, msg="error", http_status=400, exception=True) 异常返回 {"status": 1, "msg": "error"}

以上是 15二次封装Response类[Python基础] 的全部内容， 来源链接： utcz.com/z/530751.html

回到顶部