python - 是否可以根据其他可选参数将 argparse 的可选参数设置为默认值？

Question

我正在使用 argparse 并且我有两个可选参数:

  parser.add_argument('-a', '--arg1', default=1, type=int)
  parser.add_argument('-b', '--arg2', action='store_true', default=False)

如果仅当 arg2 设置为 True 时，是否可以将 arg1 默认值设置为“1”？

换句话说，只有当 arg2 设置为 True 时，我才想执行以下操作:

  parser.add_argument('-a', '--arg1', default=1, type=int)

否则，它将被设置为:

  parser.add_argument('-a', '--arg1', type=int)

Answer 1

解析后进行测试是实现此目的的最简单方法:

if args.arg2 and args.arg1 is None:
    args.arg1 = 1

您可以将此类测试放入自定义操作中。您必须考虑发生的顺序(或不考虑)。我还没有弄清楚细节，但我无法想象它比这个解析后测试更简单。