okay fine

companies/signals.py

@@ -1,5 +1,6 @@
 from django.db.models.signals import post_save, post_delete
 from django.dispatch import receiver
+from django.db.utils import ProgrammingError
 from parks.models import Park
 from rides.models import Ride
 from .models import Company, Manufacturer
@@ -8,34 +9,47 @@ from .models import Company, Manufacturer
 def update_company_stats(sender, instance, **kwargs):
     """Update company statistics when a park is added, modified, or deleted."""
     if instance.owner:
-        # Update total parks
-        total_parks = Park.objects.filter(owner=instance.owner).count()
-        total_rides = Ride.objects.filter(park__owner=instance.owner).count()
-        
-        Company.objects.filter(id=instance.owner.id).update(
-            total_parks=total_parks,
-            total_rides=total_rides
-        )
+        try:
+            # Update total parks
+            total_parks = Park.objects.filter(owner=instance.owner).count()
+            total_rides = Ride.objects.filter(park__owner=instance.owner).count()
+            
+            Company.objects.filter(id=instance.owner.id).update(
+                total_parks=total_parks,
+                total_rides=total_rides
+            )
+        except ProgrammingError:
+            # If rides table doesn't exist yet, just update parks count
+            total_parks = Park.objects.filter(owner=instance.owner).count()
+            Company.objects.filter(id=instance.owner.id).update(
+                total_parks=total_parks
+            )
 
 @receiver([post_save, post_delete], sender=Ride)
 def update_manufacturer_stats(sender, instance, **kwargs):
     """Update manufacturer statistics when a ride is added, modified, or deleted."""
     if instance.manufacturer:
-        # Update total rides and roller coasters
-        total_rides = Ride.objects.filter(manufacturer=instance.manufacturer).count()
-        total_roller_coasters = Ride.objects.filter(
-            manufacturer=instance.manufacturer,
-            category='RC'
-        ).count()
-        
-        Manufacturer.objects.filter(id=instance.manufacturer.id).update(
-            total_rides=total_rides,
-            total_roller_coasters=total_roller_coasters
-        )
+        try:
+            # Update total rides and roller coasters
+            total_rides = Ride.objects.filter(manufacturer=instance.manufacturer).count()
+            total_roller_coasters = Ride.objects.filter(
+                manufacturer=instance.manufacturer,
+                category='RC'
+            ).count()
+            
+            Manufacturer.objects.filter(id=instance.manufacturer.id).update(
+                total_rides=total_rides,
+                total_roller_coasters=total_roller_coasters
+            )
+        except ProgrammingError:
+            pass  # Skip if rides table doesn't exist yet
 
 @receiver(post_save, sender=Ride)
 def update_company_ride_stats(sender, instance, **kwargs):
     """Update company ride statistics when a ride is added or modified."""
     if instance.park and instance.park.owner:
-        total_rides = Ride.objects.filter(park__owner=instance.park.owner).count()
-        Company.objects.filter(id=instance.park.owner.id).update(total_rides=total_rides)
+        try:
+            total_rides = Ride.objects.filter(park__owner=instance.park.owner).count()
+            Company.objects.filter(id=instance.park.owner.id).update(total_rides=total_rides)
+        except ProgrammingError:
+            pass  # Skip if rides table doesn't exist yet