from __future__ import annotations

READINESS_STAGE_RULES: dict[str, list[str]] = {
    "contracting": [
        "quyet_dinh_phe_duyet_kqlcnt",
        "bien_ban_thuong_thao_hop_dong",
        "hop_dong_thi_cong",
    ],
    "payment": [
        "hop_dong_thi_cong",
        "bien_ban_nghiem_thu_cong_viec",
        "bang_xac_nhan_khoi_luong",
        "ho_so_thanh_toan",
    ],
    "finalization": [
        "hop_dong_thi_cong",
        "ho_so_quan_ly_chat_luong",
        "bien_ban_nghiem_thu_hoan_thanh",
        "ho_so_hoan_cong",
        "ho_so_quyet_toan_tong_hop",
        "bien_ban_thanh_ly_hop_dong",
    ],
}


def get_readiness_rules() -> dict[str, list[str]]:
    """Return required document types for each readiness evaluation."""
    return READINESS_STAGE_RULES