from __future__ import annotations

from typing import Any

from .stage_requirements_p1 import CONTRACTING_REQUIREMENTS, PRE_AWARD_REQUIREMENTS
from .stage_requirements_p2 import (
    EXECUTION_REQUIREMENTS,
    FINALIZATION_REQUIREMENTS,
    PAYMENT_REQUIREMENTS,
)

DOC_STAGE_REQUIREMENTS: dict[str, list[dict[str, Any]]] = {
    "pre_award": PRE_AWARD_REQUIREMENTS,
    "contracting": CONTRACTING_REQUIREMENTS,
    "execution": EXECUTION_REQUIREMENTS,
    "payment": PAYMENT_REQUIREMENTS,
    "finalization": FINALIZATION_REQUIREMENTS,
}


def get_stage_requirements() -> dict[str, list[dict[str, Any]]]:
    """Return checklist requirements by dossier stage."""
    return DOC_STAGE_REQUIREMENTS