From: Tollef Fog Heen <tfheen@err.no>
Date: Sat, 28 Jul 2012 09:11:21 +0000 (+0200)
Subject: Poor man's single page alerts
X-Git-Url: https://err.no/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=78470ec6e3df7eb23ce2366d5f6003f6ff5950d2;p=peojumk

Poor man's single page alerts
---

diff --git a/invoice/views.py b/invoice/views.py
index 9b9805b..64a0f55 100644
--- 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):
diff --git a/templates/base.html b/templates/base.html
index 1c28fe1..367e2c9 100644
--- a/templates/base.html
+++ b/templates/base.html
@@ -84,6 +84,16 @@
 	      </div>
 	    {% endfor %}
 	  {% endif %}
+	  {% if alert_error %}
+	    <div class="alert alert-error">
+	      {{ alert_error }}
+	    </div>
+	  {% endif %}
+	  {% if alert_warning %}
+	    <div class="alert alert-warning">
+	      {{ alert_warning }}
+	    </div>
+	  {% endif %}
 	{% endblock messages %}
 
 	  {% block pretitle %}{% endblock %}