如果同时存在 3 个对象，则删除或更新 - Django Rest Framework

Question

我正在为我的博客网站构建 LIKE、UNLIKE 和 CLAPPING 功能。我有这样的模型： 模型反应（喜欢、不喜欢和鼓掌）

class Reaction(models.Model):
    REACT_TYPES = (
        (LIKE, 'Like'),
        (CLAPPING, 'Clapping')
    )
    user = models.ForeignKey(User)
    react_type = models.CharField(max_length=100, choices=REACT_TYPES, default='LIKE')
    timestamp = models.DateTimeField(auto_now_add=True, null=True)
    content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE, null=True)
    object_id = models.PositiveIntegerField(null=True)
    content_object = GenericForeignKey('content_type', 'object_id')

    class Meta:
        unique_together = ('user', 'content_type', 'object_id')

序列化器模型 REACTION 从模型导入反应

class ReactCreateUpdateSerializer(ModelSerializer):
    class Meta:
        model = Reaction
        fields = [
            'user',
            'react_type',
            'content_type',
            'object_id',
        ] 

我的 REACT 视图集：

from rest_framework.generics import CreateAPIView
class ReactCreateAPIView(CreateAPIView):
    queryset = Reaction.objects.all()
    serializer_class = ReactCreateUpdateSerializer

我假设之前创建了一个带有 LIKE 的对象。我想构建一个函数可以做这些事情：

• 如果用户再次使用存在 4 个对象的 POST 方法：user, content_type, object_id, react_type=LIKE。它成为删除对象。 （不喜欢）。然后，如果再发布一个，它将再次成为创建对象。 （喜欢）
• 如果用户使用存在 4 个对象的 POST 方法：user, content_type, object_id, react_type=CLAPPING。它变为使用更新 2 字段更新数据库中的可用反应：react_type 和时间戳，所有其他字段均未更改。

我认为这是每个人在为网站构建 API 需要反应时都会遇到的问题。所以希望您的热心帮助。提前致谢！

Answer 1

您只需在请求数据中发送react_type， 如果您将 @detail_route 创建为 pk，object_id 将始终存在，如果用户以 身份登录，user 对象将在请求中可用request.user

任何用户的反应对象都将在该用户提出第一个请求时创建。

不需要删除记录，如果用户已经LIKED，则必须更新反应对象react_type=UNLIKE。

class PostViewSet(viewsets.ModelViewSet):
    # Data model on which user is reacting
    queryset = Post.objects.all()
    serializer_class = PostSerializer

    @detail_route(methods=['post'])
    def react(self, request, pk=None):

        react_type = request.data['react_type']
        obj = self.get_object()

        # GenericForeignKey does not support get_or_create
        try:
            reaction_obj = Reaction.objects.get(content_type=ContentType.objects.get_for_model(obj), object_id=obj.id, user=request.user)  
            reaction_obj.react_type = react_type
            reaction_obj.save()
        except Reaction.DoesNotExist:
            obj.reactions.create(user=request.user, react_type=react_type)

        return Response({'success': True})

用户反应的模型应该具有多态连接

class Post(models.Model)
    # Other fields
    ...
    reactions = GenericRelation(Reaction, related_query_name='posts')

将您的视图注册为

url(r'^posts/(?P<pk>[0-9]+)/$', PostViewSet.as_view(), name='post-reaction')
# http://domian.com/api/v1/posts/1/react 
# method: POST
# data : {'react_type': 'UNLIKE'}
# content-type: 'application/json'