#!/usr/bin/env python3
import argparse

STAGES = {
    1: ("内部产品准备", [
        "智慧食堂标准产品、核心能力与非标边界手册",
        "政企与学校智慧食堂产品定位及销售标准讲解课件",
        "智慧食堂对外数据、客户案例及图片素材审核清单",
    ]),
    2: ("潜在客户接触", [
        "{customer}一页式产品介绍",
        "{customer}五页轻介绍",
        "智慧食堂两分钟产品介绍视频脚本与分镜",
    ]),
    3: ("销售机会判断", [
        "智慧食堂客户现状、需求与决策链访谈表",
        "智慧食堂标准方案、第三方接口与非标需求判断清单",
        "智慧食堂项目需求确认与后续行动纪要",
    ]),
    4: ("首次沟通与方案形成", [
        "{customer}首次沟通与标准解决方案",
        "智慧食堂业务场景、部署架构与软硬件产品配置清单",
        "智慧食堂客户价值测算与典型案例证明集",
    ]),
    5: ("产品验证与采购成交", [
        "政企与学校智慧食堂产品演示脚本、验证计划及结果报告",
        "智慧食堂项目客户报价、服务范围与专项费用说明",
        "智慧食堂渠道价格、内部控价与项目报价审批表",
        "智慧食堂项目技术方案、参数应答与投标证据资料包",
    ]),
    6: ("合同签订与项目交付", [
        "智慧食堂项目合同范围、产品配置与责任边界确认附件",
        "智慧食堂项目销售至交付信息移交与环境检查清单",
        "智慧食堂项目实施计划、用户培训、验收测试与服务保障方案",
    ]),
    7: ("客户运营与持续经营", [
        "智慧食堂项目上线运营月报与30、60、90天阶段复盘报告",
        "智慧食堂客户案例、续费维保、产品升级与站点扩容建议书",
    ]),
}

CUSTOMERS = {
    "government": "政企智慧营养健康食堂",
    "enterprise": "政企智慧营养健康食堂",
    "政企": "政企智慧营养健康食堂",
    "school": "校园食品安全与营养健康数智化平台",
    "学校": "校园食品安全与营养健康数智化平台",
}


def main():
    parser = argparse.ArgumentParser(description="推荐智慧食堂当前阶段的必要销售材料")
    parser.add_argument("--customer", required=True, choices=sorted(CUSTOMERS))
    parser.add_argument("--stage", required=True, type=int, choices=range(1, 8))
    parser.add_argument("--format", choices=["ppt", "docx", "xlsx", "html", "all"], default="all")
    args = parser.parse_args()
    customer = CUSTOMERS[args.customer]
    stage_name, materials = STAGES[args.stage]
    print(f"客户类型：{customer}")
    print(f"项目阶段：{args.stage}. {stage_name}")
    print(f"优先格式：{args.format}")
    print("必要材料：")
    for i, material in enumerate(materials, 1):
        print(f"{i}. {material.format(customer=customer)}")
    if args.stage <= 4:
        print("对外发送前：检查案例授权、数字口径和非标边界。")
    if args.stage >= 5:
        print("成交与交付阶段：确保配置、报价、合同和交付清单的名称、版本、数量一致。")


if __name__ == "__main__":
    main()