[python经验]argparse的互斥参数等小技巧

argparse常见问题 - 互斥参数

问题

--items 和 --all-items 必有其一：required参数

• --items 可以指定1个或者多个item：nargs参数
• --items 不要硬编码：用list间接指定
• --items 忽略大小写：用type参数直接转换为小写
• --all-items 对应boolean数据：用action参数实现出现即为True，缺席为False
• --all-items 用--items的default参数实现all-items为True选择全部items

实现

items = [
    'audio',
    'video',
    'network',
]
parser = argparse.ArgumentParser()
items_group = parser.add_mutually_exclusive_group(required=True)
items_group.add_argument('--items', nargs='+', type=str.lower,
                         choices=items, default=items, 
                         help='available items: ' + ', '.join(items))
items_group.add_argument('--all-items', action='store_true',
                         default=False, help='all items')
args = parser.parse_args()
print(args)

测试一下

我不写expect了，读者自己想想UT怎么写，然后动手测试一下：

1. 测试用例：没有参数
2. 测试用例：两个参数--items audio --all-items
3. 测试用例：1个参数--all-items
4. 测试用例：1个item参数--items audio
5. 测试用例：2个item参数--items audio video
6. 测试用例：3个item参数--items audio video network
7. 测试用例：item参数的大小写--items audio Video NETWORK
8. 测试用例：不合法的item参数--items invallable_item
9. 测试用例：不合法的item参数--items audio invallable_item

好，先到这里。