here we go

companies/views.py

@@ -1,11 +1,165 @@
-from django.views.generic import DetailView, ListView
+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 CompanyDetailView(SlugRedirectMixin, DetailView):
+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'
@@ -27,7 +181,7 @@ class CompanyDetailView(SlugRedirectMixin, DetailView):
     def get_redirect_url_pattern(self):
         return 'company_detail'
 
-class ManufacturerDetailView(SlugRedirectMixin, DetailView):
+class ManufacturerDetailView(SlugRedirectMixin, EditSubmissionMixin, PhotoSubmissionMixin, HistoryMixin, DetailView):
     model = Manufacturer
     template_name = 'companies/manufacturer_detail.html'
     context_object_name = 'manufacturer'