15 二次封裝Response類

二次封裝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"}