drf自己封装response

from rest_framework.response import Response

# 第一种
class APIResponse(Response):
    def __init__(self, code=100, msg='成功', status=None, headers=None, **kwargs):
        res_data = {
            'code': code,
            'msg': msg,
        }
        if kwargs:
            res_data.update(kwargs)
        super().__init__(data=res_data, status=status, headers=headers)     

'''
        return APIResponse()  
    	# 成功，没有数据  
        {
            "code": 100,
            "msg": "成功"
        }
        
        return APIResponse(result=[])  
        {
            "code": 100,
            "msg": "成功"
             result:[]
        }
        
        return APIResponse(token='xxxxx')  
        {
            "code": 100,
            "msg": "成功"
            token:'xxxxx'
        }
        
         return APIResponse(headers={name:amgl})  
        # 响应头中会多name=amgl
        {
            "code": 100,
            "msg": "成功"
        }
'''
# 第二种
# class MyResponse():
#     def __init__(self):
#         self.status=100
#         self.msg='成功'
#     @property
#     def get_dict(self):
#         return self.__dict__

# if __name__ == '__main__':
#     res=MyResponse()
#     res.status=101
#     res.msg='查询失败'
#     # res.data={'name':'AAA'}
#     print(res.get_dict) # {'status': 101, 'msg': '查询失败'}

__EOF__

本文作者：amgulen
本文链接：https://www.cnblogs.com/amgulen/p/16849923.html
关于博主：评论和私信会在第一时间回复。或者直接私信我。
版权声明：本博客所有文章除特别声明外，均采用 BY-NC-SA 许可协议。转载请注明出处！
声援博主：如果您觉得文章对您有帮助，可以点击文章右下角【推荐】一下。您的鼓励是博主的最大动力！