backend/passfinder/recomendations/api/serializers.py

from rest_framework import serializers
from passfinder.events.api.serializers import (
    EventSerializer,
    HotelSerializer,
    ObjectRouteSerializer,
)


class TinderProceedSerializer(serializers.Serializer):
    action = serializers.ChoiceField(["left", "right"], write_only=True)
    event = EventSerializer(read_only=True)


class AddToPreferenceSerializer(serializers.Serializer):
    oid = serializers.CharField(write_only=True)


class EventOnboardingRetrieve(serializers.Serializer):
    events = serializers.ListField(child=EventSerializer(), read_only=True)
    types = serializers.ListField(
        child=serializers.ChoiceField(["park", "monument", "museum", "unseco"]),
        write_only=True,
    )


class HotelOnboardingRetrieve(serializers.Serializer):
    stars = serializers.ListField(
        child=serializers.ChoiceField([1, 2, 3, 4, 5]), write_only=True
    )
    hotels = serializers.ListField(child=HotelSerializer(), read_only=True)


class TinderGetEventFilterSerializer(serializers.Serializer):
    type = serializers.ListField(
        child=serializers.ChoiceField(
            ["attraction", "museum", "movie", "play", "concert"]
        )
    )
    event = EventSerializer()


class DailySelectionNodeSerializer(serializers.Serializer):
    action = serializers.ChoiceField(["left", "right"])
    oid = serializers.CharField()


class DailySelectionSerializerInput(serializers.Serializer):
    city = serializers.CharField(read_only=True)
    events = serializers.ListField(child=EventSerializer(), read_only=True)


class DailySelectionSerializer(serializers.Serializer):
    nodes = serializers.ListField(child=DailySelectionNodeSerializer(), write_only=True)
    type = serializers.ListField(
        child=serializers.ChoiceField(
            ["attraction", "museum", "movie", "play", "concert"]
        )
    )
    event = EventSerializer()


class StarSelectionSerializer(serializers.Serializer):
    stars = serializers.ListField(child=serializers.IntegerField(), write_only=True)


class CategorySelectionSerializer(serializers.Serializer):
    categories = serializers.ListField(
        child=serializers.ChoiceField(
            [
                "attractions",
                "museum",
                "movie",
                "concert",
                "artwork",
                "plays",
                "shop",
                "gallery",
                "theme_park",
                "viewpoint",
                "zoo",
            ]
        )
    )


class RecomendationNode(serializers.Serializer):
    category = serializers.CharField()
    events = serializers.ListField(child=ObjectRouteSerializer())


class SelfRecomendationSerializer(serializers.Serializer):
    recomendations = serializers.ListField(child=RecomendationNode(), write_only=True)