weird header stuff

parks/management/commands/update_park_counts.py

@@ -0,0 +1,34 @@
+from django.core.management.base import BaseCommand
+from django.db.models import Count, Q
+from parks.models import Park
+
+class Command(BaseCommand):
+    help = 'Update total_rides and total_roller_coasters counts for all parks'
+
+    def handle(self, *args, **options):
+        parks = Park.objects.all()
+        operating_rides = Q(status='OPERATING')
+        updated = 0
+
+        for park in parks:
+            # Count total operating rides
+            total_rides = park.rides.filter(operating_rides).count()
+            
+            # Count total operating roller coasters
+            total_coasters = park.rides.filter(
+                operating_rides,
+                category='RC'
+            ).count()
+            
+            # Update park counts
+            Park.objects.filter(id=park.id).update(
+                total_rides=total_rides,
+                total_roller_coasters=total_coasters
+            )
+            updated += 1
+
+        self.stdout.write(
+            self.style.SUCCESS(
+                f'Successfully updated counts for {updated} parks'
+            )
+        )