mirror of
https://github.com/pacnpal/thrillwiki_django_no_react.git
synced 2025-12-20 22:51:09 -05:00
66ed4347a9
- Cleaned up and standardized assertions in ApiTestMixin for API response validation. - Updated ASGI settings to use os.environ for setting the DJANGO_SETTINGS_MODULE. - Removed unused imports and improved formatting in settings.py. - Refactored URL patterns in urls.py for better readability and organization. - Enhanced view functions in views.py for consistency and clarity. - Added .flake8 configuration for linting and style enforcement. - Introduced type stubs for django-environ to improve type checking with Pylance.
30 lines
969 B
Python
30 lines
969 B
Python
from django.core.management.base import BaseCommand
|
|
from django.db.models import 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")
|
|
)
|