Migrate to collections

1 files changed, 12 insertions, 0 deletions

diff --git a/archaeological_warehouse/migrations/0103_auto_container_views.py b/archaeological_warehouse/migrations/0103_auto_container_views.py
index 0a5bacc81..f49b7be1f 100644
--- a/archaeological_warehouse/migrations/0103_auto_container_views.py
+++ b/archaeological_warehouse/migrations/0103_auto_container_views.py
@@ -8,6 +8,17 @@ import archaeological_warehouse.models
 import archaeological_finds.models
 
 
+def migrate_to_collections(apps, schema_editor):
+    Container = apps.get_model('archaeological_warehouse', 'Container')
+    for c in Container.objects.all():
+        if c.responsible_id and c.location_id != c.responsible_id:
+            for find in c.finds_ref.all():
+                find.collection = c.responsible
+                find.skip_history_when_saving = True
+                find._no_move = True
+                find.save()
+
+
 class Migration(migrations.Migration):
 
     dependencies = [
@@ -23,4 +34,5 @@ class Migration(migrations.Migration):
             archaeological_warehouse.models.ContainerTree.CREATE_SQL),
         migrations.RunSQL(
             archaeological_finds.models.FindInsideContainer.CREATE_SQL),
+        migrations.RunPython(migrate_to_collections),
     ]