Treatment: manage treatment with baskets

1 files changed, 16 insertions, 18 deletions

diff --git a/archaeological_finds/models.py b/archaeological_finds/models.py
index 01c904cc8..3e0d09ef8 100644
--- a/archaeological_finds/models.py
+++ b/archaeological_finds/models.py
@@ -879,27 +879,25 @@ class Treatment(BaseHistorizedItem, ImageModel, OwnPerms):
             user = kwargs.pop('user')
         if "extra_args_for_new" in kwargs:
             extra_args_for_new = kwargs.pop('extra_args_for_new')
-        is_new = self.pk is None
         self.pre_save()
         super(Treatment, self).save(*args, **kwargs)
         updated = []
-        if is_new:
-            if hasattr(items, "items"):
-                items = items.items.all()
-            for item in items:
-                new = item.duplicate(user)
-                item.downstream_treatment = self
-                item.save()
-                new.upstream_treatment = self
-                for k in extra_args_for_new:
-                    setattr(new, k, extra_args_for_new[k])
-                new.save()
-                updated.append(new.pk)
-                # update baskets
-                for basket in \
-                        FindBasket.objects.filter(items__pk=item.pk).all():
-                    basket.items.remove(item)
-                    basket.items.add(new)
+        if hasattr(items, "items"):
+            items = items.items.all()
+        for item in items:
+            new = item.duplicate(user)
+            item.downstream_treatment = self
+            item.save()
+            new.upstream_treatment = self
+            for k in extra_args_for_new:
+                setattr(new, k, extra_args_for_new[k])
+            new.save()
+            updated.append(new.pk)
+            # update baskets
+            for basket in \
+                    FindBasket.objects.filter(items__pk=item.pk).all():
+                basket.items.remove(item)
+                basket.items.add(new)
         # manage containers
         for find in Find.objects.filter(upstream_treatment=self).all():
             if find.container != self.container: