akshar-raaj

# polls/models.py
class Question(models.Model):
    ....
    def choices(self):
        if not hasattr(self, '_choices'):
            self._choices = self.choice_set.all()
        return self._choices

# polls/serializers.py
class ChoiceSerializer(serializers.ModelSerializer):

akshar-raaj

class QuestionSerializer(serializers.ModelSerializer):
    choices = ChoiceSerializer(read_only=True, many=True)

    class Meta:
        model = Question
        fields = '__all__'

akshar-raaj

class QuestionListPageSerializer(serializers.ModelSerializer):

    was_published_recently = serializers.BooleanField(read_only=True)

    class Meta:
        model = Question
        fields = '__all__'


class QuestionDetailPageSerializer(QuestionListPageSerializer):

akshar-raaj

@api_view(['GET', 'POST'])
def questions_view(request):
    if request.method == 'GET':
        questions = Question.objects.all()
        serializer = QuestionListPageSerializer(questions, many=True)
        return Response(serializer.data)
    elif request.method == 'POST':
        serializer = QuestionSerializer(data=request.data)
        if serializer.is_valid():
            question = serializer.save()

akshar-raaj

class VoteSerializer(serializers.Serializer):
    choice_id = serializers.IntegerField()

akshar-raaj

path('questions/<int:question_id>/vote/', apiviews.vote_view, name='vote_view'),

akshar-raaj

@api_view(['PATCH'])
def vote_view(request, question_id):
    question = get_object_or_404(Question, pk=question_id)
    serializer = VoteSerializer(data=request.data)
    if serializer.is_valid():
        choice = get_object_or_404(Choice, pk=serializer.validated_data['choice_id'], question=question)
        choice.votes += 1
        choice.save()
        return Response("Voted")
    return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

akshar-raaj

class ChoiceSerializerWithVotes(ChoiceSerializer):

    class Meta(ChoiceSerializer.Meta):
        fields = ChoiceSerializer.Meta.fields + ('votes',)

class QuestionResultPageSerializer(QuestionListPageSerializer):
    choices = ChoiceSerializerWithVotes(many=True, read_only=True)

akshar-raaj

@api_view(['GET'])
def question_result_view(request, question_id):
    question = get_object_or_404(Question, pk=question_id)
    serializer = QuestionResultPageSerializer(question)
    return Response(serializer.data)

akshar-raaj

path('questions/<int:question_id>/result/', apiviews.question_result_view, name='question_result_view')