Django:在所有queryset对象上运行一个方法

时间:2013-02-15 22:20:17

标签: django django-rest-framework

我想知道以下是否可行以及是否有人可以解释如何。我正在使用Django REST Framework

我有一个模型,在该模型中我有一个名为产品的类。产品具有名为 is_product_safe_for_user 的方法。它需要用户对象和自我(产品)。

model.py

class Product(models.Model):

    title = models.CharField(max_length=60, help_text="Title of the product.")
    for_age = models.CharField(max_length=2,)

    def is_product_safe_for_user(self, user):
        if self.for_age > user.age
        return "OK"

(忽略上面的语法,它只是为了给你一个想法)

我想要做的是运行所有查询集对象的方法,如下所示,但我不知道如何......

class ProductListWithAge(generics.ListAPIView):
    permission_classes = (permissions.IsAuthenticated,)
    model = Product
    serializer_class = ProductSerializer

    def get_queryset(self):

      Product.is_product_safe_for_user(self,user)

       # then somehow apply this to my queryset

        return Product.objects.filter()

还有时我想在一个对象上运行方法。

还是应该进入Serializer?如果是这样的话?...

class ProductSerializer(serializers.ModelSerializer):

    safe = serializers.Field(Product='is_product_safe_for_user(self,user)')

    class Meta:
        model = Product
        fields = ('id', 'title', 'active', 'safe')

1 个答案:

答案 0 :(得分:5)

你可以为你的模型写一个custom manager。像这样:

class OnlySafeObjects(models.Manager):
    def filter_by_user(self, user):
        return super(OnlySafeObjects, self).get_query_set().filter(for_age__gte=user.age)

class Product(models.Model):
     # your normal stuff
     onlysafeobjects = OnlySafeObjects()

然后你会像这样使用它:

safe_products = Product.onlysafeobjects.filter_by_user(request.user)