From acf195f9b350fe78e96159f89035e0125597ac8c Mon Sep 17 00:00:00 2001 From: Oliver Walters Date: Mon, 22 May 2023 19:40:27 +1000 Subject: [PATCH] Update migration file - Allow reverse migration --- .../migrations/0060_auto_20230519_0344.py | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/InvenTree/company/migrations/0060_auto_20230519_0344.py b/InvenTree/company/migrations/0060_auto_20230519_0344.py index 25bca7b391..34e6d2fc69 100644 --- a/InvenTree/company/migrations/0060_auto_20230519_0344.py +++ b/InvenTree/company/migrations/0060_auto_20230519_0344.py @@ -22,6 +22,20 @@ def update_supplier_part_units(apps, schema_editor): print(f"Updated {supplier_parts.count()} supplier part units") +def reverse_pack_quantity(apps, schema_editor): + """Reverse the migrations""" + + SupplierPart = apps.get_model('company', 'SupplierPart') + + supplier_parts = SupplierPart.objects.all() + + for sp in supplier_parts: + sp.pack_size = sp.pack_quantity_native + sp.save() + + if supplier_parts.count() > 0: + print(f"Updated {supplier_parts.count()} supplier part units") + class Migration(migrations.Migration): dependencies = [ @@ -32,6 +46,6 @@ class Migration(migrations.Migration): operations = [ migrations.RunPython( code=update_supplier_part_units, - reverse_code=migrations.RunPython.noop + reverse_code=reverse_pack_quantity, ) ]