from rest_framework import serializers

from .models import Apartament, User


class ApartamentListSerializer(serializers.ModelSerializer):
    """Список всех квартиры"""

    class Meta:
        model = Apartament
        exclude = ("views",)


class ApartamentDetailSerializer(serializers.ModelSerializer):
    """Вывод квартиры"""

    class Meta:
        model = Apartament
        fields = "__all__"

class PsychTestAddResultSerializer(serializers.ModelSerializer):
    class Meta:
        model = User
        fields = ("psych_test_result",)

class PublicUserSerializer(serializers.ModelSerializer):
    class Meta:
        model = User
        exclude = ('favorites_apartments', 'comparison_apartments')