如何在带CLI的Python脚本中检查输入参数并解决main函数参数校验问题

我正在编写一个用于学习Python的小脚本，功能是为N名玩家生成棋类锦标赛对阵表，它带有一个仅含参数N的简单CLI。当前的实现代码如下：

import argparse

def parse_args(argv: list[str] | None = None) -> int:
    parser = argparse.ArgumentParser(description="Tournament tables")
    parser.add_argument('N', help="number of players (2 at least)", type=int)
    args = parser.parse_args(argv)
    if args.N < 2:
        parser.error("N must be 2 at least")
    return args.N

def main(n: int) -> None:
    print(F"Here will be the table for {n} players")

if __name__ == '__main__':
    main(parse_args())

但这个实现存在缺陷：main函数未对参数n进行有效性检查（原本由CLI解析器负责），若其他模块（如测试模块）直接调用main并传入无效值（如0），程序很可能崩溃。我考虑了几种解决方案，但不确定哪种最优：

• 在main中添加参数校验和错误处理，不过这违反DRY原则，重复了CLI的工作；
• 仅在文档中说明main仅接受n>=2，否则行为不可预测，可配合添加断言assert n >= 2, "n must be 2 or more"；
• 或许不应将main设为外部可调用函数，需重写脚本入口；
• 其他可行方案？

各方案分析与推荐 #

方案2（文档说明+断言）

这是学习场景下最实用的方案。断言在开发、测试阶段能立刻揪出非法输入，不会给后续可能的生产环境带来额外性能开销，同时通过文档明确函数的前置条件，契合Python“显式优于隐式”的设计哲学。而且完全不会重复校验逻辑，避开了DRY问题。

方案1（重复校验）

不推荐。重复的校验逻辑会提升维护成本——比如以后如果要求N必须是偶数，你得同时修改parse_args和main两个地方，很容易出现遗漏或不一致的情况。

方案3（限制main的可调用性）

可以尝试，但会降低代码灵活性。如果后续需要在其他模块复用main的核心逻辑，这种限制会造成不必要的麻烦。学习阶段还是尽量让代码保留一定的可复用性更有意义。

其他可行方案：抽离独立校验函数

把参数校验逻辑单独封装成一个函数，在parse_args和main中统一调用。这样既遵守DRY原则，又能保证所有调用入口都做了参数校验，是更专业的实现方式。示例代码如下：

import argparse

def validate_player_count(n: int) -> None:
    if n < 2:
        raise ValueError("n must be 2 or more")

def parse_args(argv: list[str] | None = None) -> int:
    parser = argparse.ArgumentParser(description="Tournament tables")
    parser.add_argument('N', help="number of players (2 at least)", type=int)
    args = parser.parse_args(argv)
    validate_player_count(args.N)
    return args.N

def main(n: int) -> None:
    validate_player_count(n)
    print(f"Here will be the table for {n} players")

if __name__ == '__main__':
    main(parse_args())

内容的提问来源于stack exchange，提问作者Phant