通过外键过滤 Tastypie 资源

Question

我有一些通过外键连接的 Django (3.1) 模型

class Drone(models.Model):
    serial = models.CharField(max_length=200, null=False, unique=True)

class Dive(models.Model):
    measurement_time = models.DateTimeField(db_index=True)
    drone = models.ForeignKey(
        Drone,
        on_delete=models.PROTECT,
        null=True,
    )

class Density(models.Model):
    measurement_time = models.DateTimeField(db_index=True)
    depth = models.IntegerField()
    density = models.FloatField(null=True)
    dive = models.ForeignKey(
        Dive,
        on_delete=models.PROTECT,
        db_index=True,
        null=True,
    )

我的 API 有一个 Tastypie (0.14.3) ModelResource class，定义如下

class DensityResource(ModelResource):
    class Meta:
        queryset = Density.objects.filter(dive__drone_id=13).order_by('-measurement_time', 'depth')

正如我对单曲 dive__drone_id 所期望的那样。我想实现获取参数过滤，这样我就可以做类似 /?order_by=-measurement_time&order_by=depth&dive__drone_id=13 的事情所以我将这个 class 重写为

class DensityResource(ModelResource):
    class Meta:
        queryset = Density.objects.all()
        filtering = {
            'dive__drone_id': ALL
        }
        ordering = ['measurement_time', 'depth']

并且排序工作正常但过滤不正常。我在这里错过了什么？

Answer 1

我找到了修改 build_filters 方法的解决方案：

class DensityResource(ModelResource):
    class Meta:
        queryset = Density.objects.all()
        ordering = ['measurement_time', 'depth']

    def build_filters(self, filters=None, **kwargs):
        orm_filters = super(DensityResource, self).build_filters(filters, **kwargs)
        if 'dive__drone_id' in filters.keys():
            orm_filters['dive__drone_id'] = filters['dive__drone_id']
        return orm_filters

通过外键过滤 Tastypie 资源

Tastypie resource filtering through foreign keys

django

tastypie