Question

我需要制作一个简单的API，在其中发送创建投票实例的表单并返回201状态代码，但是如果用户已经投票，它将返回400错误请求。

我的模型如下：

class Company(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    short_name = models.CharField(max_length=24)

    def __str__(self):
        return f"{self.short_name}"


class Vote(models.Model):
    voter = models.ForeignKey(Company, on_delete=models.CASCADE)
    date_voted = models.DateField(auto_now=True)

    def __str__(self):
        return f"{self.id}/{self.voter}"

views.py：

class VoteView(CreateAPIView):
    queryset = Vote.objects.all()
    serializer_class = TaskCreateSerializer
    permission_classes = (IsAuthenticated,)

    def perform_create(self, serializer):
        if serializer.is_valid():
            user = self.request.user
            voter = Company.objects.get(user=user)
            has_voted = Vote.objects.filter(voter=voter)
            serializer.save(voter=voter)

和serializers

class TaskCreateSerializer(ModelSerializer):

    class Meta:
        model = Vote
        fields = [
        ]

现在它总是总是返回201。我正在寻找一种方法，可以编写可以检查当前用户的投票实例是否存在然后返回Response(data={"message":"You have already voted"}, status=status.HTTP_400_BAD_REQUEST)的逻辑。最简单的方法。

如果用户尚未投票，则DRF返回201；如果投票，则返回400

0 个答案: