Как проверить вхождение пользователя в таблицу БД?

Question

[Кирилл Писарев]

Добрый день. Суть проблемы в том что имеется страница с обсуждением на которой отображается опрос, не получается сделать проверку на вхождение текущего пользователя в список проголосовавших пользователей.

Models.py (part)

...

class Question(models.Model):
    question_text = models.CharField(max_length=255)
    discussion = models.ForeignKey(Post)
    pub_date = models.DateTimeField(auto_now_add=True)
    voted_users = models.ManyToManyField(CustomUser, default=None, blank=True)

    def get_voted_users(self):
        return "\n".join([u.get_full_name() + "," for u in self.voted_users.all()])

    def __unicode__(self):
        return self.question_text

class Choice(models.Model):
    question = models.ForeignKey(Question)
    choice_text = models.CharField(max_length=100)
    votes = models.IntegerField(default=0)

    def __unicode__(self):
        return self.choice_text

view.py

def discussionDetail(request, pk, *args, **kwargs):
    '''Func that show discussion details'''
    post = get_object_or_404(Post, pk=pk)
    form = CommentForm()
    if request.user.is_authenticated():
        if request.method =="POST":
            # Check what form actually send a POST request
            if 'comment_post' in request.POST:
                form = {
                    'comment_body' : request.POST['comment_body'],
                }
                if form['comment_body']:
                    Comment.objects.create(
                        comment_body = form['comment_body'],
                        post = post,
                        user_id = getting_current_user(request),
                    )
                    return redirect('discussion_app:discussion_detail', pk)
                else:
                    messages.add_message(request, messages.INFO, "Comment can't be empty")
                    form = CommentForm()
            elif 'Vote' in request.POST:
                voted = request.POST['choice']
                voted_choice = get_object_or_404(Choice, pk=voted)
                voted_choice.votes += 1
                voted_choice.save()
                voted_question = get_object_or_404(Question, pk=voted_choice.question.id)
                voted_question.voted_users.add(getting_current_user(request))

                messages.add_message(request, messages.INFO, "Voted " + voted )
            else:
                form = CommentForm()

    return render(request, 'discussion_app/discussion_detail.html', {'post':post, 'form':form})

discussion_detail.html (part)

{% if user.is_authenticated %}

    <div id="poll">
        {% for poll in post.question_set.all %}
            {% if user in poll.voted_user.all %}
                <p>You are voted already</p>
            {% else %}
        {{ poll.question_text }}
        <form action="" method="POST">
            {% csrf_token %}
                {% for choice in poll.choice_set.all %}
                <input type="radio" name="choice" id="choice{{ forloop.counter }}" value="{{ choice.id }}"/>
                <label for="choice{{ forloop.counter }}">{{ choice.choice_text }}</label><br/>
                {% endfor %}
                <input class="btn btn-primary" type="submit" value="Vote" name="Vote" />
        </form>
            {{ poll.voted_users.count }}
            {% endif %}
        {% endfor %}

    </div>

{% endif %}

Собственно я думал что проверка {% if user in poll.voted_user.all %} будет работать, но к сожаление ничего не дает... Подскажите пожалуйста как решить данный вопрос. Если не туда копаю, то хотя бы в какую сторону нужно.

Answer 1

[Алесандр Савчук]

Лучше проверку на то что пользователь голосовал делать во view.
Например так:
is_voted = user in question.voted_users.all()
или

is_voted = question.voted_users.filter(pk=user.pk).exists()

И да, так не стоит делать :-)

voted = request.POST['choice']
voted_choice = get_object_or_404(Choice, pk=voted)

Я про то что брать данные прямо из запроса. Лучше используйте формы для этого.

Comments

[Кирилл Писарев]

Приводит к ошибке: <'ReverseManyRelatedObjectsDescriptor' object has no attribute 'filter'>

[Алесандр Савчук]

Хм... Попробуйте question.voted_users.all().filter()

[Кирилл Писарев]

Заработало таким способом:
***
elif 'Vote' in request.POST:
voted_choice = get_object_or_404(Choice, pk=request.POST['choice'])
voted_question = get_object_or_404(Question, pk=voted_choice.question.id)
current_question_voted_users = voted_question.voted_users.all()
is_voted = current_question_voted_users.filter(id=getting_current_user(request).id).exists()
if not is_voted:
voted_choice.votes += 1
voted_choice.save()
voted_question.voted_users.add(getting_current_user(request))
else:
messages.add_message(request, messages.INFO, "You are voted already")
***
Однако в данном случае на странице в любом случае отображается опрос, просто нету возможности проголосовать повторно. Но хотелось бы вынести логику на страницу, дабы не отображать опрос если человек голосовал...

[Алесандр Савчук]

Передавайте в контекст представления переменную is_voted,

return render(request, 'discussion_app/discussion_detail.html', {
    'post':post, 
    'form':form,
    'is_voted': is_voted,
})

И уже в шаблоне используйте следующую конструкцию для проверки того голосовал ли уже человек:

{% if is_voted %}
    <человек_голосовал>
{% else %}
    <человек_не_голосовал>
{% endif %}

[Кирилл Писарев]

не получиться... суть в том что is_voted содержит в себе voted_question который получаем только при нажатии формы...

[Кирилл Писарев]

Ладно, я думаю это тема другого вопроса ) Вам спасибо за помощь.