diff options
Diffstat (limited to 'kitchen/migrations')
| -rw-r--r-- | kitchen/migrations/0003_pantryitem_state.py | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/kitchen/migrations/0003_pantryitem_state.py b/kitchen/migrations/0003_pantryitem_state.py new file mode 100644 index 0000000..a380a39 --- /dev/null +++ b/kitchen/migrations/0003_pantryitem_state.py @@ -0,0 +1,42 @@ +from django.db import migrations, models + + +def backfill_state(apps, schema_editor): + """Items at quantity 0 are 'out'; everything else is 'in' (the field default).""" + PantryItem = apps.get_model("kitchen", "PantryItem") + PantryItem.objects.filter(quantity=0).update(state="out") + + +class Migration(migrations.Migration): + + dependencies = [ + ("kitchen", "0002_add_rating_and_preferences"), + ] + + operations = [ + migrations.AddField( + model_name="pantryitem", + name="state", + field=models.CharField( + choices=[("in", "In stock"), ("low", "Running low"), ("out", "Out")], + default="in", + max_length=10, + ), + ), + # Runs while quantity is still a Decimal column — 0 matches fine. + migrations.RunPython(backfill_state, migrations.RunPython.noop), + migrations.AlterField( + model_name="pantryitem", + name="quantity", + field=models.PositiveIntegerField( + blank=True, + help_text="Optional rough count; In/Low/Out is the primary signal", + null=True, + ), + ), + migrations.AlterField( + model_name="pantryitem", + name="unit", + field=models.CharField(blank=True, max_length=20), + ), + ] |
