Operation from file creation - fix automatic town association and parcel recopy

1 files changed, 31 insertions, 0 deletions

diff --git a/archaeological_operations/models.py b/archaeological_operations/models.py
index 57f538383..293b1e933 100644
--- a/archaeological_operations/models.py
+++ b/archaeological_operations/models.py
@@ -2260,6 +2260,12 @@ def operation_post_save(sender, **kwargs):
     post_save_geo(sender=sender, **kwargs)
 
     operation = kwargs["instance"]
+
+    # on creation associate parcel from a file if the file exist
+    if kwargs.get("created", False) and operation.associated_file:
+        for parcel in operation.associated_file.parcels.all():
+            parcel.copy_from_file_to_operation(operation)
+
     operation.skip_history_when_saving = True
     if operation.fnap_financing and operation.cost:
         fnap_cost = int(float(operation.cost) / 100 * operation.fnap_financing)
@@ -3194,6 +3200,31 @@ class Parcel(LightHistorizedItem):
         self.operation = None
         self.save()
 
+    def copy_from_file_to_operation(self, operation):
+        """
+        Used to copy parcel from file on operation creation
+        """
+        if not self.associated_file:
+            return
+        keys = {
+            "town": self.town,
+            "section": self.section,
+            "parcel_number": self.parcel_number,
+            "operation": operation,
+            "associated_file": None,
+            "defaults": {"address": self.address, "year": self.year},
+        }
+        new_p, created = Parcel.objects.get_or_create(**keys)
+        # copy owning only if created
+        if created:
+            for owning in self.owners.all():
+                ParcelOwner.objects.create(
+                    owner=owning.owner,
+                    parcel=new_p,
+                    start_date=owning.start_date,
+                    end_date=owning.end_date,
+                )
+
     def clean_orphan(self):
         """
         Remove when the parcel is linked to nothing