backend/passfinder/events/api/serializers.py

from drf_spectacular.types import OpenApiTypes
from drf_spectacular.utils import extend_schema_field
from rest_framework import serializers
from rest_framework.generics import get_object_or_404

from passfinder.events.models import (
    Hotel,
    HotelPhone,
    City,
    Event,
    BasePoint,
    Region,
    Restaurant,
    UserRoute,
    UserRouteDate,
)


class HotelPhoneSerializer(serializers.ModelSerializer):
    class Meta:
        model = HotelPhone
        exclude = ("hotel",)


class HotelSerializer(serializers.ModelSerializer):
    phones = HotelPhoneSerializer(many=True)
    source = serializers.CharField(source="parser_source")

    class Meta:
        model = Hotel
        exclude = ("oid", "parser_source")


class MuseumSerializer(serializers.ModelSerializer):
    phones = HotelPhoneSerializer(many=True)

    class Meta:
        model = Hotel
        exclude = ("oid",)


class EventSerializer(serializers.ModelSerializer):
    class Meta:
        model = Event
        fields = ("type", "title", "description", "city", "oid")


class PointSerializer(serializers.ModelSerializer):
    class Meta:
        model = BasePoint
        fields = ["oid", "title", "description", "location", "icon"]


class RouteSerializer(serializers.Serializer):
    name = serializers.CharField()
    date = serializers.DateField(allow_null=True)
    description = serializers.CharField()
    points = serializers.ListSerializer(child=PointSerializer())


class RouteInputSerializer(serializers.Serializer):
    date_from = serializers.DateField(required=False, allow_null=True)
    date_to = serializers.DateField(required=False, allow_null=True)
    city = serializers.CharField(
        min_length=24, max_length=24, required=False, allow_blank=True, allow_null=True
    )
    movement = serializers.ChoiceField(['walk', 'bike', 'scooter', 'auto'], required=False, allow_blank=True)
    stars = serializers.ListField(
        child=serializers.ChoiceField([1, 2, 3, 4, 5]),
        required=False,
        allow_empty=True,
        allow_null=True
    )
    what_to_see = serializers.ListField(
        child=serializers.ChoiceField(        
            [
                'attractions', 
                'museum', 
                'movie', 
                'concert', 
                'artwork', 
                'plays', 
                'shop', 
                'gallery', 
                'theme_park', 
                'viewpoint', 
                'zoo'
            ]
        ),
        required=False,
        allow_empty=True,
        allow_null=True
    )
    where_stay = serializers.ListField(
        child=serializers.ChoiceField([
            'hotel', 'apartment', 'hostel'
        ]),
        required=False,
        allow_empty=True,
        allow_null=True
    )
    where_eat = serializers.ListField(
        child=serializers.ChoiceField(['restaurant', 'bar', 'cafe']),
        required=False,
        allow_empty=True,
        allow_null=True
    )
    with_kids = serializers.BooleanField(required=False, allow_null=True)
    with_animals = serializers.BooleanField(required=False, allow_null=True)


class CitySerializer(serializers.ModelSerializer):
    class Meta:
        model = City
        fields = ["oid", "title"]


class RegionSerializer(serializers.ModelSerializer):
    cities = CitySerializer(many=True)

    class Meta:
        model = Region
        fields = ["oid", "title", "description_short", "cities"]


class InputRoutePointSerializer(serializers.Serializer):
    type = serializers.ChoiceField(choices=["point", "transition"])
    duration = serializers.IntegerField(min_value=0, required=True)

    # point
    point = serializers.CharField(
        min_length=24, max_length=24, required=False, allow_blank=True, allow_null=True
    )
    point_type = serializers.CharField(
        required=False, allow_blank=True, allow_null=True
    )

    # transition
    distance = serializers.FloatField(min_value=0, required=False, allow_null=True)

    def validate(self, data):
        if data["type"] == "point":
            if "point" not in data or not data["point"]:
                raise serializers.ValidationError("Point id is required")
            get_object_or_404(BasePoint, oid=data["point"])
            if "distance" not in data or not data["point_type"]:
                raise serializers.ValidationError("Point type is required")
        else:
            if "distance" not in data or not data["distance"]:
                raise serializers.ValidationError("Distance is required")

        return data


class InputRouteDateSerializer(serializers.Serializer):
    date = serializers.DateField()
    points = serializers.ListSerializer(child=InputRoutePointSerializer())


class InputRouteSerializer(serializers.Serializer):
    dates = serializers.ListSerializer(child=InputRouteDateSerializer())


class ListUserRouteSerializer(serializers.ModelSerializer):
    class Meta:
        model = UserRoute
        fields = ["id", "created"]


class UserRouteDateSerializer(serializers.ModelSerializer):
    points = serializers.SerializerMethodField(method_name="get_points")

    @extend_schema_field(InputRoutePointSerializer)
    def get_points(self, obj):
        return [x.get_json() for x in obj.points.all()]

    class Meta:
        model = UserRouteDate
        fields = ["date", "points"]


class UserRouteSerializer(serializers.ModelSerializer):
    class Meta:
        model = UserRoute
        fields = ["created", "dates"]


class RestaurantSerializer(serializers.ModelSerializer):
    class Meta:
        model = Restaurant
        exclude = ("phones",)


class ObjectRouteSerializer(serializers.Serializer):
    lat = serializers.FloatField()
    lon = serializers.FloatField()
    title = serializers.CharField()
    description = serializers.CharField()
    oid = serializers.CharField()