Preventive file: add extra field for export

1 files changed, 103 insertions, 2 deletions

diff --git a/archaeological_files/models.py b/archaeological_files/models.py
index 5881710e6..7d9391178 100644
--- a/archaeological_files/models.py
+++ b/archaeological_files/models.py
@@ -26,7 +26,7 @@ from django.contrib.gis.db import models
 from django.contrib.postgres.indexes import GinIndex
 from django.core.cache import cache
 from django.core.validators import MinValueValidator, MaxValueValidator
-from django.db.models import Q, Count, Sum
+from django.db.models import Q, Count, Sum, Max
 from django.db.models.signals import post_save, m2m_changed, post_delete
 from django.core.urlresolvers import reverse
 from ishtar_common.utils import ugettext_lazy as _, pgettext_lazy
@@ -188,6 +188,13 @@ ES_UNITS = (
 
 DCT_ES_UNITS = dict(ES_UNITS)
 
+ES_UNITS_DAYS = {
+    "D": 1,
+    "W": 7,
+    "M": 30,
+    "L": 0
+}
+
 
 class EquipmentServiceCost(models.Model):
     price_agreement = models.ForeignKey(
@@ -719,6 +726,14 @@ class File(
         "general_contractor_address_2",
         "general_contractor_address_3",
         "get_locality",
+        "excavation_working_time_planned",
+        "excavation_working_time_worked",
+        "study_working_time_planned",
+        "study_working_time_worked",
+        "excavation_average_team_planned",
+        "excavation_average_team_worked",
+        "study_average_team_planned",
+        "study_average_team_worked",
     ]
 
     class Meta:
@@ -914,11 +929,49 @@ class File(
         )
         return cls._return_get_owns(owns, values, get_short_menu_class)
 
+    def get_dynamic_values(self, prefix, values, filtr=None):
+        q = GenericEquipmentServiceType.objects.filter(available=True)
+        for equipment_service in q.all():
+            pk = equipment_service.pk
+            #if not filtr or not any(
+            #        key for f in filtr if f.startswith(prefix + key)
+            #):
+            q = self.equipment_costs.filter(
+                equipment_service_cost__equipment_service_type__generic_equipment_type_id=pk,
+                equipment_service_cost__unit__isnull=False,
+            ).exclude(
+                equipment_service_cost__unit__in=("L", ""),  # exclude linear meter
+            )
+            slug = equipment_service.txt_idx.replace("-", "_")
+            key_time_planned = slug + "_equipment_time_planned"
+            values[prefix + key_time_planned] = 0
+            key_time_worked = slug + "_equipment_time_worked"
+            values[prefix + key_time_worked] = 0
+            key_qt_planned = slug + "_equipment_sum_planned"
+            values[prefix + key_qt_planned] = 0
+            key_qt_worked = slug + "_equipment_sum_worked"
+            values[prefix + key_qt_worked] = 0
+            for cost in q.all():
+                values[prefix + key_qt_planned] += (cost.quantity_by_day_planned or 0)
+                values[prefix + key_qt_worked] += (cost.quantity_by_day_worked or 0)
+                print()
+                value = ES_UNITS_DAYS[cost.equipment_service_cost.unit] * (
+                    cost.days_planned or 0)
+                if value > values[prefix + key_time_planned]:
+                    values[prefix + key_time_planned] = value
+                value = ES_UNITS_DAYS[cost.equipment_service_cost.unit] * (
+                        cost.days_worked or 0)
+                if value > values[prefix + key_time_planned]:
+                    values[prefix + key_time_worked] = value
+        return values
+
     def get_values(self, prefix="", no_values=False, filtr=None, **kwargs):
         values = super(File, self).get_values(
             prefix=prefix, no_values=no_values, filtr=filtr, **kwargs
         )
-        return get_values_town_related(self, prefix, values, filtr=filtr)
+        values = get_values_town_related(self, prefix, values, filtr=filtr)
+        values = self.get_dynamic_values(prefix, values, filtr=filtr)
+        return values
 
     def render_parcels(self):
         Parcel = apps.get_model("archaeological_operations", "Parcel")
@@ -1070,6 +1123,54 @@ class File(
         return True
 
     @property
+    def excavation_working_time_planned(self):
+        return self.ground_jobs.all().aggregate(Max("days_planned"))["days_planned__max"] or 0
+
+    @property
+    def study_working_time_planned(self):
+        return self.jobs.all().aggregate(Max("days_planned"))["days_planned__max"] or 0
+
+    @property
+    def excavation_working_time_worked(self):
+        return self.ground_jobs.all().aggregate(Max("days_worked"))["days_worked__max"] or 0
+
+    @property
+    def study_working_time_worked(self):
+        return self.jobs.all().aggregate(Max("days_worked"))["days_worked__max"] or 0
+
+    @property
+    def excavation_average_team_planned(self):
+        return round(
+            sum([(job.man_by_day_planned or 0) * (job.days_planned or 0)
+                 for job in self.ground_jobs.all()]) /
+            (self.excavation_working_time_planned or 1)
+        )
+
+    @property
+    def excavation_average_team_worked(self):
+        return round(
+            sum([(job.man_by_day_worked or 0) * (job.days_worked or 0)
+                 for job in self.ground_jobs.all()]) /
+            (self.excavation_working_time_worked or 1)
+        )
+
+    @property
+    def study_average_team_planned(self):
+        return round(
+            sum([(job.man_by_day_planned or 0) * (job.days_planned or 0)
+                 for job in self.jobs.all()]) /
+            (self.study_working_time_planned or 1)
+        )
+
+    @property
+    def study_average_team_worked(self):
+        return round(
+            sum([(job.man_by_day_worked or 0) * (job.days_worked or 0)
+                 for job in self.jobs.all()]) /
+            (self.study_working_time_worked or 1)
+        )
+
+    @property
     def job_cost_planned(self):
         return sum(job.cost_planned for job in self.jobs.all())