From ec67cee42fd0f8ac989b5b93949938059b661f0c Mon Sep 17 00:00:00 2001 From: Mitch Riedstra Date: Tue, 9 Jan 2018 01:26:02 -0500 Subject: Add the ability to download all of the paperwork from a customer view --- app/dispatch/models.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) (limited to 'app/dispatch/models.py') 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() -- cgit v1.2.3