mirror of
https://github.com/pacnpal/thrillwiki_django_no_react.git
synced 2025-12-20 17:31:09 -05:00
99 lines
3.6 KiB
Python
99 lines
3.6 KiB
Python
from django.views.generic import DetailView, ListView
|
|
from django.shortcuts import get_object_or_404
|
|
from django.core.serializers.json import DjangoJSONEncoder
|
|
from django.urls import reverse
|
|
from django.db.models import Q
|
|
from .models import Ride, RollerCoasterStats
|
|
from parks.models import Park
|
|
from core.views import SlugRedirectMixin
|
|
|
|
class RideDetailView(SlugRedirectMixin, DetailView):
|
|
model = Ride
|
|
template_name = 'rides/ride_detail.html'
|
|
context_object_name = 'ride'
|
|
slug_url_kwarg = 'ride_slug'
|
|
|
|
def get_object(self, queryset=None):
|
|
if queryset is None:
|
|
queryset = self.get_queryset()
|
|
park_slug = self.kwargs.get('park_slug')
|
|
ride_slug = self.kwargs.get('ride_slug')
|
|
# Try to get by current or historical slug
|
|
obj, is_old_slug = self.model.get_by_slug(ride_slug)
|
|
if obj.park.slug != park_slug:
|
|
raise self.model.DoesNotExist("Park slug doesn't match")
|
|
return obj
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
if self.object.category == 'RC':
|
|
context['coaster_stats'] = RollerCoasterStats.objects.filter(ride=self.object).first()
|
|
return context
|
|
|
|
def get_redirect_url_pattern(self):
|
|
return 'ride_detail'
|
|
|
|
def get_redirect_url_kwargs(self):
|
|
return {
|
|
'park_slug': self.object.park.slug,
|
|
'ride_slug': self.object.slug
|
|
}
|
|
|
|
class RideListView(ListView):
|
|
model = Ride
|
|
template_name = 'rides/ride_list.html'
|
|
context_object_name = 'rides'
|
|
|
|
def get_queryset(self):
|
|
queryset = Ride.objects.select_related('park', 'coaster_stats', 'manufacturer').prefetch_related('photos')
|
|
|
|
search = self.request.GET.get('search', '').strip() or None
|
|
category = self.request.GET.get('category', '').strip() or None
|
|
status = self.request.GET.get('status', '').strip() or None
|
|
manufacturer = self.request.GET.get('manufacturer', '').strip() or None
|
|
|
|
if search:
|
|
queryset = queryset.filter(
|
|
Q(name__icontains=search) |
|
|
Q(park__name__icontains=search)
|
|
)
|
|
if category:
|
|
queryset = queryset.filter(category=category)
|
|
if status:
|
|
queryset = queryset.filter(status=status)
|
|
if manufacturer:
|
|
# Example: Exclude null manufacturers in RideListView
|
|
queryset = Ride.objects.exclude(manufacturer__isnull=True)
|
|
|
|
return queryset
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
|
|
# Exclude cases where manufacturer is null or an empty string
|
|
context['manufacturers'] = list(
|
|
Ride.objects.exclude(manufacturer__isnull=True)
|
|
.exclude(manufacturer__isnull=True)
|
|
.values_list('manufacturer__name', flat=True)
|
|
.distinct().order_by('manufacturer__name')
|
|
)
|
|
|
|
# Add current filter values to context
|
|
context['current_filters'] = {
|
|
'search': self.request.GET.get('search', ''),
|
|
'category': self.request.GET.get('category', ''),
|
|
'status': self.request.GET.get('status', ''),
|
|
'manufacturer': self.request.GET.get('manufacturer', '')
|
|
}
|
|
|
|
return context
|
|
|
|
|
|
|
|
def get(self, request, *args, **kwargs):
|
|
# Check if this is an HTMX request
|
|
if request.htmx:
|
|
# If it is, return just the rides list partial
|
|
self.template_name = 'rides/partials/ride_list.html'
|
|
return super().get(request, *args, **kwargs)
|