| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768 |
- from copy import deepcopy
- from rest_framework.response import Response
- from ...acl.objectacl import add_acl_to_obj
- from ..permissions import allow_vote_poll
- from ..serializers import NewVoteSerializer, PollSerializer
- def poll_vote_create(request, thread, poll):
- poll.make_choices_votes_aware(request.user)
- allow_vote_poll(request.user_acl, poll)
- serializer = NewVoteSerializer(
- data={"choices": request.data},
- context={"allowed_choices": poll.allowed_choices, "choices": poll.choices},
- )
- if not serializer.is_valid():
- return Response({"detail": serializer.errors["choices"][0]}, status=400)
- remove_user_votes(request.user, poll, serializer.data["choices"])
- set_new_votes(request, poll, serializer.data["choices"])
- add_acl_to_obj(request.user_acl, poll)
- serialized_poll = PollSerializer(poll).data
- poll.choices = list(map(presave_clean_choice, deepcopy(poll.choices)))
- poll.save()
- return Response(serialized_poll)
- def presave_clean_choice(choice):
- del choice["selected"]
- return choice
- def remove_user_votes(user, poll, final_votes):
- removed_votes = []
- for choice in poll.choices:
- if choice["selected"] and choice["hash"] not in final_votes:
- poll.votes -= 1
- choice["votes"] -= 1
- choice["selected"] = False
- removed_votes.append(choice["hash"])
- if removed_votes:
- poll.pollvote_set.filter(voter=user, choice_hash__in=removed_votes).delete()
- def set_new_votes(request, poll, final_votes):
- for choice in poll.choices:
- if not choice["selected"] and choice["hash"] in final_votes:
- poll.votes += 1
- choice["votes"] += 1
- choice["selected"] = True
- poll.pollvote_set.create(
- category=poll.category,
- thread=poll.thread,
- voter=request.user,
- voter_name=request.user.username,
- voter_slug=request.user.slug,
- choice_hash=choice["hash"],
- )
|
