Question

我有一些产品，用户可以将其标记为他最喜欢的产品。

收藏夹被建模为用户个人资料和产品之间的多对多关系。

我现在想通过API调用来获取产品，但首先按收藏夹排序，其他所有内容排在第二位。现在因为这取决于当前请求，我不能通过Product.objects.all().order_by(...)执行此操作，因为模型不知道当前用户，并且我现在知道如何通过序列化程序或ModelViewSet告诉查询。

我尝试获取两个查询集，一个包含所有产品，一个包含ModelViewSet中用户个人资料中的所有收藏，如下所示：

class ProductViewSet(viewsets.ModelViewSet):
    serializer_class = ProductCreateSerializer
    # ...

    def get_queryset(self):
        favorited_products = self.request.user.profile.favorites.all()
        products = Product.objects.all()

        queryset = favorited_products | products

        return queryset

这确实有效，因为实体保持此顺序。但是这个视图返回了几百个实体，所以当我尝试用return queryset[:30]限制查询集时，默认排序似乎接管了。

我尝试做的事情是否容易实现？有没有人解决过类似的问题？

以下是我的模特：

class Product(models.Model):
    product_name = models.CharField(max_length=200)
    # ...


    def __unicode__(self):
        return self.product_name


class Profile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    favorites = models.ManyToManyField(Product)

    @receiver(post_save, sender=User)
    def create_user_profile(sender, instance, created, **kwargs):
        if created:
            Profile.objects.get_or_create(user=instance)

    @receiver(post_save, sender=User)
    def save_user_profile(sender, instance, **kwargs):
        instance.profile.save()

我的序列化器：

class FavoriteField(serializers.BooleanField):
    def get_attribute(self, instance):

        return self.context['request'].user.profile.favorites.filter(pk=instance.pk).exists()

class ProductSerializer(serializers.ModelSerializer):
    favorite = FavoriteField()

    def update(self, instance, validated_data):
        instance = super().update(instance, validated_data)
        is_favourite = validated_data.get('favorite')  # can be None
        if is_favourite is True:
            self.context['request'].user.profile.favorites.add(instance)
        elif is_favourite is False:
            self.context['request'].user.profile.favorites.remove(instance)
        return instance

Answer 1

我找到了一个解决方案，使用Count，Case和IntegerField：

user_id = self.request.user.pk
queryset = Product.objects \
        .annotate(favorited=Count(Case(When(favorited_by__user__pk=user_id, then=1), output_field=IntegerField()))) \
        .order_by("-favorited")

Django Rest Framework：按用户依赖的字段排序

1 个答案: