Merge pull request #736 from ImageMarkup/isic-138-migrate-image-types

Migrate clinical image types in preparation for TBP types

isic/ingest/migrations/0040_migrate_image_types.py

@@ -0,0 +1,31 @@
+# Generated by Django 4.1.10 on 2023-08-01 21:01
+
+from django.db import migrations, transaction
+
+
+def migrate_image_types(apps, schema_editor):
+    Accession = apps.get_model("ingest", "Accession")
+
+    with transaction.atomic():
+        objs = []
+        clinical_qs = Accession.objects.filter(metadata__image_type="clinical").select_for_update()
+
+        for accession in clinical_qs.iterator():
+            accession.metadata["image_type"] = "clinical close-up"
+            objs.append(accession)
+
+        overview_qs = Accession.objects.filter(metadata__image_type="overview").select_for_update()
+
+        for accession in overview_qs.iterator():
+            accession.metadata["image_type"] = "clinical overview"
+            objs.append(accession)
+
+        Accession.objects.bulk_update(objs, ["metadata"], batch_size=100)
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("ingest", "0039_alter_accession_copyright_license"),
+    ]
+
+    operations = [migrations.RunPython(migrate_image_types)]

setup.py

@@ -62,7 +62,7 @@
         "drf-yasg",
         "google-api-python-client",
         "hashids",
-        "isic-metadata>=0.0.6",
+        "isic-metadata>=0.0.8",
         "jaro-winkler",
         "more_itertools",
         "oauth2client",