first commit

companies/views.py

@@ -0,0 +1,105 @@
+from django.views.generic import DetailView, ListView
+from django.shortcuts import get_object_or_404
+from .models import Company, Manufacturer
+from rides.models import Ride
+from parks.models import Park
+from core.views import SlugRedirectMixin
+
+class CompanyDetailView(SlugRedirectMixin, DetailView):
+    model = Company
+    template_name = 'companies/company_detail.html'
+    context_object_name = 'company'
+
+    def get_object(self, queryset=None):
+        if queryset is None:
+            queryset = self.get_queryset()
+        slug = self.kwargs.get(self.slug_url_kwarg)
+        # Try to get by current or historical slug
+        return self.model.get_by_slug(slug)[0]
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context['parks'] = Park.objects.filter(
+            owner=self.object
+        ).select_related('owner')
+        return context
+
+    def get_redirect_url_pattern(self):
+        return 'company_detail'
+
+class ManufacturerDetailView(SlugRedirectMixin, DetailView):
+    model = Manufacturer
+    template_name = 'companies/manufacturer_detail.html'
+    context_object_name = 'manufacturer'
+
+    def get_object(self, queryset=None):
+        if queryset is None:
+            queryset = self.get_queryset()
+        slug = self.kwargs.get(self.slug_url_kwarg)
+        # Try to get by current or historical slug
+        return self.model.get_by_slug(slug)[0]
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        context['rides'] = Ride.objects.filter(
+            manufacturer=self.object
+        ).select_related('park', 'coaster_stats')
+        return context
+
+    def get_redirect_url_pattern(self):
+        return 'manufacturer_detail'
+
+class CompanyListView(ListView):
+    model = Company
+    template_name = 'companies/company_list.html'
+    context_object_name = 'companies'
+    paginate_by = 12
+
+    def get_queryset(self):
+        queryset = Company.objects.all()
+        
+        # Filter by country if specified
+        country = self.request.GET.get('country')
+        if country:
+            queryset = queryset.filter(headquarters__icontains=country)
+            
+        # Search by name if specified
+        search = self.request.GET.get('search')
+        if search:
+            queryset = queryset.filter(name__icontains=search)
+            
+        return queryset.order_by('name')
+
+class ManufacturerListView(ListView):
+    model = Manufacturer
+    template_name = 'companies/manufacturer_list.html'
+    context_object_name = 'manufacturers'
+    paginate_by = 12
+
+    def get_queryset(self):
+        queryset = Manufacturer.objects.all()
+        
+        # Filter by country if specified
+        country = self.request.GET.get('country')
+        if country:
+            queryset = queryset.filter(headquarters__icontains=country)
+            
+        # Search by name if specified
+        search = self.request.GET.get('search')
+        if search:
+            queryset = queryset.filter(name__icontains=search)
+            
+        return queryset.order_by('name')
+
+    def get_context_data(self, **kwargs):
+        context = super().get_context_data(**kwargs)
+        # Add stats for filtering
+        context['total_manufacturers'] = self.model.objects.count()
+        context['total_rides'] = Ride.objects.filter(
+            manufacturer__isnull=False
+        ).count()
+        context['total_roller_coasters'] = Ride.objects.filter(
+            manufacturer__isnull=False,
+            category='RC'
+        ).count()
+        return context