From 8e9f798d09c9c65e7b0205ce5e307fd966012ca2 Mon Sep 17 00:00:00 2001 From: Alex Swindler Date: Wed, 13 Mar 2024 13:10:12 -0600 Subject: [PATCH] Fixed historical note backfilling --- seed/migrations/0215_accountability_hierarchy.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/seed/migrations/0215_accountability_hierarchy.py b/seed/migrations/0215_accountability_hierarchy.py index 42ed68c79b..5dffeef409 100644 --- a/seed/migrations/0215_accountability_hierarchy.py +++ b/seed/migrations/0215_accountability_hierarchy.py @@ -44,6 +44,17 @@ def assign_analyses_to_root_access_level(apps, schema_editor): analysis.save(update_fields=['access_level_instance']) +@transaction.atomic +def backfill_historical_notes(apps, schema_editor): + Property = apps.get_model('seed', 'Property') + HistoricalNote = apps.get_model('seed', 'HistoricalNote') + + properties_ids = Property.objects.filter(historical_note__isnull=True).values_list('id', flat=True) + + historical_notes_to_create = [HistoricalNote(property_id=property_id, text='') for property_id in properties_ids] + HistoricalNote.objects.bulk_create(historical_notes_to_create) + + class Migration(migrations.Migration): dependencies = [ @@ -135,6 +146,9 @@ class Migration(migrations.Migration): code=assign_analyses_to_root_access_level, reverse_code=migrations.RunPython.noop, ), + migrations.RunPython( + code=backfill_historical_notes, + ), migrations.AlterField( model_name='property', name='access_level_instance',