inventory.py

from enums import SupplyType
from constants import RESTOCK_THRESHOLD

class Inventory:
    def __init__(self):
        self.stock = {
            SupplyType.MEDICATION:      50,
            SupplyType.BANDAGES:        60,
            SupplyType.OXYGEN:          40,
            SupplyType.SURGICAL_KIT:    15,
            SupplyType.BLOOD:           25
        }

    def consume(self, supply_type, amount=1):
        if self.stock[supply_type] >= amount:
            self.stock[supply_type] -= amount
            return True
        return False
    
    def restock(self, supply_type, amount):
        self.stock[supply_type] += amount
    
    def is_low(self, supply_type):
        return self.stock[supply_type] <= RESTOCK_THRESHOLD[supply_type]
    
    def __repr__(self):
        lines = ["Inventory:"]
        for supply, qty in self.stock.items():
            low = "LOW" if self.is_low(supply) else ""
            lines.append(f"  {supply.value:<15}: {qty}{low}")
        return "\n".join(lines)