65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
from rest_framework import serializers
|
|
from booking.models import models
|
|
|
|
|
|
class BookingSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = models.Booking
|
|
fields = (
|
|
'id',
|
|
'type',
|
|
)
|
|
|
|
|
|
class CheckBookingSerializer(serializers.ModelSerializer):
|
|
available = serializers.BooleanField()
|
|
type = serializers.ChoiceField(choices=models.Booking.AVAILABLE_SERVICES, allow_null=True)
|
|
details = serializers.DictField()
|
|
|
|
class Meta:
|
|
model = models.Booking
|
|
fields = (
|
|
'available',
|
|
'type',
|
|
'details',
|
|
)
|
|
|
|
|
|
class PendingBookingSerializer(serializers.ModelSerializer):
|
|
restaurant_id = serializers.CharField()
|
|
booking_id = serializers.CharField(allow_null=True, allow_blank=True)
|
|
id = serializers.ReadOnlyField()
|
|
user = serializers.ReadOnlyField()
|
|
|
|
class Meta:
|
|
model = models.Booking
|
|
fields = (
|
|
'id',
|
|
'type',
|
|
'restaurant_id',
|
|
'booking_id',
|
|
'pending_booking_id',
|
|
'user',
|
|
)
|
|
|
|
|
|
class UpdateBookingSerializer(serializers.ModelSerializer):
|
|
id = serializers.ReadOnlyField()
|
|
|
|
class Meta:
|
|
model = models.Booking
|
|
fields = ('booking_id', 'id')
|
|
|
|
|
|
class GetBookingSerializer(serializers.ModelSerializer):
|
|
details = serializers.SerializerMethodField()
|
|
|
|
def get_details(self, obj):
|
|
booking = self.instance
|
|
service = booking.get_service_by_type(booking.type)
|
|
return service.get_booking_details(booking.booking_id)
|
|
|
|
class Meta:
|
|
model = models.Booking
|
|
fields = '__all__'
|