Модуль argparse Как добавить параметр без каких-либо аргументов? - PullRequest
132 голосов
/ 10 марта 2011

Я создал скрипт, используя argparse.

Сценарий должен принять имя файла конфигурации в качестве опции, и пользователь может указать, нужно ли ему полностью выполнить сценарий или только имитировать его.

Аргументы для передачи: ./script -f config_file -s или ./script -f config_file.

Это нормально для части -f config_file, но он продолжает спрашивать меня об аргументах для -s, который не является обязательным и не должен сопровождаться никакими.

Я пробовал это:

parser = argparse.ArgumentParser()
parser.add_argument('-f', '--file')
#parser.add_argument('-s', '--simulate', nargs = '0')
args = parser.parse_args()
if args.file:
    config_file = args.file
if args.set_in_prod:
        simulate = True
else:
    pass

Со следующими ошибками:

File "/usr/local/lib/python2.6/dist-packages/argparse.py", line 2169, in _get_nargs_pattern
nargs_pattern = '(-*%s-*)' % '-*'.join('A' * nargs)
TypeError: can't multiply sequence by non-int of type 'str'

И та же ошибка с '' вместо 0.

Ответы [ 2 ]

190 голосов
/ 11 марта 2011

Как @ Феликс Клинг предложил использовать action='store_true':

>>> from argparse import ArgumentParser
>>> p = ArgumentParser()
>>> _ = p.add_argument('-f', '--foo', action='store_true')
>>> args = p.parse_args()
>>> args.foo
False
>>> args = p.parse_args(['-f'])
>>> args.foo
True
80 голосов
/ 10 марта 2011

Чтобы создать параметр, который не требует значения, установите для action [документы] его значение 'store_const', 'store_true' или 'store_false'.

Пример:

parser.add_argument('-s', '--simulate', action='store_true')
...