ATP/CTP动态承诺交期计算 — 基于实时库存+PLT的订单交货承诺日期引擎

"""
ATP/CTP 动态承诺交期计算引擎
功能：ATP计算 / CTP推导 / 承诺日期生成 / 可靠性标签 / 批量承诺
"""
from dataclasses import dataclass, field
from datetime import datetime, timedelta
from typing import Optional
import warnings
warnings.filterwarnings('ignore')


@dataclass
class SKUSupplyPosition:
    sku_id: str
    warehouse_id: str
    on_hand_qty: int
    reserved_qty: int
    planned_receipts: list = field(default_factory=list)  # [(date, qty)]
    plt_p85_days: float = 35.0
    daily_demand: float = 10.0

    @property
    def atp_today(self) -> int:
        return max(0, self.on_hand_qty - self.reserved_qty)

    @property
    def dos(self) -> float:
        return self.on_hand_qty / max(0.1, self.daily_demand)


@dataclass
class PromiseResult:
    order_id: str
    sku_id: str
    requested_qty: int
    promise_type: str      # ATP / CTP / PARTIAL / REJECT
    promise_date: Optional[datetime]
    available_qty: int
    reliability: str       # HIGH / MEDIUM / LOW
    reason: str
    tags: dict = field(default_factory=dict)


class PromiseDateEngine:

    TRANSIT_DAYS = {
        ("WH-NJ", "US-East"): 2, ("WH-NJ", "US-Midwest"): 4,
        ("WH-CA", "US-West"): 2, ("WH-CA", "US-East"): 5,
    }

    def calculate_promise(self, order_id: str, sku_pos: SKUSupplyPosition,
                           qty: int, destination: str, priority: str) -> PromiseResult:
        now = datetime.now()
        transit = self.TRANSIT_DAYS.get((sku_pos.warehouse_id, destination), 4)

        # 场景1: ATP充足
        if sku_pos.atp_today >= qty and sku_pos.dos >= 7:
            promise_date = now + timedelta(days=1 + transit)  # 1天出库+运输
            reliability = "HIGH" if sku_pos.dos >= 14 else "MEDIUM"
            return PromiseResult(