Initial commit

app/companies.py

@@ -0,0 +1,59 @@
+"""
+Company resolution.
+
+Two ways an order tells us which company it belongs to:
+  - JIRA orders: the SKU prefix (SH... -> Signify Health, OK... -> Oak Street Health)
+  - ShipStation orders: which store the order lives in (each company has
+    its own store because they have separate UPS accounts)
+
+Both mappings are configurable in Settings so adding a third company
+later doesn't require touching code - just add another "PREFIX:Company
+Name" or "storeId:Company Name" entry.
+"""
+from __future__ import annotations
+
+UNKNOWN_COMPANY = "Unknown"
+
+
+def parse_mapping(raw: str) -> dict[str, str]:
+    """Parse a 'KEY:Value,KEY2:Value2' settings string into a dict."""
+    mapping: dict[str, str] = {}
+    if not raw:
+        return mapping
+    for part in raw.split(","):
+        part = part.strip()
+        if not part or ":" not in part:
+            continue
+        key, _, value = part.partition(":")
+        key, value = key.strip(), value.strip()
+        if key and value:
+            mapping[key] = value
+    return mapping
+
+
+def resolve_company_by_sku(sku: str, sku_map: dict[str, str]) -> str:
+    """Match a SKU against configured prefixes, e.g. 'SH-1029' -> 'Signify Health'."""
+    if not sku:
+        return UNKNOWN_COMPANY
+    sku_upper = sku.strip().upper()
+    for prefix, company in sku_map.items():
+        if sku_upper.startswith(prefix.upper()):
+            return company
+    return UNKNOWN_COMPANY
+
+
+def resolve_company_for_skus(skus: list[str], sku_map: dict[str, str]) -> str:
+    """
+    Resolve the company for a set of SKUs on one ticket. Per business rule,
+    companies never mix on a single ticket, so the first recognizable SKU
+    decides it.
+    """
+    for sku in skus:
+        company = resolve_company_by_sku(sku, sku_map)
+        if company != UNKNOWN_COMPANY:
+            return company
+    return UNKNOWN_COMPANY
+
+
+def resolve_company_by_store(store_id: str, store_map: dict[str, str]) -> str:
+    return store_map.get(str(store_id), UNKNOWN_COMPANY)