update

apps/rides/models/company.py

@@ -0,0 +1,96 @@
+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"