Create equipment_catalog.py

equipment_catalog.py

@@ -0,0 +1,145 @@
+from dataclasses import dataclass, asdict
+from typing import Dict, Any, Optional, List
+import json, os
+from datetime import date
+
+CATALOG_PATH = "equipment_catalog.json"
+ASSETS_PATH = "purchased_assets.json"
+
+@dataclass
+class EquipmentSpec:
+    # Identity
+    id: str
+    name: str
+    category: str   # "tractor", "planter", etc.
+    brand: str
+    model: str
+
+    # Economics
+    list_price: float
+    expected_life_years: int
+    salvage_value_pct: float  # straight-line baseline (% of cost)
+
+    # Ops (extend as needed)
+    rated_hp: Optional[int] = None
+    transport_speed_mph: Optional[float] = None
+
+    # Depreciation (ASAE defaults for tractors; can refine later)
+    dep_method_default: str = "ASAE"
+    asae_dep1: Optional[float] = 0.67
+    asae_dep2: Optional[float] = 0.94
+
+    # K-State / ASAE repair model (tractor-ish defaults; refine per class)
+    repair_rf1: Optional[float] = 0.007
+    repair_rf2: Optional[float] = 2.0
+    eul_hours: Optional[int] = 12000
+
+    fuel_type: Optional[str] = "diesel"
+    notes: Optional[str] = None
+
+
+def default_catalog() -> List[EquipmentSpec]:
+    return [
+        EquipmentSpec(
+            id="jd-6155m",
+            name="John Deere 6155M",
+            category="tractor",
+            brand="John Deere",
+            model="6155M",
+            list_price=185000.0,
+            expected_life_years=12,
+            salvage_value_pct=0.18,
+            rated_hp=155,
+            transport_speed_mph=24.0,
+            notes="Mid-range row-crop tractor.",
+        ),
+        EquipmentSpec(
+            id="caseih-maxxum-125",
+            name="Case IH Maxxum 125",
+            category="tractor",
+            brand="Case IH",
+            model="Maxxum 125",
+            list_price=165000.0,
+            expected_life_years=12,
+            salvage_value_pct=0.17,
+            rated_hp=125,
+            transport_speed_mph=25.0,
+        ),
+        EquipmentSpec(
+            id="kubota-m7-151",
+            name="Kubota M7-151",
+            category="tractor",
+            brand="Kubota",
+            model="M7-151",
+            list_price=150000.0,
+            expected_life_years=12,
+            salvage_value_pct=0.16,
+            rated_hp=150,
+            transport_speed_mph=24.0,
+        ),
+        EquipmentSpec(
+            id="nh-t7-190",
+            name="New Holland T7.190",
+            category="tractor",
+            brand="New Holland",
+            model="T7.190",
+            list_price=178000.0,
+            expected_life_years=12,
+            salvage_value_pct=0.17,
+            rated_hp=165,
+            transport_speed_mph=31.0,
+        ),
+    ]
+
+
+def _atomic_write_json(path: str, data: Any):
+    tmp = path + ".tmp"
+    with open(tmp, "w") as f:
+        json.dump(data, f, indent=2, default=str)
+    os.replace(tmp, path)
+
+def ensure_catalog_file():
+    if not os.path.exists(CATALOG_PATH):
+        items = [asdict(x) for x in default_catalog()]
+        _atomic_write_json(CATALOG_PATH, items)
+
+def load_catalog() -> Dict[str, EquipmentSpec]:
+    ensure_catalog_file()
+    with open(CATALOG_PATH, "r") as f:
+        raw = json.load(f)
+    cat: Dict[str, EquipmentSpec] = {}
+    for item in raw:
+        cat[item["id"]] = EquipmentSpec(**item)
+    return cat
+
+def load_assets() -> list:
+    if not os.path.exists(ASSETS_PATH):
+        _atomic_write_json(ASSETS_PATH, [])
+        return []
+    with open(ASSETS_PATH, "r") as f:
+        return json.load(f)
+
+def add_asset_from_catalog(spec: EquipmentSpec, purchase_price: Optional[float] = None) -> Dict[str, Any]:
+    assets = load_assets()
+    entry = {
+        "asset_id": f"{spec.id}-{len(assets)+1}",
+        "catalog_id": spec.id,
+        "name": spec.name,
+        "category": spec.category,
+        "brand": spec.brand,
+        "model": spec.model,
+        "purchase_date": str(date.today()),
+        "cost": float(purchase_price) if purchase_price is not None else float(spec.list_price),
+        "expected_life_years": spec.expected_life_years,
+        "salvage_value_pct": spec.salvage_value_pct,
+        "dep_method": spec.dep_method_default,
+        "asae_dep1": spec.asae_dep1,
+        "asae_dep2": spec.asae_dep2,
+        "repair": {"rf1": spec.repair_rf1, "rf2": spec.repair_rf2, "eul_hours": spec.eul_hours},
+        "ops": {"rated_hp": spec.rated_hp, "transport_speed_mph": spec.transport_speed_mph, "fuel": spec.fuel_type},
+        "notes": spec.notes,
+        "hours_used": 0,
+    }
+    assets.append(entry)
+    _atomic_write_json(ASSETS_PATH, assets)
+    return entry