mohd
411180e312
Added a staff availability model and migration, a booking validation service, and serializer enforcement.
99 lines
3.5 KiB
Python
99 lines
3.5 KiB
Python
from django.conf import settings
|
|
from django.db import models
|
|
|
|
|
|
class Salon(models.Model):
|
|
owner = models.ForeignKey(
|
|
settings.AUTH_USER_MODEL,
|
|
on_delete=models.CASCADE,
|
|
related_name="owned_salons",
|
|
)
|
|
name = models.CharField(max_length=200)
|
|
description = models.TextField(blank=True)
|
|
address = models.CharField(max_length=255)
|
|
city = models.CharField(max_length=100)
|
|
latitude = models.DecimalField(max_digits=9, decimal_places=6, null=True, blank=True)
|
|
longitude = models.DecimalField(max_digits=9, decimal_places=6, null=True, blank=True)
|
|
phone_number = models.CharField(max_length=20, blank=True)
|
|
email = models.EmailField(blank=True)
|
|
website = models.URLField(blank=True)
|
|
rating_avg = models.DecimalField(max_digits=3, decimal_places=2, default=0)
|
|
rating_count = models.PositiveIntegerField(default=0)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class SalonPhoto(models.Model):
|
|
salon = models.ForeignKey(Salon, on_delete=models.CASCADE, related_name="photos")
|
|
image_url = models.URLField()
|
|
alt_text = models.CharField(max_length=200, blank=True)
|
|
sort_order = models.PositiveIntegerField(default=0)
|
|
|
|
def __str__(self):
|
|
return f"{self.salon.name} photo"
|
|
|
|
|
|
class Service(models.Model):
|
|
salon = models.ForeignKey(Salon, on_delete=models.CASCADE, related_name="services")
|
|
name = models.CharField(max_length=200)
|
|
description = models.TextField(blank=True)
|
|
duration_minutes = models.PositiveIntegerField()
|
|
price_amount = models.DecimalField(max_digits=10, decimal_places=2)
|
|
currency = models.CharField(max_length=10, default=getattr(settings, "DEFAULT_CURRENCY", "SAR"))
|
|
is_active = models.BooleanField(default=True)
|
|
|
|
def __str__(self):
|
|
return f"{self.name} - {self.salon.name}"
|
|
|
|
|
|
class StaffProfile(models.Model):
|
|
user = models.OneToOneField(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)
|
|
salon = models.ForeignKey(Salon, on_delete=models.CASCADE, related_name="staff")
|
|
title = models.CharField(max_length=200, blank=True)
|
|
bio = models.TextField(blank=True)
|
|
is_active = models.BooleanField(default=True)
|
|
|
|
def __str__(self):
|
|
return f"{self.user.email} - {self.salon.name}"
|
|
|
|
|
|
class StaffAvailability(models.Model):
|
|
DAY_CHOICES = [
|
|
(0, "Monday"),
|
|
(1, "Tuesday"),
|
|
(2, "Wednesday"),
|
|
(3, "Thursday"),
|
|
(4, "Friday"),
|
|
(5, "Saturday"),
|
|
(6, "Sunday"),
|
|
]
|
|
|
|
staff = models.ForeignKey(
|
|
StaffProfile,
|
|
on_delete=models.CASCADE,
|
|
related_name="availability",
|
|
)
|
|
day_of_week = models.PositiveSmallIntegerField(choices=DAY_CHOICES)
|
|
start_time = models.TimeField()
|
|
end_time = models.TimeField()
|
|
is_active = models.BooleanField(default=True)
|
|
|
|
class Meta:
|
|
ordering = ["staff_id", "day_of_week", "start_time"]
|
|
|
|
def __str__(self):
|
|
return f"{self.staff.user.email} {self.get_day_of_week_display()} {self.start_time}-{self.end_time}"
|
|
|
|
|
|
class Review(models.Model):
|
|
salon = models.ForeignKey(Salon, on_delete=models.CASCADE, related_name="reviews")
|
|
customer = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, related_name="reviews")
|
|
rating = models.PositiveSmallIntegerField()
|
|
comment = models.TextField(blank=True)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return f"Review {self.rating} for {self.salon.name}"
|