from django.db import models from django.urls import reverse from django.utils.text import slugify from django.contrib.contenttypes.fields import GenericRelation from django.core.exceptions import ValidationError from decimal import Decimal, ROUND_DOWN, InvalidOperation from simple_history.models import HistoricalRecords from companies.models import Company from media.models import Photo from history_tracking.models import HistoricalModel from location.models import Location class Park(HistoricalModel): STATUS_CHOICES = [ ("OPERATING", "Operating"), ("CLOSED_TEMP", "Temporarily Closed"), ("CLOSED_PERM", "Permanently Closed"), ("UNDER_CONSTRUCTION", "Under Construction"), ("DEMOLISHED", "Demolished"), ("RELOCATED", "Relocated"), ] name = models.CharField(max_length=255) slug = models.SlugField(max_length=255, unique=True) description = models.TextField(blank=True) status = models.CharField( max_length=20, choices=STATUS_CHOICES, default="OPERATING" ) # Location fields using GenericRelation location = GenericRelation(Location, related_query_name='park') # Details opening_date = models.DateField(null=True, blank=True) closing_date = models.DateField(null=True, blank=True) operating_season = models.CharField(max_length=255, blank=True) size_acres = models.DecimalField( max_digits=10, decimal_places=2, null=True, blank=True ) website = models.URLField(blank=True) # Statistics average_rating = models.DecimalField( max_digits=3, decimal_places=2, null=True, blank=True ) total_rides = models.IntegerField(null=True, blank=True) total_roller_coasters = models.IntegerField(null=True, blank=True) # Relationships owner = models.ForeignKey( Company, on_delete=models.SET_NULL, null=True, blank=True, related_name="parks" ) photos = GenericRelation(Photo, related_query_name="park") # Metadata created_at = models.DateTimeField(auto_now_add=True, null=True) updated_at = models.DateTimeField(auto_now=True) history = HistoricalRecords() class Meta: ordering = ["name"] def __str__(self): return self.name def save(self, *args, **kwargs): if not self.slug: self.slug = slugify(self.name) super().save(*args, **kwargs) def get_absolute_url(self): return reverse("parks:park_detail", kwargs={"slug": self.slug}) @property def formatted_location(self): if self.location.exists(): location = self.location.first() return location.get_formatted_address() return "" @property def coordinates(self): """Returns coordinates as a tuple (latitude, longitude)""" if self.location.exists(): location = self.location.first() return location.coordinates return None @classmethod def get_by_slug(cls, slug): """Get park by current or historical slug""" try: return cls.objects.get(slug=slug), False except cls.DoesNotExist: # Check historical slugs history = cls.history.filter(slug=slug).order_by("-history_date").first() if history: try: return cls.objects.get(id=history.id), True except cls.DoesNotExist: pass raise cls.DoesNotExist() class ParkArea(HistoricalModel): park = models.ForeignKey(Park, on_delete=models.CASCADE, related_name="areas") name = models.CharField(max_length=255) slug = models.SlugField(max_length=255) description = models.TextField(blank=True) opening_date = models.DateField(null=True, blank=True) closing_date = models.DateField(null=True, blank=True) # Metadata created_at = models.DateTimeField(auto_now_add=True, null=True) updated_at = models.DateTimeField(auto_now=True) history = HistoricalRecords() class Meta: ordering = ["name"] unique_together = ["park", "slug"] def __str__(self): return f"{self.name} at {self.park.name}" def save(self, *args, **kwargs): if not self.slug: self.slug = slugify(self.name) super().save(*args, **kwargs) def get_absolute_url(self): return reverse( "parks:area_detail", kwargs={"park_slug": self.park.slug, "area_slug": self.slug}, ) @classmethod def get_by_slug(cls, slug): """Get area by current or historical slug""" try: return cls.objects.get(slug=slug), False except cls.DoesNotExist: # Check historical slugs history = cls.history.filter(slug=slug).order_by("-history_date").first() if history: try: return cls.objects.get(id=history.id), True except cls.DoesNotExist: pass raise cls.DoesNotExist()