summaryrefslogtreecommitdiff
path: root/kitchen/migrations/0003_pantryitem_state.py
blob: a380a39c66abfef3126c730042d491b0b63384a2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
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),
        ),
    ]