88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
from django.utils.translation import gettext_lazy as _
|
|
from rest_framework import serializers
|
|
from apps.bookings.models import Booking, BookingStatus
|
|
from apps.bookings.services import validate_booking_request
|
|
from apps.salons.models import Service, StaffProfile
|
|
|
|
|
|
class BookingSerializer(serializers.ModelSerializer):
|
|
salon_name = serializers.CharField(source="salon.name", read_only=True)
|
|
service_name = serializers.CharField(source="service.name", read_only=True)
|
|
staff_name = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = Booking
|
|
fields = [
|
|
"id",
|
|
"salon",
|
|
"salon_name",
|
|
"service",
|
|
"service_name",
|
|
"staff",
|
|
"staff_name",
|
|
"start_time",
|
|
"end_time",
|
|
"status",
|
|
"price_amount",
|
|
"currency",
|
|
"notes",
|
|
"created_at",
|
|
]
|
|
read_only_fields = ["id", "salon", "price_amount", "currency", "created_at"]
|
|
|
|
def get_staff_name(self, obj):
|
|
if not obj.staff:
|
|
return None
|
|
first = obj.staff.user.first_name or ""
|
|
last = obj.staff.user.last_name or ""
|
|
return (first + " " + last).strip() or obj.staff.user.email
|
|
|
|
def validate(self, attrs):
|
|
if not self.instance or "status" not in attrs:
|
|
return attrs
|
|
|
|
new_status = attrs["status"]
|
|
old_status = self.instance.status
|
|
if new_status == old_status:
|
|
return attrs
|
|
|
|
user = self.context["request"].user
|
|
role = getattr(user, "role", None)
|
|
|
|
if new_status == BookingStatus.CONFIRMED and role not in {"manager", "staff", "admin"}:
|
|
raise serializers.ValidationError({"status": _("Only staff or managers can confirm bookings.")})
|
|
if new_status == BookingStatus.COMPLETED and role not in {"manager", "staff", "admin"}:
|
|
raise serializers.ValidationError({"status": _("Only staff or managers can complete bookings.")})
|
|
if new_status == BookingStatus.CANCELLED and role not in {"manager", "staff", "admin", "customer"}:
|
|
raise serializers.ValidationError({"status": _("You are not allowed to cancel this booking.")})
|
|
|
|
return attrs
|
|
|
|
|
|
class BookingCreateSerializer(serializers.ModelSerializer):
|
|
class Meta:
|
|
model = Booking
|
|
fields = ["service", "staff", "start_time", "end_time", "notes"]
|
|
extra_kwargs = {"staff": {"required": True}}
|
|
|
|
def validate(self, attrs):
|
|
service: Service = attrs["service"]
|
|
staff = attrs.get("staff")
|
|
validate_booking_request(service, staff, attrs["start_time"], attrs["end_time"])
|
|
return attrs
|
|
|
|
def create(self, validated_data):
|
|
request = self.context["request"]
|
|
service = validated_data["service"]
|
|
return Booking.objects.create(
|
|
salon=service.salon,
|
|
customer=request.user,
|
|
service=service,
|
|
staff=validated_data.get("staff"),
|
|
start_time=validated_data["start_time"],
|
|
end_time=validated_data["end_time"],
|
|
notes=validated_data.get("notes", ""),
|
|
price_amount=service.price_amount,
|
|
currency=service.currency,
|
|
)
|