Preventive file: cost propertues - display cost in sheet files

1 files changed, 78 insertions, 9 deletions

diff --git a/archaeological_files/models.py b/archaeological_files/models.py
index 79dabc84d..0eac73119 100644
--- a/archaeological_files/models.py
+++ b/archaeological_files/models.py
@@ -235,6 +235,12 @@ class EquipmentServiceCost(models.Model):
     def history_compress(self):
         return self.slug
 
+    @property
+    def unit_label(self):
+        if self.unit and self.unit in DCT_ES_UNITS:
+            return DCT_ES_UNITS[self.unit]
+        return ""
+
 
 class FileType(GeneralType):
     class Meta:
@@ -1033,6 +1039,47 @@ class File(
             self.general_contractor.save()
         return True
 
+    @property
+    def job_cost_planned(self):
+        return sum(job.cost_planned for job in self.jobs.all())
+
+    @property
+    def ground_job_cost_planned(self):
+        return sum(job.cost_planned for job in self.ground_jobs.all())
+
+    @property
+    def job_cost_worked(self):
+        return sum(job.cost_worked for job in self.jobs.all())
+
+    @property
+    def ground_job_cost_worked(self):
+        return sum(job.cost_worked for job in self.ground_jobs.all())
+
+    @property
+    def job_cost_diff_planned_worked(self):
+        return self.job_cost_planned - self.job_cost_worked
+
+    @property
+    def ground_job_cost_diff_planned_worked(self):
+        return self.ground_job_cost_planned - self.ground_job_cost_worked
+
+    @property
+    def used_equipments(self):
+        equipments = []
+        service_types = list(GenericEquipmentServiceType.objects.all())
+        for service_type in service_types:
+            q = self.equipment_costs.filter(
+                equipment_service_cost__equipment_service_type__generic_equipment_type=service_type)
+            if not q.count():
+                continue
+            equipments.append([service_type.label, 0, 0, 0, []])
+            for cost in q.all():
+                equipments[-1][-4] += cost.cost_planned
+                equipments[-1][-3] += cost.cost_worked
+                equipments[-1][-1].append(cost)
+            equipments[-1][-2] = equipments[-1][-4] - equipments[-1][-3]
+        return equipments
+
     def get_extra_actions(self, request):
         # url, base_text, icon, extra_text, extra css class, is a quick action
         actions = super(File, self).get_extra_actions(request)
@@ -1318,6 +1365,14 @@ class PreventiveFileGroundJob(ManDays):
     class Meta:
         ordering = ("job",)
 
+    @property
+    def cost_planned(self):
+        return (self.job.ground_daily_cost or 0) * self.quantity_planned
+
+    @property
+    def cost_worked(self):
+        return (self.job.ground_daily_cost or 0) * self.quantity_worked
+
 
 class PreventiveFileJob(ManDays):
     file = models.ForeignKey(File, related_name="jobs")
@@ -1326,8 +1381,18 @@ class PreventiveFileJob(ManDays):
     class Meta:
         ordering = ("job",)
 
+    @property
+    def cost_planned(self):
+        return (self.job.daily_cost or 0) * self.quantity_planned
+
+    @property
+    def cost_worked(self):
+        return (self.job.daily_cost or 0) * self.quantity_worked
+
 
-class TechDays(models.Model):
+class PreventiveFileEquipmentServiceCost(models.Model):
+    file = models.ForeignKey(File, related_name="equipment_costs")
+    equipment_service_cost = models.ForeignKey(EquipmentServiceCost)
     quantity_by_day_planned = models.FloatField(
         _("Quantity by day - planned"), null=True, blank=True
     )
@@ -1338,24 +1403,28 @@ class TechDays(models.Model):
     days_worked = models.FloatField(_("Days - worked"), null=True, blank=True)
 
     class Meta:
-        abstract = True
+        ordering = ("equipment_service_cost",)
 
     @property
     def quantity_planned(self):
+        if self.equipment_service_cost.flat_rate:
+            return self.quantity_by_day_planned or 0
         if not self.days_planned or not self.quantity_by_day_planned:
             return 0
         return self.days_planned * self.quantity_by_day_planned
 
     @property
     def quantity_worked(self):
-        if not self.days_worked or not self.quantity_by_day_worked:
+        if self.equipment_service_cost.flat_rate:
+            return self.quantity_by_day_worked or 0
+        if not self.quantity_by_day_worked or not self.days_worked:
             return 0
         return self.days_worked * self.quantity_by_day_worked
 
+    @property
+    def cost_planned(self):
+        return (self.equipment_service_cost.unitary_cost or 0) * self.quantity_planned
 
-class PreventiveFileEquipmentServiceCost(TechDays):
-    file = models.ForeignKey(File, related_name="equipment_costs")
-    equipment_service_cost = models.ForeignKey(EquipmentServiceCost)
-
-    class Meta:
-        ordering = ("equipment_service_cost",)
+    @property
+    def cost_worked(self):
+        return (self.equipment_service_cost.unitary_cost or 0) * self.quantity_worked