summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--archaeological_warehouse/migrations/0103_auto_container_views.py12
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),
]