Migrate main image: fix migration script

1 files changed, 24 insertions, 0 deletions

diff --git a/archaeological_finds/migrations/0057_migrate_main_image.py b/archaeological_finds/migrations/0057_migrate_main_image.py
new file mode 100644
index 000000000..dceff0ea0
--- /dev/null
+++ b/archaeological_finds/migrations/0057_migrate_main_image.py
@@ -0,0 +1,24 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.10 on 2018-12-20 19:12
+from __future__ import unicode_literals
+
+from django.db import migrations
+from ishtar_common.utils_migrations import migrate_main_image
+
+
+def migrate_main_image_script(apps, schema):
+    migrate_main_image(apps, 'archaeological_finds', 'Find', verbose=True)
+    migrate_main_image(apps, 'archaeological_finds', 'Treatment', verbose=True)
+    migrate_main_image(apps, 'archaeological_finds', 'TreatmentFile',
+                       verbose=True)
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('archaeological_finds', '0056_auto_20190118_1203'),
+    ]
+
+    operations = [
+        migrations.RunPython(migrate_main_image_script)
+    ]