def post(self, request: HttpRequest) -> Response:
"""
Create a new election.
"""
title = request.data.get("title")
initiator_name = request.data.get("initiator_name")
if not title:
return Response(
{"error": "Missing parameter: `title`."},
status=status.HTTP_400_BAD_REQUEST,
)
if not initiator_name:
return Response(
{"error": "Missing parameter: `initiator_name`."},
status=status.HTTP_400_BAD_REQUEST,
)
device_token = request.headers.get("X-Device-ID")
if not device_token:
return Response(
{"error": "Missing header: `X-Device-ID`."},
status=status.HTTP_400_BAD_REQUEST,
)
device = users_manager.get_or_create_device(device_token)
election = manager.initiate_election(device, initiator_name, title)
election_document = builders.build_election_document(election)
return Response(election_document, status=status.HTTP_201_CREATED)
def delete(self, request: HttpRequest, candidate_id: str) -> Response:
"""
Remove a vote from one of the candidates in an election.
"""
candidate = manager.get_candidate_and_related_objects(
int(candidate_id))
if not candidate:
return Response({}, status=status.HTTP_404_NOT_FOUND)
device_token = request.headers.get("X-Device-ID")
if not device_token:
return Response(
{"error": "Missing header: `X-Device-ID`."},
status=status.HTTP_400_BAD_REQUEST,
)
participant = manager.get_participant_by_election_and_device_token(
candidate.election, device_token)
if not participant:
return Response(
{
"error":
"Participant with the provided device ID does not exist in the election."
},
status=status.HTTP_400_BAD_REQUEST,
)
vote = manager.get_vote_by_participant_and_candidate(
participant, candidate)
if not vote:
return Response(
{
"error":
"Vote with the provided device ID does not exist for the candidate."
},
status=status.HTTP_400_BAD_REQUEST,
)
manager.delete_vote(vote)
# Send election server-side event.
candidate_actions_map = manager.get_candidate_actions_map_for_election(
candidate.election, candidate.participant)
election_document = builders.build_election_document(
candidate.election, candidate_actions_map)
manager.send_election_event(election_document)
actions = manager.get_candidate_actions_for_participant(
candidate, participant)
candidate_document = builders.build_candidate_document(
candidate, actions)
return Response(candidate_document, status=status.HTTP_200_OK)
def post(self, request: HttpRequest, election_id: str) -> Response:
"""
Create a new movie candidate. Called when a user finds a movie they want to suggest
to the group.
"""
try:
election = Election.objects.get(external_id=election_id)
except Election.DoesNotExist:
return Response({}, status=status.HTTP_404_NOT_FOUND)
device_token = request.headers.get("X-Device-ID")
if not device_token:
return Response(
{"error": "Missing header: `X-Device-ID`."},
status=status.HTTP_400_BAD_REQUEST,
)
participant = manager.get_participant_by_election_and_device_token(
election, device_token)
if not participant:
return Response(
{
"error":
"Participant with the provided device ID does not exist in the election."
},
status=status.HTTP_400_BAD_REQUEST,
)
movie_id = request.data.get("movie_id")
if not movie_id:
return Response(
{"error": "Missing parameter: `movie_id`."},
status=status.HTTP_400_BAD_REQUEST,
)
try:
manager.create_candidate_for_election(election, participant,
movie_id)
except (errors.CandidateAlreadyExistsError,
errors.MovieDoesNotExistError) as e:
return Response({"error": e.message},
status=status.HTTP_400_BAD_REQUEST)
candidate_actions_map = manager.get_candidate_actions_map_for_election(
election, participant)
election_document = builders.build_election_document(
election, candidate_actions_map)
manager.send_election_event(election_document)
return Response(election_document, status=status.HTTP_201_CREATED)
def post(self, request: HttpRequest, candidate_id: str) -> Response:
"""
Cast a vote for one of the candidates in an election.
"""
candidate = manager.get_candidate_and_related_objects(
int(candidate_id))
if not candidate:
return Response({}, status=status.HTTP_404_NOT_FOUND)
device_token = request.headers.get("X-Device-ID")
if not device_token:
return Response(
{"error": "Missing header: `X-Device-ID`."},
status=status.HTTP_400_BAD_REQUEST,
)
participant = manager.get_participant_by_election_and_device_token(
candidate.election, device_token)
if not participant:
return Response(
{
"error":
"Participant with the provided device ID does not exist in the election."
},
status=status.HTTP_400_BAD_REQUEST,
)
try:
manager.create_or_activate_vote_for_candidate(
participant, candidate)
except (
errors.ParticipantNotPartOfElectionError,
errors.ParticipantAlreadyVotedForCandidate,
) as e:
return Response({"error": e.message},
status=status.HTTP_400_BAD_REQUEST)
# Send election server-side event.
candidate_actions_map = manager.get_candidate_actions_map_for_election(
candidate.election, candidate.participant)
election_document = builders.build_election_document(
candidate.election, candidate_actions_map)
manager.send_election_event(election_document)
actions = manager.get_candidate_actions_for_participant(
candidate, participant)
candidate_document = builders.build_candidate_document(
candidate, actions)
return Response(candidate_document, status=status.HTTP_201_CREATED)
def put(self, request: HttpRequest, election_id: str) -> Response:
"""
Update the attributes of an election.
"""
device_token = request.headers.get("X-Device-ID")
if not device_token:
return Response(
{"error": "Missing header: `X-Device-ID`."},
status=status.HTTP_400_BAD_REQUEST,
)
title = request.data.get("title")
if not title:
return Response(
{"error": "Missing parameter: `title`."},
status=status.HTTP_400_BAD_REQUEST,
)
election = manager.get_election_and_related_objects(election_id)
if not election:
return Response({}, status=status.HTTP_404_NOT_FOUND)
participant = manager.get_participant_by_election_and_device_token(
election, device_token)
if not participant:
return Response(
{
"error":
"Participant with the provided device ID does not exist in the election."
},
status=status.HTTP_400_BAD_REQUEST,
)
try:
manager.update_election(election, participant, title)
except errors.ParticipantDidNotInitiateElectionError as e:
return Response({"error": e.message},
status=status.HTTP_400_BAD_REQUEST)
election_document = builders.build_election_document(election)
manager.send_election_event(election_document)
return Response(election_document, status=status.HTTP_200_OK)
def delete(self, request: HttpRequest, election_id: str) -> Response:
"""
Remove a participant from an election. This is called when a user chooses to leave an
election. Deleted participants and their votes will no longer appear in the election
document.
"""
try:
election = Election.objects.get(external_id=election_id)
except Election.DoesNotExist:
return Response({}, status=status.HTTP_404_NOT_FOUND)
device_token = request.headers.get("X-Device-ID")
if not device_token:
return Response(
{"error": "Missing header: `X-Device-ID`."},
status=status.HTTP_400_BAD_REQUEST,
)
participant = manager.get_participant_by_election_and_device_token(
election, device_token)
if not participant:
return Response(
{
"error":
"Participant with the provided device ID does not exist in the election."
},
status=status.HTTP_400_BAD_REQUEST,
)
manager.delete_participant(participant)
candidate_actions_map = manager.get_candidate_actions_map_for_election(
election, participant)
election_document = builders.build_election_document(
election, candidate_actions_map)
manager.send_election_event(election_document)
return Response(election_document, status=status.HTTP_200_OK)
def post(self, request: HttpRequest, election_id: str) -> Response:
"""
Create a new participant for an existing election. Called when a user clicks an
election link and is prompted to enter their name and join the election.
"""
try:
election = Election.objects.get(external_id=election_id)
except Election.DoesNotExist:
return Response({}, status=status.HTTP_404_NOT_FOUND)
name = request.data.get("name")
if not name:
return Response(
{"error": "Missing parameter: `name`."},
status=status.HTTP_400_BAD_REQUEST,
)
device_token = request.headers.get("X-Device-ID")
if not device_token:
return Response(
{"error": "Missing header: `X-Device-ID`."},
status=status.HTTP_400_BAD_REQUEST,
)
device = users_manager.get_or_create_device(device_token)
participant = manager.create_or_activate_participant_for_election(
election, device, name)
candidate_actions_map = manager.get_candidate_actions_map_for_election(
election, participant)
election_document = builders.build_election_document(
election, candidate_actions_map)
manager.send_election_event(election_document)
return Response(election_document, status=status.HTTP_201_CREATED)
def get(self, request: HttpRequest, election_id: str) -> Response:
"""
Get information about a single election, including participants and candidates.
"""
election = manager.get_election_and_related_objects(election_id)
if not election:
return Response({}, status=status.HTTP_404_NOT_FOUND)
candidate_actions_map = None
device_token = request.headers.get("X-Device-ID")
if device_token:
participant = manager.get_participant_by_election_and_device_token(
election, device_token)
if participant:
candidate_actions_map = manager.get_candidate_actions_map_for_election(
election, participant)
election_document = builders.build_election_document(
election, candidate_actions_map)
return Response(election_document, status=status.HTTP_200_OK)
def get(self, request: HttpRequest) -> Response:
"""
Retrieve all elections the user is a participating in.
"""
device_token = request.headers.get("X-Device-ID")
if not device_token:
return Response(
{"error": "Missing header: `X-Device-ID`."},
status=status.HTTP_400_BAD_REQUEST,
)
elections = manager.get_elections_and_related_objects_by_device_token(
device_token)
response_body = {
"results": [
builders.build_election_document(election)
for election in elections
]
}
return Response(response_body, status=status.HTTP_200_OK)