argparse 可选位置参数？

小编典典

argparse 可选位置参数？

all

我有一个脚本，打算像这样使用： usage: installer.py dir [-h] [-v]

dir是一个位置参数，定义如下：

parser.add_argument('dir', default=os.getcwd())

我希望dir它是可选的：当它没有被指定时，它应该只是cwd.

不幸的是，当我没有指定dir参数时，我得到Error: Too few arguments.

阅读 136

2022-03-02

共1个答案

小编典典

使用nargs='?'（或者nargs='*'如果您需要多个目录）

parser.add_argument('dir', nargs='?', default=os.getcwd())

扩展示例：

>>> import os, argparse
>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('-v', action='store_true')
_StoreTrueAction(option_strings=['-v'], dest='v', nargs=0, const=True, default=False, type=None, choices=None, help=None, metavar=None)
>>> parser.add_argument('dir', nargs='?', default=os.getcwd())
_StoreAction(option_strings=[], dest='dir', nargs='?', const=None, default='/home/vinay', type=None, choices=None, help=None, metavar=None)
>>> parser.parse_args('somedir -v'.split())
Namespace(dir='somedir', v=True)
>>> parser.parse_args('-v'.split())
Namespace(dir='/home/vinay', v=True)
>>> parser.parse_args(''.split())
Namespace(dir='/home/vinay', v=False)
>>> parser.parse_args(['somedir'])
Namespace(dir='somedir', v=False)
>>> parser.parse_args('somedir -h -v'.split())
usage: [-h] [-v] [dir]

positional arguments:
  dir

optional arguments:
  -h, --help  show this help message and exit
  -v

2022-03-02