DRF——viewsets使用

最新推荐文章于 2025-01-11 15:40:17 发布

cod16xx

最新推荐文章于 2025-01-11 15:40:17 发布

阅读量1.2k

点赞数

CC 4.0 BY-SA版权

分类专栏： django

本文链接：https://blog.youkuaiyun.com/qq_35037977/article/details/79404942

django 专栏收录该内容

24 篇文章

订阅专栏

本文介绍Django REST framework中视图集的使用方法，包括如何通过GenericViewSet实现RESTful API，以及如何利用过滤类实现数据过滤。文章还详细介绍了自定义过滤方法和排序字段的设置。

GenericViewSet继承的ViewSetMixin重写了视图类的as_view方法，viewsets可以使用router进行字段注册，也可以自己进行绑定

from rest_framework.routers import DefaultRouter
router=DefaultRouter()
router.register(r'goods',GoodsListViewSet)

class GoodsSerializer(serializers.ModelSerializer):
    """
    序列化类
    """
    class Meta:
        model=Goods
        fields="__all__"

class ModelFilter(django_filters.rest_framework.FilterSet):
    """
    过滤类
    """
    pricemin=django_filters.NumberFilter(name="shop_price",lookup_expr="gt")
    pricemax = django_filters.NumberFilter(name="shop_price", lookup_expr="lt")
    top_category=django_filters.NumberFilter(method="top_category_filter")
    def top_category_filter(self,queryset,name,value):
        queryset=queryset.filter(Q(category_id=value)|
                                 Q(category__parent_category__parent_category_id=value)|
                                 Q(category__parent_category_id=value))
        return queryset

    class Meta:
        model=Goods
        fields=["pricemin","pricemax","top_category"]

from rest_framework import mixins，viewsets,filters
from rest_framework.pagination import PageNumberPagination
from django_filters.rest_framework import DjangoFilterBackend

from .models import Model
from .serializer import ModelSerializer
from .filters import ModelFilter 

class StandardResultsSetPagination(PageNumberPagination):
    page_size = 12
    page_size_query_param = 'page_size'
    page_query_param = "page"
    max_page_size = 100

#mixins.ListModelMixin提供了list方法，接收到get请求时会被调用
class GoodsListViewSet(mixins.ListModelMixin,viewsets.GenericViewSet):
    """
    列出模型所有数据,包含分页，搜索，过滤，排序功能
    """
    queryset = Model.objects.all().order_by("id")
    serializer_class = ModelSerializer #自定义序列化类
    pagination_class = StandardResultsSetPagination #自定义分页类
    #设置过滤的backends
    filter_backends = (DjangoFilterBackend,filters.SearchFilter,filters.OrderingFilter)


    #filter_fields=(("","","")#过滤字段，也可以自定义过滤的类
    filter_class=ModelFilter #自定义过滤的类
    search_fields=("","","") #搜索字段
    ordering_fields=("","")  #排序字段