From 82fbe2e53e5f52d722d34fd6f36140f30a2ae564 Mon Sep 17 00:00:00 2001
From: Tollef Fog Heen <tfheen@err.no>
Date: Thu, 7 Jun 2012 17:36:41 +0200
Subject: [PATCH] Colour invoices on front page, add some links

---
 invoice/models.py            | 32 +++++++++++++++++++++++++++++++-
 invoice/urls.py              |  2 +-
 templates/invoice/index.html | 10 ++++++++++
 3 files changed, 42 insertions(+), 2 deletions(-)

diff --git a/invoice/models.py b/invoice/models.py
index 9dd8629..8e5a1ca 100644
--- a/invoice/models.py
+++ b/invoice/models.py
@@ -17,6 +17,19 @@ INVOICE_STATUS_CHOICES = (
     ('P', 'Paid'),
     )
 
+# Mapping from colours to bootstrap names
+BOOTSTRAP_COLOUR_MAP = {
+    'red': 'label-important',
+    'orange':'label-warning',
+    'green':'label-success',
+    'grey':'',
+    'blue':'label-info',
+    'black':'label-inverse',
+
+}
+
+from datetime import date
+
 class Invoice(models.Model):
     """An invoice"""
     customer = models.ForeignKey(Client)
@@ -28,9 +41,26 @@ class Invoice(models.Model):
 
     def _total_sum(self):
         return self.invoiceline_set.aggregate(total=Sum("amount"))["total"]
-        
     total_sum = property(_total_sum)
 
+    # represents how due the invoice is
+    def _bootstrap_colour(self):
+        delta = self.due_date - date.today() 
+        if delta.days > 10:
+            return BOOTSTRAP_COLOUR_MAP['green']
+        elif delta.days > 0:
+            return BOOTSTRAP_COLOUR_MAP['orange']
+        else:
+            return BOOTSTRAP_COLOUR_MAP['red']
+
+    def _due_label(self):
+        delta = self.due_date - date.today() 
+        return delta.days
+
+    bootstrap_label = property(_bootstrap_colour)
+    due_label = property(_due_label)
+
+
 class VAT(models.Model):
     """ Model representing different taxes to be used in Invoices"""
     name = models.CharField(_('name'), max_length=255)
diff --git a/invoice/urls.py b/invoice/urls.py
index 0ccdd54..f667fbd 100644
--- a/invoice/urls.py
+++ b/invoice/urls.py
@@ -14,5 +14,5 @@ urlpatterns = patterns('invoice.views',
                 url(r'^client/(?P<pk>\d+)/invoice_new$', CreateInvoice.as_view(), name='invoice_new_id'),
                 url(r'^accounts/login/$', django.contrib.auth.views.login),
                 url(r'^invoice/new$', CreateInvoice.as_view(), name="invoice_new"),
-                url(r'^invoice/(?P<pk>\d+)$', DetailInvoice.as_view()),
+                url(r'^invoice/(?P<pk>\d+)$', DetailInvoice.as_view(), name='invoice_view'),
 )
diff --git a/templates/invoice/index.html b/templates/invoice/index.html
index 4b268b8..432ec2f 100644
--- a/templates/invoice/index.html
+++ b/templates/invoice/index.html
@@ -7,12 +7,22 @@
 
 <table>
 <tr>
+  <th>Due in # days</th>
+  <th>Invoice #</td>
   <th>Date</td>
   <th>Customer</td>
   <th>Total sum</td>
 </tr>
 {% for i in new_invoices %}
 <tr>
+  <td>
+    <a href="{% url invoice:invoice_view i.id %}">
+      <span class="label {{ i.bootstrap_label }}">
+	{{ i.due_label }}
+      </span>
+    </a>
+  </td>
+  <td>{{ i.id }}</td>
   <td>{{ i.date }}</td>
   <td>{{ i.customer }}</td>
   <td>{{ i.currency }} {{ i.total_sum }}</td>
-- 
2.39.5