módulo argparse Como adicionar uma opção sem nenhum argumento?

165

Eu criei um script usando argparse.

O script precisa usar um nome de arquivo de configuração como opção, e o usuário pode especificar se precisa continuar totalmente o script ou apenas simulá-lo.

Os argumentos a serem passados: ./script -f config_file -sou ./script -f config_file.

Está ok para a parte -f config_file, mas continua me pedindo argumentos para o -s, que é opcional e não deve ser seguido por nenhum.

Eu tentei isso:

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

Com os seguintes erros:

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'

E mesmo erro com em ''vez de 0.

philippe
fonte

Respostas:

232

Como o @Felix Kling sugeriu o uso 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
jfs
fonte
96

Para criar uma opção que não precisa de valor, defina os action [documentos] para 'store_const', 'store_true'ou 'store_false'.

Exemplo:

parser.add_argument('-s', '--simulate', action='store_true')
Felix Kling
fonte
Olá. Obrigado pela sua resposta. Tentei conforme explicado no documento: >>> parser.add_argument ('- foo', '-f', action = 'store_true') _StoreTrueAction (option_strings = ['- foo', '-f'], dest = 'foo', nargs = 0, const = True, padrão = False, tipo = None, escolhas = None, help = None, metavar = None) >>> espaço para nome parser.parse_args () (foo = False) >> > print args.foo Traceback (última chamada mais recente): Arquivo "<stdin>", linha 1, em <module> AttributeError: o objeto 'Namespace' não possui atributo 'foo' >>> `Isso parece não funcionar; Não sei como recuperar o valor após a atribuição. Obrigado por me ajudar!
31411 philippe
@philippe: você se esqueceu de ligamento parser.parse_args()valor retornado para args stackoverflow.com/questions/5262702/...
jfs
@ philippe: args = parser.parse_args()como você já possui em seu código. Você apenas precisa substituir a linha comentada e acessar o valor com args.simulate.
precisa