时间戳

首页 > 解决方案 > 带有选择选项的条件 argparse

python - 带有选择选项的条件 argparse

问题描述

以下是我在模块中编写的三个参数。

parser.add_argument('--Type',type=str,choices=['a','b','c'],help='Options include: a,b,c.',required=True)

parser.add_argument('--Input',default=False,help='Generate input files',required=False)

parser.add_argument('--Directory',default=False,help='Secondary directory',required='--Input' in sys.argv)

--Type可以使用三个选项：a、b、c 。

目前，我已将其设置为，如果 --Directory 为真，则需要 --Input 为真。

但是，我想向 --Directory 添加一个附加条件以要求 --Type 为 == 'c'。

如何更改 --Directory 参数中的必需选项，使其同时需要 --Input 和 --Type == 'c'？

标签： pythonconditionalargparse

解决方案

将参数解析与您的需求分离。

parser.add_argument('--Type', choices=['a','b','c'], required=True)
parser.add_argument('--Input', action='store_true')
parser.add_argument('--Directory', action='store_true')

args = parser.parse_args()

if args.Directory and args.Type != 'c' and not args.input:
    raise argparse.ArgumentError("--Directory requires --Type c and --Input")

（注意action='store_true'自动设置type=bool和default=False。）

推荐阅读