快捷導(dǎo)航

Django-rest-framework中過(guò)濾器的定制實(shí)例

更新時(shí)間：2020年04月01日 12:01:35 作者：axon-繆

這篇文章主要介紹了Django-rest-framework中過(guò)濾器的定制實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

1、定義一個(gè)自定義的filter.py模塊，增加一個(gè)新的過(guò)濾類(lèi)

import django_filters
#這個(gè)Q可以支持表查詢(xún)，單下劃線獲取表字段，雙下劃線獲取關(guān)聯(lián)表，
from django.db.models import Q
#引入自己的模型
from app.item.models import ItemCategory
#自定義的過(guò)濾類(lèi)，需要繼承django_filter.rest_framework中的FilterSet類(lèi)
class ItemCategoryFilter(django_filters.rest_framework.FilterSet):
  Filter_category = django_filters.NumberFilter(method='category_filter', label='Filter_category')
  #django_filters.NumberFilter類(lèi)似，ModelForm中字段類(lèi)型的控制
  #其中method指向自己定義的過(guò)濾函數(shù)，label用于標(biāo)識(shí)在測(cè)試API界面中的過(guò)濾界面字段，F(xiàn)ilter_category控制查詢(xún)字段
  def category_filter(self, queryset, name, value):
    #這里用到多級(jí)聯(lián)表查詢(xún)
    return queryset.filter(Q(category_id=value)|Q(category__parent_id=value)|Q(category__parent__parent_id=value)|Q(category__parent__parent__parent_id=value))
  class Meta:
    model = ItemCategory
    fields = []

2、將視圖處理類(lèi)中的，過(guò)濾器后端重新指定，將過(guò)濾器類(lèi)連接到我們自定義實(shí)現(xiàn)的處理類(lèi)上

class ItemsListByCategory(generics.ListCreateAPIView):
  '''
    根據(jù)category編號(hào)獲取相關(guān)模型no。
    請(qǐng)求中請(qǐng)帶有參數(shù)category
    @:require param: category
    such as: GET /itemsListByCategory/?Filter_category=2
  '''
  from django_filters.rest_framework import DjangoFilterBackend
  from app.item.filters import ItemCategoryFilter

  queryset = ItemCategory.objects.all()
  serializer_class = ItemCategorySerializer
  pagination_class = CommonPagination
  # 在這里哦
  filter_backends = (DjangoFilterBackend,)
  filter_class =ItemCategoryFilter
  permission_classes = (ReadOnly,)

補(bǔ)充知識(shí)：Django restful framework 中的 filter 自定義過(guò)濾

class AlertFilter(django_filters.rest_framework.FilterSet):
  
   .......
  to_bearer_service = django_filters.rest_framework.CharFilter(method='to_bearer_service_filter', label='bearer_service')

  class Meta:
    model = Alert
    fields = [......, 'to_bearer_service']

  @staticmethod  # 如果想獲取請(qǐng)求信息可以去掉這個(gè)靜態(tài)方法的裝飾器，為什么要加靜態(tài)方法裝飾器我也不清楚
  def to_bearer_service_filter(queryset, name, value):
    Q_id = Q()
    for i in value.split(","):
      if i == "0":
        Q_id.add(Q(**{'to_bearer_service':None}), Q.OR)
      else:
        Q_id.add(Q(**{'to_bearer_service_id': int(i)}), Q.OR)
    queryset = queryset.filter(Q_id)
    return queryset
 
 
  def to_bearer_service_filter(self, queryset, name, value):
    Q_id = Q()
    for i in value.split(","):
      if i == "0":
        Q_id.add(Q(**{'to_bearer_service':None}), Q.OR)
      else:
        Q_id.add(Q(**{'to_bearer_service_id': int(i)}), Q.OR)
    queryset = queryset.filter(Q_id)
    return queryset

以上這篇Django-rest-framework中過(guò)濾器的定制實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: