mirror of
https://github.com/pacnpal/thrillwiki_django_no_react.git
synced 2025-12-20 17:31:09 -05:00
22 lines
730 B
Python
22 lines
730 B
Python
from django.db.models import QuerySet, Count, Q
|
|
from .models import Park
|
|
|
|
def get_base_park_queryset() -> QuerySet[Park]:
|
|
"""Get base queryset with all needed annotations and prefetches"""
|
|
from django.contrib.contenttypes.models import ContentType
|
|
|
|
park_type = ContentType.objects.get_for_model(Park)
|
|
return (
|
|
Park.objects.select_related('owner')
|
|
.prefetch_related(
|
|
'photos',
|
|
'rides',
|
|
'location',
|
|
'location__content_type'
|
|
)
|
|
.annotate(
|
|
current_ride_count=Count('rides', distinct=True),
|
|
current_coaster_count=Count('rides', filter=Q(rides__category="RC"), distinct=True)
|
|
)
|
|
.order_by('name')
|
|
) |