fix(Prices): init Price.receipt_quantity field in migration. ref #498

openfoodfacts · Oct 18, 2024 · 1822b8d · 1822b8d
1 parent 857b813
commit 1822b8d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/open_prices/prices/migrations/0003_price_receipt_quantity.py b/open_prices/prices/migrations/0003_price_receipt_quantity.py
@@ -4,6 +4,13 @@
 from django.db import migrations, models
 
 
+def init_receipt_quantity(apps, schema_editor):
+    Price = apps.get_model("prices", "Price")
+    Price.objects.filter(
+        proof__type__in=["RECEIPT", "GDPR_REQUEST"], receipt_quantity=None
+    ).update(receipt_quantity=1)
+
+
 class Migration(migrations.Migration):
     dependencies = [
         ("prices", "0002_alter_price_currency"),
@@ -20,4 +27,5 @@ class Migration(migrations.Migration):
                 verbose_name="Receipt's price quantity (user input)",
             ),
         ),
+        migrations.RunPython(init_receipt_quantity),
     ]