mirror of
https://github.com/pacnpal/thrillwiki_django_no_react.git
synced 2025-12-20 08:11:08 -05:00
112 lines
3.9 KiB
Python
112 lines
3.9 KiB
Python
from django.db import models
|
|
from django.utils.text import slugify
|
|
from django.urls import reverse
|
|
from typing import Tuple, Optional, ClassVar, TYPE_CHECKING
|
|
import pghistory
|
|
from history_tracking.models import TrackedModel, HistoricalSlug
|
|
|
|
@pghistory.track()
|
|
class Company(TrackedModel):
|
|
name = models.CharField(max_length=255)
|
|
slug = models.SlugField(max_length=255, unique=True)
|
|
website = models.URLField(blank=True)
|
|
headquarters = models.CharField(max_length=255, blank=True)
|
|
description = models.TextField(blank=True)
|
|
total_parks = models.IntegerField(default=0)
|
|
total_rides = models.IntegerField(default=0)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
objects: ClassVar[models.Manager['Company']]
|
|
|
|
class Meta:
|
|
verbose_name_plural = 'companies'
|
|
ordering = ['name']
|
|
|
|
def __str__(self) -> str:
|
|
return self.name
|
|
|
|
def save(self, *args, **kwargs) -> None:
|
|
if not self.slug:
|
|
self.slug = slugify(self.name)
|
|
super().save(*args, **kwargs)
|
|
|
|
@classmethod
|
|
def get_by_slug(cls, slug: str) -> Tuple['Company', bool]:
|
|
"""Get company by slug, checking historical slugs if needed"""
|
|
try:
|
|
return cls.objects.get(slug=slug), False
|
|
except cls.DoesNotExist:
|
|
# Check pghistory first
|
|
history_model = cls.get_history_model()
|
|
history_entry = (
|
|
history_model.objects.filter(slug=slug)
|
|
.order_by('-pgh_created_at')
|
|
.first()
|
|
)
|
|
|
|
if history_entry:
|
|
return cls.objects.get(id=history_entry.pgh_obj_id), True
|
|
|
|
# Check manual slug history as fallback
|
|
try:
|
|
historical = HistoricalSlug.objects.get(
|
|
content_type__model='company',
|
|
slug=slug
|
|
)
|
|
return cls.objects.get(pk=historical.object_id), True
|
|
except (HistoricalSlug.DoesNotExist, cls.DoesNotExist):
|
|
raise cls.DoesNotExist()
|
|
|
|
@pghistory.track()
|
|
class Manufacturer(TrackedModel):
|
|
name = models.CharField(max_length=255)
|
|
slug = models.SlugField(max_length=255, unique=True)
|
|
website = models.URLField(blank=True)
|
|
headquarters = models.CharField(max_length=255, blank=True)
|
|
description = models.TextField(blank=True)
|
|
total_rides = models.IntegerField(default=0)
|
|
total_roller_coasters = models.IntegerField(default=0)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
objects: ClassVar[models.Manager['Manufacturer']]
|
|
|
|
class Meta:
|
|
ordering = ['name']
|
|
|
|
def __str__(self) -> str:
|
|
return self.name
|
|
|
|
def save(self, *args, **kwargs) -> None:
|
|
if not self.slug:
|
|
self.slug = slugify(self.name)
|
|
super().save(*args, **kwargs)
|
|
|
|
@classmethod
|
|
def get_by_slug(cls, slug: str) -> Tuple['Manufacturer', bool]:
|
|
"""Get manufacturer by slug, checking historical slugs if needed"""
|
|
try:
|
|
return cls.objects.get(slug=slug), False
|
|
except cls.DoesNotExist:
|
|
# Check pghistory first
|
|
history_model = cls.get_history_model()
|
|
history_entry = (
|
|
history_model.objects.filter(slug=slug)
|
|
.order_by('-pgh_created_at')
|
|
.first()
|
|
)
|
|
|
|
if history_entry:
|
|
return cls.objects.get(id=history_entry.pgh_obj_id), True
|
|
|
|
# Check manual slug history as fallback
|
|
try:
|
|
historical = HistoricalSlug.objects.get(
|
|
content_type__model='manufacturer',
|
|
slug=slug
|
|
)
|
|
return cls.objects.get(pk=historical.object_id), True
|
|
except (HistoricalSlug.DoesNotExist, cls.DoesNotExist):
|
|
raise cls.DoesNotExist()
|