add reverse function in migration file

This commit is contained in:
Laetitia Getti 2023-08-09 11:31:46 +02:00
parent 6006cbe8df
commit cb2373020e

View file

@ -23,7 +23,14 @@ def set_code_default(apps, schema_editor):
for grouped_order in GroupedOrder.objects.all().iterator():
grouped_order.code = create_code_from_pk(grouped_order.pk)
grouped_order.save()
def reverse_set_code_default(apps, schema_editor):
"""Reverse the set_code default function"""
GroupedOrder = apps.get_model("order","GroupedOrder")
for grouped_order in GroupedOrder.objects.all().iterator():
grouped_order.code = ''
grouped_order.save()
class Migration(migrations.Migration):
dependencies = [
@ -36,7 +43,7 @@ class Migration(migrations.Migration):
name="code",
field=models.CharField(auto_created=True, null=True),
),
migrations.RunPython(set_code_default),
migrations.RunPython(set_code_default, reverse_set_code_default),
migrations.AlterField(
model_name="groupedorder",
name="code",