From 54c4f8c862c0d34a25eb39b7e6945fb197462c54 Mon Sep 17 00:00:00 2001
From: vrigal <rigal@teklia.com>
Date: Tue, 3 Dec 2019 16:20:21 +0100
Subject: [PATCH] Update migration metadata type and name

---
 .../migrations/0029_structural_metadata.py    | 22 +++++++++++--------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/arkindex/documents/migrations/0029_structural_metadata.py b/arkindex/documents/migrations/0029_structural_metadata.py
index ba1c6366de..0f86901fb5 100644
--- a/arkindex/documents/migrations/0029_structural_metadata.py
+++ b/arkindex/documents/migrations/0029_structural_metadata.py
@@ -5,30 +5,34 @@ from django.db.models import Q
 
 def update_balsac_pages(apps, schema_editor):
     MetaData = apps.get_model('documents', 'MetaData')
-    balsac_struct_metadata = MetaData.objects \
+    balsac_pages_ref_md = MetaData.objects \
         .filter(
             element__corpus__name__in=('Balsac | S3', 'Balsac | From S3'),
             element__type__slug='page'
         ).filter(
             Q(name="name") | Q(name="filename")
         )
-    renamed_count = balsac_struct_metadata.update(name='Reference')
-    print('Renamed {} metadata from BALSAC corpora'.format(renamed_count))
+    renamed_pages_count = balsac_pages_ref_md.update(name='balsac_id')
+    print('Renamed {} reference metadata from BALSAC pages'.format(renamed_pages_count))
+    balsac_ref_metadata = MetaData.objects \
+        .filter(element__corpus__name__in=('Balsac | S3', 'Balsac | From S3'), name='Reference')
+    renamed_other_count = balsac_ref_metadata.update(name='balsac_id')
+    print('Renamed {} reference metadata from all BALSAC elements'.format(renamed_other_count))
 
 
-def structural_metadata(apps, schema_editor):
+def reference_metadata(apps, schema_editor):
     """
     Update reference
     """
     MetaData = apps.get_model('documents', 'MetaData')
-    references = MetaData.objects.filter(Q(name='Reference') | Q(name='reference'), type=MetaType.Text)
-    moved_count = references.update(type=MetaType.Structural)
-    print('Moved {} reference metadata to Structural type'.format(moved_count))
+    references = MetaData.objects.filter(name='balsac_id', type=MetaType.Text)
+    moved_count = references.update(type=MetaType.Reference)
+    print('Moved {} reference metadata to Reference type'.format(moved_count))
 
 
 def reverse_migration(apps, schema_editor):
     MetaData = apps.get_model('documents', 'MetaData')
-    MetaData.objects.filter(type=MetaType.Structural).update(type=MetaType.Text)
+    MetaData.objects.filter(type=MetaType.Reference).update(type=MetaType.Text)
 
 
 class Migration(migrations.Migration):
@@ -39,5 +43,5 @@ class Migration(migrations.Migration):
 
     operations = [
         migrations.RunPython(update_balsac_pages, migrations.RunPython.noop),
-        migrations.RunPython(structural_metadata, reverse_migration)
+        migrations.RunPython(reference_metadata, reverse_migration)
     ]
-- 
GitLab