Python Argument Parser：支持无--的非必填多参数（伪Git脚本）

这事儿用Python标准库的argparse来处理就非常优雅，它专门用来构建命令行接口，完美适配你这种带子命令（init/add）且需要接收可变参数的场景。相比临时解决方法，它不仅代码结构清晰，还自带帮助文档、参数校验等功能，扩展性极强。

完整重构示例代码 #

import argparse
import os

def init_repo():
    """初始化伪Git仓库"""
    repo_dir = ".git"
    if not os.path.exists(repo_dir):
        os.makedirs(repo_dir)
        print(f"Initialized empty pseudo-Git repository in {os.path.abspath(repo_dir)}")
    else:
        print("Repository already initialized")

def add_files(filenames):
    """将指定文件添加到暂存区"""
    print(f"Staging files: {', '.join(filenames)}")
    # 这里可以添加实际的暂存逻辑，比如写入.git/index文件等
    # 示例：模拟记录暂存文件
    with open(".git/index", "a") as f:
        for fname in filenames:
            f.write(f"{fname}\n")

def main():
    # 创建主解析器
    parser = argparse.ArgumentParser(description="A simple pseudo-Git command-line tool")
    
    # 创建子命令解析器
    subparsers = parser.add_subparsers(dest="command", required=True, help="Available commands")

    # 配置init子命令
    init_parser = subparsers.add_parser("init", help="Initialize a new pseudo-Git repository")
    init_parser.set_defaults(func=init_repo)

    # 配置add子命令，接收多个文件名参数
    add_parser = subparsers.add_parser("add", help="Add files to the staging area")
    # nargs='+' 表示至少需要一个参数，支持接收多个文件名（包括路径）
    add_parser.add_argument(
        "filenames",
        nargs="+",
        help="One or more files/directories to add to the staging area"
    )
    add_parser.set_defaults(func=add_files)

    # 解析命令行参数
    args = parser.parse_args()

    # 根据子命令调用对应的处理函数
    if args.command == "add":
        # 把接收的文件名列表传给add_files函数
        args.func(args.filenames)
    else:
        args.func()

if __name__ == "__main__":
    main()

关键特性说明 #

• 子命令管理：通过add_subparsers创建init和add两个独立子命令，各自有自己的参数和逻辑，代码分层清晰。
• 可变参数接收：add_parser.add_argument里的nargs='+'是核心，它告诉解析器要接收一个或多个文件名参数，自动打包成列表传给add_files函数。
• 自动帮助文档：执行python script.py -h会显示所有可用命令，执行python script.py add -h会显示add命令的具体用法，无需手动编写帮助信息。
• 参数校验：如果用户执行python script.py add但不传入文件名，解析器会自动报错提示，省去了手动校验的麻烦。

使用示例 #

# 初始化仓库
python script.py init

# 添加单个文件
python script.py add README.md

# 添加多个文件或目录
python script.py add file1.txt src/utils.py docs/

这种方案不仅解决了add命令接收多参数的问题，还为后续扩展其他命令（比如commit、log）打下了良好基础，完全符合命令行工具的设计规范。

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