from datetime import date
from decimal import Decimal

from group import (
    draft_from_group,
    group_by_supplier,
    is_claimable,
    missing_mdm_reason,
    money4,
)
from repo import PurchaseContractRow


def _row(**overrides) -> PurchaseContractRow:
    data = dict(
        fid=1,
        oms_id=2026100501,
        supplier_id="SUP001",
        supplier_name="示例供应商",
        contract_sign_date=date(2026, 10, 1),
        contract_start_date=date(2026, 10, 2),
        contract_end_date=date(2027, 9, 30),
        payment_term=90,
        goods_id="G0001",
        goods_name="阿莫西林胶囊",
        reg_no="国药准字H20201234",
        spec="0.25g*24粒",
        unit="盒",
        qty=Decimal("10.0000"),
        unit_price=Decimal("100.0000"),
        purchaser_id="P001",
        purchaser_name="张三",
        planner_id="PL001",
        planner_name="李四",
        contract_entry_status=None,
        contract_no=None,
        process_status=0,
        org_code="901",
        goods_mdm_code="QX0100167583",
        supplier_mdm_code="DW0100046361",
    )
    data.update(overrides)
    return PurchaseContractRow(**data)


def test_protocol_rerun_is_not_claimable():
    row = _row(contract_entry_status=0, contract_no="HT-OLD")
    assert is_claimable(row) is False


def test_blank_contract_no_is_still_claimable():
    assert is_claimable(_row(contract_entry_status=0, contract_no="  ")) is True
    assert is_claimable(_row(contract_entry_status=0, contract_no="")) is True
    assert is_claimable(_row(contract_entry_status=0, contract_no=None)) is True


def test_processing_or_done_is_not_claimable():
    assert is_claimable(_row(process_status=3)) is False
    assert is_claimable(_row(process_status=1)) is False


def test_group_by_supplier_keeps_fid_order_and_first_header():
    rows = [
        _row(fid=2, supplier_id="B", payment_term=30, supplier_mdm_code="MDM-B"),
        _row(fid=1, supplier_id="A", payment_term=90, supplier_mdm_code="MDM-A"),
        _row(fid=3, supplier_id="A", payment_term=60, oms_id=2026100502, goods_id="G0002"),
    ]
    groups = group_by_supplier(rows)
    assert [g[0].supplier_id for g in groups] == ["A", "B"]
    assert [r.fid for r in groups[0]] == [1, 3]
    draft = draft_from_group(groups[0])
    assert draft.customer_code == "MDM-A"
    assert draft.payment_days == "90"
    assert draft.order_date == "2026-10-01"
    assert draft.contract_sign_date == "2026-10-01"
    assert draft.contract_start_date == "2026-10-02"
    assert draft.contract_end_date == "2027-09-30"
    assert draft.amount == "2000.0000"
    assert draft.source_type == ""
    assert draft.is_prepayment == "否"
    assert len(draft.goods_lines) == 2
    assert draft.goods_lines[0].goods_code == "QX0100167583"
    assert draft.goods_lines[0].operation_code == "QX0100167583"
    assert draft.goods_lines[0].mdm_code == "QX0100167583"
    assert draft.goods_lines[0].unit_price == "100.0000"
    assert draft.belong_year == "2026"


# 同供应商但货主或三个日期任一不同，不得收成同一张草稿。
def test_same_supplier_splits_when_org_or_dates_differ():
    rows = [
        _row(fid=1, supplier_id="A"),
        _row(fid=2, supplier_id="A", org_code="902", oms_id=2),
        _row(fid=3, supplier_id="A", contract_sign_date=date(2026, 11, 1), oms_id=3),
        _row(fid=4, supplier_id="A", contract_start_date=date(2026, 11, 2), oms_id=4),
        _row(fid=5, supplier_id="A", contract_end_date=date(2028, 1, 1), oms_id=5),
        _row(fid=6, supplier_id="A", oms_id=6, goods_id="G0009"),
    ]
    groups = group_by_supplier(rows)
    drafts = {
        tuple(row.fid for row in group): draft_from_group(group) for group in groups
    }
    assert len(groups) == 5
    assert (1, 6) in drafts
    assert drafts[(1, 6)].amount == "2000.0000"
    assert len(drafts[(1, 6)].goods_lines) == 2
    for fid in (2, 3, 4, 5):
        assert (fid,) in drafts
        assert len(drafts[(fid,)].goods_lines) == 1


def test_missing_mdm_reason():
    rows = [_row(supplier_mdm_code="")]
    assert missing_mdm_reason(rows) == "缺少供应商MDM码"
    assert missing_mdm_reason([_row(goods_mdm_code="")]) == "缺少货品MDM码"
    assert missing_mdm_reason([_row()]) is None


def test_money4():
    assert money4(Decimal("10.5")) == "10.5000"