Django Rest Framework中与用户相关的字段具有多对多关系

debugcn 发表于 Dev

温格

在我的应用程序中，我有一些产品，并且已登录的用户应该能够将这些产品标记为收藏。

我将它们建模为我中的多对多关系models.py，如下所示：

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()

我现在想要实现的是，在产品详细信息视图上，用户可以查看该产品是否已经是他的最爱，并且应该能够选中一个框使其成为他的最爱。

在模型级别上，这是多对多关系，但在视图上，它更多是布尔关系（当前用户最喜欢的是/否）。

我已经设法将喜欢的状态显示在详细信息视图中，并带有 SerializerMethodField

class ProductSerializer(serializers.ModelSerializer):
    # ...
    favorite = serializers.SerializerMethodField()

    def get_favorite(self, obj):
    try:
        self.context['request'].user.profile.favorites.get(pk=obj.pk)
        return True
    except Product.DoesNotExist:
        return False

我目前正在努力的是使用此设置将收藏夹添加到当前用户的收藏夹列表中，因为它SerializerMethodField是只读的，因此不会出现在update()方法的validated_data中。

是否有人对此或类似问题有解决方案？

一次

您可以创建自定义字段。

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

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。