Poor man's single page alerts

[peojumk] / invoice / views.py

diff --git a/invoice/views.py b/invoice/views.py
index 9b9805bcba5a5c9fae587e1dea2dfc03fa5cb8c9..64a0f55201e67f7ab2a65d354af1d8528492313b 100644 (file)
--- a/invoice/views.py
+++ b/invoice/views.py
@@ -64,13 +64,13 @@ class InvoiceLineInline(InlineFormSet):
 class DetailInvoice(DetailView):
     model = Invoice
 
-    def get(self, request, *args, **kwargs):
-        i = get_object_or_404(Invoice, pk=kwargs["pk"])
-        if i.due_days < 0:
-            messages.error(request, "This invoice is overdue")
-        elif i.due_days < 3:
-            messages.warning(request, "This invoice is soon due")
-        return super(DetailInvoice, self).get(request, *args, **kwargs)
+    def get_context_data(self, *args, **kwargs):
+        context_data = super(DetailInvoice, self).get_context_data(*args, **kwargs)
+        if self.object.due_days < 0:
+            context_data.update({'alert_error': "This invoice is overdue" })
+        elif self.object.due_days < 3:
+            context_data.update({'alert_warning': "This invoice is soon due"})
+        return context_data
 
     @method_decorator(login_required)
     def dispatch(self, *args, **kwargs):