Preventive file form: fix JS - fix default cost add with price agreement

1 files changed, 11 insertions, 2 deletions

diff --git a/archaeological_files/views.py b/archaeological_files/views.py
index d91688ac4..eaeccd1ad 100644
--- a/archaeological_files/views.py
+++ b/archaeological_files/views.py
@@ -546,8 +546,13 @@ def file_edit_preventive_add_default(request, pk, current_right=None):
         ("default_daily_need_on_ground", models.PreventiveFileGroundJob),
         ("default_daily_need", models.PreventiveFileJob),
     )
+    price_agreement_id = None
+    q = models.File.objects.filter(pk=pk).values("price_agreement_id")
+    if q.count():
+        price_agreement_id = q.all()[0]["price_agreement_id"]
     for attr, job_model in job_attrs:
-        jobs = models.Job.objects.exclude(**{attr: 0})
+        jobs = models.Job.objects.exclude(**{attr: 0}).filter(
+            price_agreement_id=price_agreement_id)
         for job in jobs:
             q = job_model.objects.filter(job=job, file_id=pk)
             if not q.count():
@@ -558,7 +563,11 @@ def file_edit_preventive_add_default(request, pk, current_right=None):
                     "man_by_day_planned": getattr(job, attr),
                 }
                 job_model.objects.create(**dct)
-    q = models.EquipmentServiceCost.objects.exclude(default_quantity_by_day=0)
+    q = models.EquipmentServiceCost.objects.exclude(
+        default_quantity_by_day=0
+    ).filter(
+        price_agreement_id=price_agreement_id
+    )
     for cost in list(q.all()):
         q = models.PreventiveFileEquipmentServiceCost.objects.filter(
             equipment_service_cost=cost, file_id=pk