mirror of
https://github.com/pacnpal/thrillwiki_django_no_react.git
synced 2025-12-20 11:51:10 -05:00
97 lines
3.5 KiB
Python
97 lines
3.5 KiB
Python
import pghistory
|
|
from django.contrib.postgres.fields import ArrayField
|
|
from django.db import models
|
|
from django.urls import reverse
|
|
from django.utils.text import slugify
|
|
from django.conf import settings
|
|
|
|
from apps.core.history import HistoricalSlug
|
|
from apps.core.models import TrackedModel
|
|
from apps.core.choices.fields import RichChoiceField
|
|
|
|
|
|
@pghistory.track()
|
|
class Company(TrackedModel):
|
|
name = models.CharField(max_length=255)
|
|
slug = models.SlugField(max_length=255, unique=True)
|
|
roles = ArrayField(
|
|
RichChoiceField(choice_group="company_roles", domain="rides", max_length=20),
|
|
default=list,
|
|
blank=True,
|
|
)
|
|
description = models.TextField(blank=True)
|
|
website = models.URLField(blank=True)
|
|
|
|
# General company info
|
|
founded_date = models.DateField(null=True, blank=True)
|
|
|
|
# Manufacturer-specific fields
|
|
rides_count = models.IntegerField(default=0)
|
|
coasters_count = models.IntegerField(default=0)
|
|
|
|
# Frontend URL
|
|
url = models.URLField(blank=True, help_text="Frontend URL for this company")
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def save(self, *args, **kwargs):
|
|
if not self.slug:
|
|
self.slug = slugify(self.name)
|
|
|
|
# Generate frontend URL based on primary role
|
|
# CRITICAL: Only MANUFACTURER and DESIGNER are for rides domain
|
|
# OPERATOR and PROPERTY_OWNER are for parks domain and handled separately
|
|
if self.roles:
|
|
frontend_domain = getattr(
|
|
settings, "FRONTEND_DOMAIN", "https://thrillwiki.com"
|
|
)
|
|
primary_role = self.roles[0] # Use first role as primary
|
|
|
|
if primary_role == "MANUFACTURER":
|
|
self.url = f"{frontend_domain}/rides/manufacturers/{self.slug}/"
|
|
elif primary_role == "DESIGNER":
|
|
self.url = f"{frontend_domain}/rides/designers/{self.slug}/"
|
|
# OPERATOR and PROPERTY_OWNER URLs are handled by parks domain, not here
|
|
|
|
super().save(*args, **kwargs)
|
|
|
|
def get_absolute_url(self):
|
|
# This will need to be updated to handle different roles
|
|
return reverse("companies:detail", kwargs={"slug": self.slug})
|
|
|
|
@classmethod
|
|
def get_by_slug(cls, slug):
|
|
"""Get company by current or historical slug"""
|
|
try:
|
|
return cls.objects.get(slug=slug), False
|
|
except cls.DoesNotExist:
|
|
# Check pghistory first
|
|
try:
|
|
from django.apps import apps
|
|
history_model = apps.get_model('rides', f'{cls.__name__}Event')
|
|
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
|
|
except LookupError:
|
|
# History model doesn't exist, skip pghistory check
|
|
pass
|
|
|
|
# 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("No company found with this slug")
|
|
|
|
class Meta(TrackedModel.Meta):
|
|
app_label = "rides"
|
|
ordering = ["name"]
|
|
verbose_name_plural = "Companies"
|