from rest_framework import serializers

<<<<<<< HEAD
from .models import Apartament, User, PsychTestAnswers
=======
from .models import Apartament, User, AuthToken
>>>>>>> 4f5f817d835f6a53448e7d60bb7591c24c825e96


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')

class PsychTestReultsSerializer(serializers.ModelSerializer):
    class Meta:
        model = PsychTestAnswers
        fields = "__all__"

class UserSerializer(serializers.ModelSerializer):
    class Meta:
        model = User
        fields = "__all__"
class TokenSerializer(serializers.ModelSerializer):
    class Meta:
        model = AuthToken
        fields = '__all__'