thrillwiki_django_no_react/companies/views.py

from django.views.generic import DetailView, ListView, CreateView, UpdateView
from django.shortcuts import get_object_or_404
from django.urls import reverse
from django.contrib.auth.mixins import LoginRequiredMixin
from django.contrib.contenttypes.models import ContentType
from django.contrib import messages
from django.http import HttpResponseRedirect
from .models import Company, Manufacturer
from .forms import CompanyForm, ManufacturerForm
from rides.models import Ride
from parks.models import Park
from core.views import SlugRedirectMixin
from moderation.mixins import EditSubmissionMixin, PhotoSubmissionMixin, HistoryMixin
from moderation.models import EditSubmission

class CompanyCreateView(LoginRequiredMixin, CreateView):
    model = Company
    form_class = CompanyForm
    template_name = 'companies/company_form.html'

    def form_valid(self, form):
        cleaned_data = form.cleaned_data.copy()

        # Create submission record
        submission = EditSubmission.objects.create(
            user=self.request.user,
            content_type=ContentType.objects.get_for_model(Company),
            submission_type='CREATE',
            changes=cleaned_data,
            reason=self.request.POST.get('reason', ''),
            source=self.request.POST.get('source', '')
        )
        
        # If user is moderator or above, auto-approve
        if self.request.user.role in ['MODERATOR', 'ADMIN', 'SUPERUSER']:
            self.object = form.save()
            submission.object_id = self.object.id
            submission.status = 'APPROVED'
            submission.handled_by = self.request.user
            submission.save()
            messages.success(self.request, f'Successfully created {self.object.name}')
            return HttpResponseRedirect(self.get_success_url())
        
        messages.success(self.request, 'Your company submission has been sent for review')
        return HttpResponseRedirect(reverse('companies:company_list'))

    def get_success_url(self):
        return reverse('companies:company_detail', kwargs={'slug': self.object.slug})

class CompanyUpdateView(LoginRequiredMixin, UpdateView):
    model = Company
    form_class = CompanyForm
    template_name = 'companies/company_form.html'

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['is_edit'] = True
        return context

    def form_valid(self, form):
        cleaned_data = form.cleaned_data.copy()

        # Create submission record
        submission = EditSubmission.objects.create(
            user=self.request.user,
            content_type=ContentType.objects.get_for_model(Company),
            object_id=self.object.id,
            submission_type='EDIT',
            changes=cleaned_data,
            reason=self.request.POST.get('reason', ''),
            source=self.request.POST.get('source', '')
        )
        
        # If user is moderator or above, auto-approve
        if self.request.user.role in ['MODERATOR', 'ADMIN', 'SUPERUSER']:
            self.object = form.save()
            submission.status = 'APPROVED'
            submission.handled_by = self.request.user
            submission.save()
            messages.success(self.request, f'Successfully updated {self.object.name}')
            return HttpResponseRedirect(self.get_success_url())
        
        messages.success(self.request, f'Your changes to {self.object.name} have been sent for review')
        return HttpResponseRedirect(reverse('companies:company_detail', kwargs={'slug': self.object.slug}))

    def get_success_url(self):
        return reverse('companies:company_detail', kwargs={'slug': self.object.slug})

class ManufacturerCreateView(LoginRequiredMixin, CreateView):
    model = Manufacturer
    form_class = ManufacturerForm
    template_name = 'companies/manufacturer_form.html'

    def form_valid(self, form):
        cleaned_data = form.cleaned_data.copy()

        # Create submission record
        submission = EditSubmission.objects.create(
            user=self.request.user,
            content_type=ContentType.objects.get_for_model(Manufacturer),
            submission_type='CREATE',
            changes=cleaned_data,
            reason=self.request.POST.get('reason', ''),
            source=self.request.POST.get('source', '')
        )
        
        # If user is moderator or above, auto-approve
        if self.request.user.role in ['MODERATOR', 'ADMIN', 'SUPERUSER']:
            self.object = form.save()
            submission.object_id = self.object.id
            submission.status = 'APPROVED'
            submission.handled_by = self.request.user
            submission.save()
            messages.success(self.request, f'Successfully created {self.object.name}')
            return HttpResponseRedirect(self.get_success_url())
        
        messages.success(self.request, 'Your manufacturer submission has been sent for review')
        return HttpResponseRedirect(reverse('companies:manufacturer_list'))

    def get_success_url(self):
        return reverse('companies:manufacturer_detail', kwargs={'slug': self.object.slug})

class ManufacturerUpdateView(LoginRequiredMixin, UpdateView):
    model = Manufacturer
    form_class = ManufacturerForm
    template_name = 'companies/manufacturer_form.html'

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['is_edit'] = True
        return context

    def form_valid(self, form):
        cleaned_data = form.cleaned_data.copy()

        # Create submission record
        submission = EditSubmission.objects.create(
            user=self.request.user,
            content_type=ContentType.objects.get_for_model(Manufacturer),
            object_id=self.object.id,
            submission_type='EDIT',
            changes=cleaned_data,
            reason=self.request.POST.get('reason', ''),
            source=self.request.POST.get('source', '')
        )
        
        # If user is moderator or above, auto-approve
        if self.request.user.role in ['MODERATOR', 'ADMIN', 'SUPERUSER']:
            self.object = form.save()
            submission.status = 'APPROVED'
            submission.handled_by = self.request.user
            submission.save()
            messages.success(self.request, f'Successfully updated {self.object.name}')
            return HttpResponseRedirect(self.get_success_url())
        
        messages.success(self.request, f'Your changes to {self.object.name} have been sent for review')
        return HttpResponseRedirect(reverse('companies:manufacturer_detail', kwargs={'slug': self.object.slug}))

    def get_success_url(self):
        return reverse('companies:manufacturer_detail', kwargs={'slug': self.object.slug})

class CompanyDetailView(SlugRedirectMixin, EditSubmissionMixin, PhotoSubmissionMixin, HistoryMixin, 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, EditSubmissionMixin, PhotoSubmissionMixin, HistoryMixin, 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