Added audit logs, and removed DeleteDriver as that shouldn't happen

1 files changed, 4 insertions, 10 deletions

diff --git a/app/dispatch/views.py b/app/dispatch/views.py
index 904639e..b5f3e0d 100644
--- a/app/dispatch/views.py
+++ b/app/dispatch/views.py
@@ -93,7 +93,7 @@ class DriverDetail(LoadDateSort):
     def test_func(self):
         return self.request.user.is_superuser
 
-class DriverUpdate(UpdateView):
+class DriverUpdate(UpdateView): 
     template_name = "dispatch/drivers/edit.html"
     model = User
     success_url = reverse_lazy('driver_list')
@@ -102,14 +102,6 @@ class DriverUpdate(UpdateView):
     def test_func(self):
         return self.request.user.is_superuser
 
-class DriverDelete(DeleteView):
-    template_name = "dispatch/loads/delete.html"
-    model = User
-    success_url = reverse_lazy('driver_list')
-
-    def test_func(self):
-        return self.request.user.is_superuser
-
 # Company CRUD
 
 class CompanyList(ListView):
@@ -136,7 +128,9 @@ class CompanyDelete(DeleteView):
     template_name = "dispatch/loads/delete.html"
     model = Company
     success_url = reverse_lazy('company_list')
-
+    
+    def test_func(self):
+        return self.request.user.is_superuser
 # Load CRUD
 
 class LoadList(FilteredListView):