如何让argparse支持通配符及正则表达式批量指定skip类命令行选项？

核心思路 #

argparse本身不支持直接解析通配符或正则式风格的选项，所以需要先拦截原始命令行参数，将通配符参数转换为对应的具体选项，再交给argparse处理。

具体实现代码 #

import sys
import argparse
import re

def main():
    # 提前定义所有skip选项的完整名称
    skip_options = ['--skip-part1', '--skip-part2', '--skip-part3', '--skip-part4']
    
    # 处理原始命令行参数，替换通配符/正则式参数
    new_args = []
    for arg in sys.argv[1:]:
        # 识别需要处理的skip类通配符参数
        if arg.startswith('--skip-') and ('*' in arg or '[' in arg):
            # 将shell风格通配符转为正则表达式：先转义特殊字符，再还原*和[]的正则语义
            regex_pattern = re.escape(arg).replace(r'\*', '.*').replace(r'\[', '[').replace(r'\]', ']')
            # 匹配所有符合模式的skip选项
            matched_opts = [opt for opt in skip_options if re.fullmatch(regex_pattern, opt)]
            if not matched_opts:
                raise ValueError(f"错误：没有匹配到任何skip选项，模式为 '{arg}'")
            new_args.extend(matched_opts)
        else:
            new_args.append(arg)
    
    # 初始化argparse解析器
    parser = argparse.ArgumentParser(description="带skip选项的命令行工具")
    # 添加所有skip布尔型选项
    parser.add_argument('--skip-part1', action='store_true', help="跳过第1部分")
    parser.add_argument('--skip-part2', action='store_true', help="跳过第2部分")
    parser.add_argument('--skip-part3', action='store_true', help="跳过第3部分")
    parser.add_argument('--skip-part4', action='store_true', help="跳过第4部分")
    # 示例其他选项
    parser.add_argument('--other-option', help="其他演示选项")
    
    # 用处理后的参数执行解析
    args = parser.parse_args(new_args)
    
    # 输出解析结果验证
    print("解析后的参数：")
    print(f"skip-part1: {args.skip_part1}")
    print(f"skip-part2: {args.skip_part2}")
    print(f"skip-part3: {args.skip_part3}")
    print(f"skip-part4: {args.skip_part4}")
    if args.other_option:
        print(f"other-option: {args.other_option}")

if __name__ == "__main__":
    main()

使用说明 #

1. 选中所有skip选项：
   必须用引号包裹通配符，避免shell自动展开为文件列表：

   python myscript.py "--skip-*"
   

   执行后所有--skip-part1到--skip-part4都会被设为True。

2. 选中前三个skip选项：
   同样用引号包裹正则式风格参数：

   python myscript.py "--skip-part[1-3]"
   

   此时--skip-part1、--skip-part2、--skip-part3会被设为True，--skip-part4保持False。

3. 混合使用场景：
   可以同时搭配通配符和普通选项：

   python myscript.py "--skip-part[1-2]" --skip-part4 --other-option "测试内容"
   

注意事项 #

• 含*或[]的参数必须加引号，否则shell会将其解析为文件系统通配符，导致参数被替换为当前目录下的文件名，而非传递给脚本处理。
• 如果输入的模式没有匹配到任何skip选项，脚本会抛出错误提示，你可以根据需求调整错误逻辑（比如忽略该参数或输出更友好的提示）。

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