moduł argparse Jak dodać opcję bez argumentów?

165

Stworzyłem skrypt przy użyciu argparse.

Skrypt musi wybrać nazwę pliku konfiguracyjnego jako opcję, a użytkownik może określić, czy ma wykonać cały skrypt, czy tylko go symulować.

Argumenty do przekazania: ./script -f config_file -slub ./script -f config_file.

Jest ok dla części -f config_file, ale ciągle pyta mnie o argumenty dla -s, które są opcjonalne i nie powinny po nich następować.

Próbowałem tego:

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

Z następującymi błędami:

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'

I ten sam błąd z ''zamiast 0.

python argparse philippe
źródło

Odpowiedzi:

233

Jak sugerował @Felix Kling, użyj 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
źródło
97

Aby utworzyć opcję, która potrzebuje żadnej wartości, ustaw action [DOCS] z nim do 'store_const', 'store_true'lub 'store_false'.

Przykład:

parser.add_argument('-s', '--simulate', action='store_true')
Felix Kling
źródło
Witam, dziękuję za odpowiedź. Próbowałem, jak wyjaśniono w dokumencie: >>> parser.add_argument ('- foo', '-f', action = 'store_true') _StoreTrueAction (option_strings = ['- foo', '-f'], dest = 'foo', nargs = 0, const = True, default = False, type = None, choice = None, help = None, metavar = None) >>> parser.parse_args () Namespace (foo = False) >> > print args.foo Traceback (ostatnie wywołanie ostatnie): Plik "<stdin>", wiersz 1, w <module> AttributeError: Obiekt 'Namespace' nie ma atrybutu 'foo' >>> `To nie wydaje się działać; Nie wiem, jak pobrać wartość po przypisaniu. Dziękuję za pomoc!
philippe
@philippe: zapomniałeś powiązać parser.parse_args()zwróconą wartość ze args stackoverflow.com/questions/5262702/ ...
jfs
@philippe: args = parser.parse_args()jak już masz w swoim kodzie. Wystarczy zastąpić skomentowaną linię, a następnie uzyskać dostęp do wartości za pomocą args.simulate.
Felix Kling