民意调查正在添加多票答案

【问题标题】：Poll is adding multiple votes民意调查正在添加多票
【发布时间】：2021-05-28 10:14:43
【问题描述】：

我正在构建一个 PollApp，但遇到了一个问题。

我想要做什么：-

我试图只对One choice at a time 进行投票，如果用户想更改voting choice，那么用户也可以更改选择。但是此时，当用户选择first choice 时，它会一次又一次地添加投票，如果用户选择second option，那么两者都在添加（第一选择和第二选择）。但我试图一次只添加一个选择。

views.py

def poll_vote(request, poll_id):
    poll = get_object_or_404(Poll, pk=poll_id)
    choice_id = request.POST.get('choice')

    if choice_id:
        choice = Choice.objects.get(id=choice_id)
        vote = Vote(user=request.user, poll=poll, choice=choice)
        vote.save()
        print(vote)
        return render(request, 'polls/poll_result.html', {'poll': poll})
    else:
        messages.error(
            request, "No choice selected")
        return redirect("polls:polls_detail_view", poll_id)
    return render(request, 'polls/poll_result.html', {'poll': poll})

我也在poll_vote 视图中尝试过这个

    if not poll.user_can_vote(request.user):
        messages.error(request, "You already voted this poll")
        return redirect("polls:polls_detail_view",poll_id=poll_id)

这很好用：- 当用户投票 first choice 时，它会阻止用户选择第二个选项，但我想要，用户可以在投票后更改另一个选项。

models.py


class Poll(models.Model):
    owner = models.ForeignKey(User,null=True,on_delete=models.CASCADE)
    text = models.TextField()
    date = models.DateTimeField(default=timezone.now)

    def user_can_vote(self, user):

        user_votes = user.vote_set.all()
        qs = user_votes.filter(poll=self)
        if qs.exists():
            return False
        return True

class Choice(models.Model):
    poll = models.ForeignKey(Poll,on_delete=models.CASCADE)
    choice_text = models.CharField(max_length=30)


class Vote(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    poll = models.ForeignKey(Poll,on_delete=models.CASCADE)
    choice = models.ForeignKey(Choice,on_delete=models.CASCADE)

我不知道，如何让用户在投票后改变投票选择。

任何帮助将不胜感激。

在此先感谢您。

【问题讨论】：

标签： python django django-models django-views

【解决方案1】：

您可以使用update_or_create [Django docs] 更新现有投票或创建新投票：

def poll_vote(request, poll_id):
    poll = get_object_or_404(Poll, pk=poll_id)
    choice_id = request.POST.get('choice')

    if choice_id:
        choice = Choice.objects.get(id=choice_id)
        vote, created = Vote.objects.update_or_create(
            user=request.user,
            poll=poll,
            defaults={'choice': choice}
        )
        print(vote)
        return render(request, 'polls/poll_result.html', {'poll': poll})
    else:
        messages.error(
            request, "No choice selected")
        return redirect("polls:polls_detail_view", poll_id)
    return render(request, 'polls/poll_result.html', {'poll': poll})

如果您有任何重复的 Vote 实例，您应该在使用它之前先删除它们。此外，您应该在 Vote 模型中添加 UniqueConstraint [Django docs] 以防止重复：

class Vote(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    poll = models.ForeignKey(Poll,on_delete=models.CASCADE)
    choice = models.ForeignKey(Choice,on_delete=models.CASCADE)
    
    class Meta:
        constraints = [
            models.UniqueConstraint(fields=['user', 'poll'], name='unique_user_poll')
        ]

【讨论】：

【解决方案2】：

需要检查该用户是否已有投票。每次发布带有选项的表单时，您的代码都会创建一个新投票。

choice = Choice.objects.get(id=choice_id)
if existing_vote := Vote.objects.filter(user=request.user, poll=poll).first():
    existing_vote.choice = choice
    existing_vote.save()
else:
    vote = Vote(user=request.user, poll=poll, choice=choice)
    vote.save()
return render(request, 'polls/poll_result.html', {'poll': poll})

【讨论】：