Add reverse code
This commit is contained in:
@@ -13,6 +13,18 @@ def forward(apps, schema_editor):
|
|||||||
segment.save()
|
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):
|
class Migration(migrations.Migration):
|
||||||
|
|
||||||
dependencies = [
|
dependencies = [
|
||||||
@@ -25,7 +37,7 @@ class Migration(migrations.Migration):
|
|||||||
name='enabled',
|
name='enabled',
|
||||||
field=models.BooleanField(default=True, help_text='Should the segment be active?'),
|
field=models.BooleanField(default=True, help_text='Should the segment be active?'),
|
||||||
),
|
),
|
||||||
migrations.RunPython(forward),
|
migrations.RunPython(forward, reverse_code=backward),
|
||||||
migrations.RemoveField(
|
migrations.RemoveField(
|
||||||
model_name='segment',
|
model_name='segment',
|
||||||
name='status',
|
name='status',
|
||||||
|
Reference in New Issue
Block a user