mirror of
https://github.com/task-17-lct/backend.git
synced 2024-11-15 01:46:36 +03:00
109 lines
3.5 KiB
Python
109 lines
3.5 KiB
Python
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
|
|
|
|
|
|
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)
|
|
region = serializers.CharField(
|
|
min_length=24, max_length=24, required=False, allow_blank=True, 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"])
|
|
time = 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
|
|
)
|
|
|
|
# transition
|
|
point_from = serializers.CharField(
|
|
min_length=24, max_length=24, required=False, allow_blank=True, allow_null=True
|
|
)
|
|
point_to = serializers.CharField(
|
|
min_length=24, max_length=24, required=False, allow_blank=True, allow_null=True
|
|
)
|
|
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"])
|
|
else:
|
|
if "point_to" not in data or not data["point_to"]:
|
|
raise serializers.ValidationError("Point to id is required")
|
|
get_object_or_404(BasePoint, oid=data["point_to"])
|
|
if "point_from" not in data or not data["point_from"]:
|
|
raise serializers.ValidationError("Point from id is required")
|
|
get_object_or_404(BasePoint, oid=data["point_from"])
|
|
if "distance" not in data or not data["distance"]:
|
|
raise serializers.ValidationError("Distance is required")
|
|
|
|
return data
|
|
|
|
|
|
class InputRouteSerializer(serializers.Serializer):
|
|
points = serializers.ListSerializer(child=InputRoutePointSerializer())
|