Django Tastypie - 使用URL参数过滤ToManyField资源

时间:2013-11-17 20:02:11

标签: python django python-2.7 tastypie

我正在使用Tastypie为我的Django(v1.5)应用程序实现API。我希望能够过滤/限制父资源时获得的相关资源。

以下是我的(简化)模型:

# myapp/models.py
class User(models.Model):
    number = models.IntegerField()
    device_id = models.CharField(verbose_name="Device ID", max_length=255)
    timezone = models.CharField(max_length=255, blank=True)

    def data(self, limit=0):
        result = Data.objects.filter(patient_id = self.id).order_by('-datetime').values('datetime', 'value')
        if limit != 0:
            result = result[:limit]
        return result

class Data(models.Model):
    user = models.ForeignKey(User)
    datetime = models.DateTimeField()
    value = models.IntegerField()

我的资源:

# api/resources.py
class DataResource(ModelResource):
    class Meta:
        queryset = Data.objects.all()
        resource_name = 'cgm'
        fields = ['value', 'datetime']
        serializer = Serializer(formats=['json', 'xml'])
        filtering = {
            'datetime': ('gte', 'lte'),
        }
        include_resource_uri = False

    def dehydrate(self, bundle):
        bundle.data['timestamp'] = calendar.timegm(bundle.data['datetime'].utctimetuple())
        return bundle


class UserResource(ModelResource):

    data = fields.ToManyField(DataResource, attribute=lambda bundle: Data.objects.filter(patient_id=bundle.obj.id), full=True, related_name='data', null=True)

    class Meta:
        queryset = User.objects.all().order_by('number')
        resource_name = 'user'
        fields = ['number', 'timezone', 'device_id'],
        serializer = Serializer(formats=['json', 'xml'])
        filtering = {
            'data': ALL_WITH_RELATIONS,
        }

我希望能够使用URL参数通过 User 资源中的'datetime'过滤 Data 资源,例如:

127.0.0.1:8000/api/v1/user/1/?format=json&datetime__gte=2013-11-14%2012:00:00 

127.0.0.1:8000/api/v1/user/1/?format=json&data__datetime__gte=2013-11-14%2012:00:00

使用给定的获取用户的时区设备ID 数据列表>日期时间

我不想单独查询 Data 资源来过滤它们,我希望整个事物捆绑在 User 资源中。 有没有办法使用框架实现应用于嵌套资源的过滤器?

感谢您的时间,我将不胜感激任何建议!

1 个答案:

答案 0 :(得分:2)

您可以使用完整功能将您传递到attribute字段的data参数扩展为DataResource

def filter_data_items(bundle):
    res = DataResource()
    new_bundle = Bundle(request=bundle.request)
    objs = res.obj_get_list(new_bundle)
    return objs.filter(parent_id=bundle.obj.pk)

res.obj_get_list根据您的DataResource处理构建和应用过滤器。您只需要在parent_id上进一步过滤它。

Reference