1 files changed, 13 insertions, 3 deletions
diff --git a/kitchen/models.py b/kitchen/models.py
index ab0a720..07bf231 100644
--- a/kitchen/models.py
+++ b/kitchen/models.py
@@ -50,9 +50,18 @@ class PantryItem(models.Model):
         FREEZER = "freezer", "Freezer"
         CUPBOARD = "cupboard", "Cupboard"
 
+    class State(models.TextChoices):
+        IN = "in", "In stock"
+        LOW = "low", "Running low"
+        OUT = "out", "Out"
+
     ingredient = models.ForeignKey(Ingredient, on_delete=models.CASCADE)
-    quantity = models.DecimalField(max_digits=8, decimal_places=2)
-    unit = models.CharField(max_length=20)
+    # In/Low/Out is the primary signal. quantity is an optional rough count.
+    state = models.CharField(max_length=10, choices=State.choices, default=State.IN)
+    quantity = models.PositiveIntegerField(
+        null=True, blank=True, help_text="Optional rough count; In/Low/Out is the primary signal"
+    )
+    unit = models.CharField(max_length=20, blank=True)
     location = models.CharField(max_length=20, choices=Location.choices)
     stored_date = models.DateField(auto_now_add=True)
     expiry_date = models.DateField(null=True, blank=True)
@@ -65,7 +74,8 @@ class PantryItem(models.Model):
         ordering = ["expiry_date", "ingredient__name"]
 
     def __str__(self):
-        return f"{self.ingredient.name} ({self.quantity} {self.unit}) [{self.location}]"
+        qty = f"{self.quantity} {self.unit}" if self.quantity is not None else self.get_state_display()
+        return f"{self.ingredient.name} ({qty}) [{self.location}]"
 
     def clean(self):
         """Business rules for pantry items."""