from rest_framework import serializers
from .models import (
    Survey,
    ScoreVariable,
    Statement,
    QuestionPair,
    AgreementQuestion,
    EducationalInstitution,
    Profession,
    Specialty,
    ProfessionSpecialtyLink,
    ProfessionTestResult,
)

class SurveySerializer(serializers.ModelSerializer):
    class Meta:
        model = Survey
        fields = "__all__"


class ScoreVariableSerializer(serializers.ModelSerializer):
    class Meta:
        model = ScoreVariable
        fields = "__all__"


class StatementSerializer(serializers.ModelSerializer):
    class Meta:
        model = Statement
        fields = "__all__"


class QuestionPairSerializer(serializers.ModelSerializer):
    statement_a_text = serializers.SerializerMethodField()
    statement_b_text = serializers.SerializerMethodField()

    class Meta:
        model = QuestionPair
        fields = ["id", "survey", "statement_a", "statement_b", "statement_a_text", "statement_b_text"]

    def get_statement_a_text(self, obj):
        return obj.statement_a.statement

    def get_statement_b_text(self, obj):
        return obj.statement_b.statement


class AgreementQuestionSerializer(serializers.ModelSerializer):
    class Meta:
        model = AgreementQuestion
        fields = "__all__"


class EducationalInstitutionSerializer(serializers.ModelSerializer):
    class Meta:
        model = EducationalInstitution
        fields = "__all__"


class ProfessionSerializer(serializers.ModelSerializer):
    class Meta:
        model = Profession
        fields = "__all__"


class SurveySpecialtySerializer(serializers.ModelSerializer):
    class Meta:
        model = Specialty
        fields = "__all__"


class ProfessionSpecialtyLinkSerializer(serializers.ModelSerializer):
    class Meta:
        model = ProfessionSpecialtyLink
        fields = "__all__"


class ProfessionTestResultSerializer(serializers.ModelSerializer):
    class Meta:
        model = ProfessionTestResult
        fields = "__all__"