DRF框架（八）——二次封装response类

最新推荐文章于 2024-10-24 18:46:07 发布

Believer_abby

最新推荐文章于 2024-10-24 18:46:07 发布

阅读量420

点赞数

分类专栏： Django rest framework 文章标签： django

本文链接：https://blog.youkuaiyun.com/weixin_41635393/article/details/105820361

版权

Django rest framework 专栏收录该内容

8 篇文章

订阅专栏

本文介绍如何在Django REST框架中创建自定义的APIResponse类，以统一API响应格式。通过重写Response类的初始化方法，可以灵活地设置状态码、消息和数据体。在视图中使用该自定义响应，能提高代码的可读性和维护性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、自定义APIResponse

重写父类Response的__init__方法

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):
        # data的初始状态：状态码和状态信息
        data = {
            'status': data_status,
            'msg': data_msg,
        }
        # data的响应数据体
        # results可能是false、0等数据，这些数据某些情况下也会作为合法数据返回
        if results is not None:
            data['results'] = results
        # data响应的其他内容
        # if kwags 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)
        # 重写父类的__init__方法

二、在views.py中使用APIResponse

导入

from utils.response import APIResponse

使用

return APIResponse(1, 'ok', result_list)