Как объединить два кверисета из двух моделей? Django Rest Framework

У меня есть две модели

class Answer(models.Model):
    answer = models.TextField()
    created_at = models.DateTimeField(editable=False, default=timezone.now)
    updated_at = models.DateTimeField(default=timezone.now)
    user = models.ForeignKey('users.CustomUser', on_delete=models.PROTECT)
    question = models.ForeignKey('Question', on_delete=models.PROTECT)
    number_of_points = models.IntegerField(default=0)
    moderate_status = models.BooleanField(default=False)

и 

class Question(models.Model):
    question_subject = models.TextField()
    question_text = models.TextField(default=None, null=True, blank=True)
    slug = models.SlugField(max_length=128, unique=True, null=False, editable=False)
    created_at = models.DateTimeField(editable=False, default=timezone.now)
    updated_at = models.DateTimeField(default=timezone.now)
    user = models.ForeignKey('users.CustomUser', on_delete=models.PROTECT)
    animal = models.ForeignKey('animals.Animal', on_delete=models.PROTECT)

serializers.py

class QuestionDetailSerializer(serializers.ModelSerializer):
    answers = serializers.PrimaryKeyRelatedField(many=True, read_only=True)

    class Meta:
        model = Question
        fields = '__all__'

views.py

class QuestionsDetailView(generics.ListAPIView):
    queryset = Question.objects.all()
    serializer_class = QuestionsSerializer

    def get_queryset(self):
        return super().get_queryset().filter(
            id=self.kwargs['pk']
        )

url.py

path('questions/<int:pk>', QuestionsDetailView.as_view()),

Я хочу объединить 2 кверисета, один уже есть, который фильтрует вопрос по pk, указанному в url, а другой кверисет я хочу дать Answer.objects.all().filter(question__id='pk'). По сути, я хочу показать все вопросы с ответами на конкретный вопрос.

Я настоятельно рекомендую использовать отношения везде, где только можно.

Добавьте related_name к Answer.question в models.py:

class Answer(models.Model):
    ...
    question = models.ForeignKey('Question', on_delete=models.PROTECT, related_name='answers')

Если id достаточно в сериализаторе, то в serializers.py:

class QuestionDetailSerializer(serializers.ModelSerializer):
    # this is not needed anymore: answers = serializers.PrimaryKeyRelatedField(many=True, read_only=True)

    class Meta:
        model = Question
        fields = ['question_subject ', 'question_text', ..., 'answers`]

Но вы также можете создать функцию model, которую можно включить аналогичным образом:

class Question(models.Model):
    ...
    def get_answers(self):
        return [answer.answer for answer in self.answers.all()]    # it will return a list of TextFields of answers

serializers.py:

class QuestionDetailSerializer(serializers.ModelSerializer):
    class Meta:
        model = Question
        fields = ['question_subject ', 'question_text', ..., 'get_answers`]

Вы не сможете точно объединить 2 кверисета из разных таблиц. Но вы можете использовать вложенный сериализатор.

class AnswerSerializer(serializers.ModelSerializer):

    class Meta:
        model = Anwser
        fields = ['id', ...]

class QuestionDetailSerializer(serializers.ModelSerializer):
    answer_set = AnswerSerializer(many=True, read_only=True)

    class Meta:
        model = Question
        fields = ['id', ..., 'answer_set']

# OR


class QuestionDetailSerializer(serializers.ModelSerializer):

    class Meta:
        model = Question
        fields = ['id']
    
    def to_representation(self, instance):
        data = super().to_representation(instance)
        data['answers'] = AnswerSerializer(many=True, instance=instance.anwser_set.all()).data
        return data
Вернуться на верх