diff options
Diffstat (limited to 'app/dispatch/views.py')
| -rw-r--r-- | app/dispatch/views.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/app/dispatch/views.py b/app/dispatch/views.py index 63cc1aa..95aff28 100644 --- a/app/dispatch/views.py +++ b/app/dispatch/views.py @@ -151,7 +151,6 @@ class DriverSummary(UserPassesTestMixin, LoadDateSort): class DriverUpdate(UserPassesTestMixin, UpdateView): template_name = "dispatch/drivers/edit.html" model = User - success_url = reverse_lazy('driver_list') # fields = ['username', 'first_name','last_name','email','groups'] fields = [] @@ -174,8 +173,10 @@ class DriverUpdate(UserPassesTestMixin, UpdateView): def test_func(self): # Seems a little hacky at first but it works! - return self.request.user.is_superuser or \ + test = self.request.user.is_superuser or \ self.get_object().pk is self.request.user.id + print(test) + return test # Customer CRUD @@ -434,7 +435,7 @@ class IdentityUpdate(UserPassesTestMixin, UpdateView): model = Identity fields = [] - defualt_fields = ['name', 'address', 'city', 'state', 'zip_code'] + default_fields = ['name', 'address', 'city', 'state', 'zip_code'] superuser_fields = ['user', 'name', 'address', 'city', 'state', 'zip_code'] def get(self, request, *args, **kwargs): @@ -454,6 +455,12 @@ class InvoiceList(FilteredListView): template_name = "dispatch/invoice/list.html" model = Invoice + def get_queryset(self): + # TODO: allow for a pagination + base_qs = super(InvoiceList, self).get_queryset() + # Give me the newest ones first + return base_qs.order_by('-pk') + class InvoiceDetail(DetailView): template_name = "dispatch/invoice/detail.html" |