Archaeological files: fix parcel clean on save

1 files changed, 15 insertions, 13 deletions

diff --git a/archaeological_operations/models.py b/archaeological_operations/models.py
index 368b2e33c..d0de32be2 100644
--- a/archaeological_operations/models.py
+++ b/archaeological_operations/models.py
@@ -682,9 +682,23 @@ class ClosedItem(object):
         return {'date': date, 'user': user}
 
 
+class ParcelItem:
+    def clean_parcel_duplicates(self):
+        parcels = []
+        for p in self.parcels.order_by('pk').all():
+            if p.associated_file:
+                continue
+            key = (p.section, p.parcel_number, p.year, p.town.pk,
+                   p.public_domain)
+            if key in parcels:
+                p.delete()
+            else:
+                parcels.append(key)
+
+
 class Operation(ClosedItem, DocumentItem, BaseHistorizedItem, QRCodeItem,
                 GeoItem, OwnPerms, ValueGetter, MainItem,
-                DashboardFormItem, RelationItem):
+                DashboardFormItem, RelationItem, ParcelItem):
     SLUG = 'operation'
     APP = "archaeological-operations"
     MODEL = "operation"
@@ -1739,18 +1753,6 @@ class Operation(ClosedItem, DocumentItem, BaseHistorizedItem, QRCodeItem,
         res['mode'] = " ; ".join([str(m) for m in mode(finds)])
         return res
 
-    def clean_parcel_duplicates(self):
-        parcels = []
-        for p in self.parcels.order_by('pk').all():
-            if p.associated_file:
-                continue
-            key = (p.section, p.parcel_number, p.year, p.town.pk,
-                   p.public_domain)
-            if key in parcels:
-                p.delete()
-            else:
-                parcels.append(key)
-
     def save(self, *args, **kwargs):
         # put a default year if start_date is defined
         if self.start_date and not self.year: