drf-过滤、排序、异常处理、自封装Response

过滤

过滤就是根据路由url?后的信息过滤出符合?后条件的数据而非全部,比如…/?name=weer就是只查name是weer的数据,其余不返回。

1、安装:pip3 install django-filter
 
2、注册:在settings.py中的app中注册'django-filter'
 
3、使用
    全局配置:还是在REST_FRAMEWORK中写:
        'DEFAULT_FILTER_BACKENDS':['django_filters.rest_framework.DjangoFilterBackend']
    然后在需要使用的视图类中写:
        filter_fields = ('name', )       # 指定可过滤字段,一般都用在查中
    局部配置也是在视图类下配置filter_backends = […],同方法嘛

排序

就是将查出来的数据按某某排序展示,比如按价格升序、按id降序

是依赖于django-filter实现的,所以前面安装过这不再书写了

也是可全局使用和局部使用

此介绍局部使用:

views.py
 
    from rest_framework.filters import OrderingFilter
    class BookView(ListAPIView):
        queryset = models.Book.objects.all()
        serializer_class = BookSerializer
        filter_backends = [OrderingFilter]
        ordering_fields = ('id', 'price') # 指定可用排序字段
 
# 使用时路由url中?后应带固定关键字比如ordering=id,支持降序就是加负号-比如/?ordering=-price就是按价格降序展示

异常处理

通过前面的经验,当我们写项目的时候出现错误,drf或Django会自动帮你处理错误并返回你它自己设置的错误信息或页面。但在实际项目开发中,我们写接口文档会根据自己公司的要求来自定义错误信息,并且一般都返回字典json格式,里面包括什么状态码、出错信息、headers啊等,而不是像Django给你出现的如下页面错误信息:

drf中的自动帮你处理错误信息的方法在rest_framework.views.exception_handler中:

def exception_handler(exc, context):
    # exc就是异常信息对象, context可以取出视图函数来
    if isinstance(exc, Http404):
        exc = exceptions.NotFound()
    elif isinstance(exc, PermissionDenied):
        exc = exceptions.PermissionDenied()
    if isinstance(exc, exceptions.APIException):
        headers = {}
        if getattr(exc, 'auth_header', None):
            headers['WWW-Authenticate'] = exc.auth_header
        if getattr(exc, 'wait', None):
            headers['Retry-After'] = '%d' % exc.wait
    
        if isinstance(exc.detail, (list, dict)):
            data = exc.detail
        else:
            data = {'detail': exc.detail}
        set_rollback()
        return Response(data, status=exc.status_code, headers=headers)
    return None
 
"""
研究源码可以看到,drf自动帮你处理了404、是否允许访问,还有APIException错误信息,
另一个返回结果None就很灵性,意味着该异常处理不全,返回None的交给了Django处理,
因此展示那种错误页面
"""
研究exception_handler()源码

因此我们需要写自己的异常处理方法,一是为了统一错误信息返回格式,二是为了记录错误日志

当然,像APIException错误这种,drf已然帮我们写好了,为此我们可以直接用

我是新建的my_exception_handler.py:

from rest_framework.views import exception_handler
from rest_framework.views import Response
from rest_framework import status
 
def my_exception_handler(exc, content):
    response = exception_handler(exc, content) # 沿用exception_handler处理方法 返回None时就是写我们自己的,没有返回None时就是对APIException处理了的
    if not response:
        # 写自己的异常处理逻辑 按需求来
        # 以下为示例
        return Response(data={'code':104,'msg':str(exc)}, status=status.HTTP_400_BAD_REQUEST)
    else:
        # 这返回的是drf内置的处理错误信息,如APIException的
        return Response(data={'code':101,'msg':response.data.get('detail')}, status=status.HTTP_403_FORBIDDEN)

然后去REST_FRAMEWORK中全局配置,没有局部配

'EXCEPTION_HANDLER':'app01.my_exception_handler.my_exception_handler'

对应着来之后只要出现异常都返回类似这种格式的了:

自封装Response对象

虽然drf提供的Response对象已经包含了很多功能,比如data, status,template_name, headers,exception,content_type这种,但其可扩展性还是不高,需要结合项目需求来,比如要加一个token字段跟在其后便没有办法,因此也是在实际开发中,我们都需要自己封装一个Response对象,不过继承了原生Response对象而已,以实现更多需求

下为一个简单示例↓

 

class APIResponse(Response):
    def __init__(self,code=200,msg='成功',data=None, status=None,headers=None,**kwargs):
        dic = {'code':code, 'msg':msg}
        dic.update(kwargs)
        super().__init__(data=dic,status=status,headers=headers)

 

以后我们一般都使用自己的Response对象,因为要啥功能自己配就行了。

posted @ 2023-10-24 10:42  weer-wmq  阅读(10)  评论(0编辑  收藏  举报