Migrate images and sources to new document model (refs #4107)

2 files changed, 48 insertions, 0 deletions

diff --git a/archaeological_warehouse/migrations/0019_warehouse_documents.py b/archaeological_warehouse/migrations/0019_warehouse_documents.py
new file mode 100644
index 000000000..08ee5cb34
--- /dev/null
+++ b/archaeological_warehouse/migrations/0019_warehouse_documents.py
@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.10 on 2018-05-30 19:00
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ishtar_common', '0055_auto_20180530_1900'),
+        ('archaeological_warehouse', '0018_auto_20180528_1141'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='warehouse',
+            name='documents',
+            field=models.ManyToManyField(blank=True, related_name='warehouses', to='ishtar_common.Document', verbose_name='Documents'),
+        ),
+    ]
diff --git a/archaeological_warehouse/migrations/0020_migrate_source_and_image.py b/archaeological_warehouse/migrations/0020_migrate_source_and_image.py
new file mode 100644
index 000000000..1db07eee4
--- /dev/null
+++ b/archaeological_warehouse/migrations/0020_migrate_source_and_image.py
@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.10 on 2018-05-30 19:38
+from __future__ import unicode_literals
+
+from django.db import migrations
+
+
+from ishtar_common.utils_migrations import migrate_images
+
+
+def migrate_source_and_image(apps, schema_editor):
+    Warehouse = apps.get_model(
+        'archaeological_warehouse', 'Warehouse')
+    WarehouseImage = apps.get_model(
+        'archaeological_warehouse', 'WarehouseImage')
+    migrate_images(apps, Warehouse, WarehouseImage)
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('archaeological_warehouse', '0019_warehouse_documents'),
+    ]
+
+    operations = [
+        migrations.RunPython(migrate_source_and_image)
+    ]