From acd273c06c56e952ed9bc5097ea069c1fd7e4b59 Mon Sep 17 00:00:00 2001 From: Jasper Berghoef Date: Thu, 19 Jul 2018 16:09:55 +0200 Subject: [PATCH] Add reverse code --- .../migrations/0024_segment_status_as_boolean.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/src/wagtail_personalisation/migrations/0024_segment_status_as_boolean.py b/src/wagtail_personalisation/migrations/0024_segment_status_as_boolean.py index 08ce77b..413b542 100644 --- a/src/wagtail_personalisation/migrations/0024_segment_status_as_boolean.py +++ b/src/wagtail_personalisation/migrations/0024_segment_status_as_boolean.py @@ -13,6 +13,18 @@ def forward(apps, schema_editor): segment.save() +def backward(apps, schema_editor): + Segment = apps.get_model('wagtail_personalisation', 'Segment') + + for segment in Segment.objects.all(): + if segment.enabled: + segment.status = 'enabled' + else: + segment.status = 'disabled' + + segment.save() + + class Migration(migrations.Migration): dependencies = [ @@ -25,7 +37,7 @@ class Migration(migrations.Migration): name='enabled', field=models.BooleanField(default=True, help_text='Should the segment be active?'), ), - migrations.RunPython(forward), + migrations.RunPython(forward, reverse_code=backward), migrations.RemoveField( model_name='segment', name='status',