Add the ability to download all of the paperwork from a customer view

1 files changed, 17 insertions, 1 deletions

diff --git a/app/dispatch/models.py b/app/dispatch/models.py
index 58d47fb..b0f76e5 100644
--- a/app/dispatch/models.py
+++ b/app/dispatch/models.py
@@ -5,7 +5,8 @@ from auditlog.models import AuditlogHistoryField
 # from django.contrib.auth import get_user_model
 from datetime import datetime
 from django.core.exceptions import ObjectDoesNotExist
-from .misc import get_week_dates, paperwork_user_directory_path
+from .misc import get_week_dates, paperwork_user_directory_path, \
+        get_week_dates_datetime
 from django.core.urlresolvers import reverse
 import re
 
@@ -24,6 +25,21 @@ class Customer(models.Model):
     def __str__(self):
         return self.name
 
+    def has_paperwork(self, dt=None):
+        week_dates = get_week_dates_datetime(dt)
+
+        loads = self.load_set.filter(
+                        date__range=(week_dates['start_date'],
+                                     week_dates['end_date']))
+
+        if len(loads) >= 1:
+            for l in loads:
+                if len(l.paperwork_set.all()) >=1:
+                    return True
+
+        return False
+
+
 
 class Load(models.Model):
     history         = AuditlogHistoryField()