通过Django-rest-framework中的一系列对象属性过滤对象

时间:2017-08-08 04:01:15

标签: django django-rest-framework django-filter

我正在Django Rest Framework中创建一个Rest API,我有一个存储这样数据的模型 -

   class Commodity(models.Model):
      def countryname(self):
        return self.country.country_name
      country=models.ForeignKey(Country)
      commodity_name=models.CharField(max_length=255)
      commodity_type=models.IntegerField(choices=COMMODITY_TYPE)
      commodity_year=models.IntegerField(default=None)
      commodity_production=models.IntegerField(default=None)

      def __unicode__(self):
        return self.commodity_name +"-"+str(self.country.country_name)

现在我想得到所有那些属于commodity_year范围的对象,例如,commodity_year在1999年到2014年之间(包括两者)的所有对象。我一直在使用Django-filter课,但没有运气。

class CommodityFilter(filters.FilterSet):
   start_date = DateFilter(name='commodity_year' )
   end_date = DateFilter(name='commodity_year')
   class Meta:
    model = Commodity
    fields =  [ 'country','commodity_name', 'commodity_type','start_date','end_date']


 class CommodityList(generics.ListCreateAPIView):

   queryset = Commodity.objects.all()
   serializer_class = CommoditySerializer
   filter_backends = (filters.DjangoFilterBackend,)
   filter_class = CommodityFilter
   permission_classes = [AllowAny]

如果有人有更好的方法,请指导我。

2 个答案:

答案 0 :(得分:1)

您可以使用小于等于和大于等于的范围进行过滤。

queryset = Commodity.objects.filter(commodity_year__gte=1999,
                                    commodity_year__lte=2014)

以上内容将为您提供一个commodity_year小于2014且大于1999的查询集。在django docs上了解gt的更多信息

注意:我不认为需要过滤类,因为它是一个非常简单的查询

答案 1 :(得分:0)

CommodityFilter课程中尝试此操作:

class CommodityFilter(filters.FilterSet):
    start_date = django_filters.NumberFilter(
        name='commodity_year', lookup_expr='gte')
    end_date = django_filters.NumberFilter(
        name='commodity_year', lookup_expr='lte')

    class Meta:
        model = Commodity
        fields =  [ 'country','commodity_name', 'commodity_type','start_date','end_date']