jak uczynić argument opcjonalnym w python argparse

Question

Chciałbym dokonać tych wywołań pracy myprog, a nie innych.

$ python3 myprog.py -i infile -o outfile 
$ python3 myprog.py -o outfile 
$ python3 myprog.py -o 
$ python3 myprog.py 

W szczególności chcę, aby to nielegalne określić plik_we ale nie plik_wy.

W trzecim przypadku przyjmuje się domyślną nazwę pliku wyjściowego "out.json". W drugim, trzecim i czwartym przypadku przyjmuje się domyślną nazwę pliku wejściowego "file.n.json", gdzie n jest liczbą całkowitą wersji. W czwartym przypadku plikiem wyjściowym będzie "file.n + 1.json", gdzie n + 1 jest wersją numer jeden większą niż ta na pliku wejściowym. Odpowiedni fragment mojego kodu jest:

import argparse 

parser = argparse.ArgumentParser(description="first python version") 
parser.add_argument('-i', '--infile', nargs=1, type=argparse.FileType('r'), help='input file, in JSON format') 
parser.add_argument('-o', '--outfile', nargs='?', type=argparse.FileType('w'), default='out.json', help='output file, in JSON format') 

args = parser.parse_args() 

print("Here's what we saw on the command line: ") 
print("args.infile",args.infile) 
print("args.outfile",args.outfile) 

if args.infile and not args.outfile: 
    parser.error("dont specify an infile without specifying an outfile") 
elif not args.infile: 
    print("fetching infile") 
else: # neither was specified on the command line 
    print("fetching both infile and outfile") 

Problem jest, gdy biegnę

$ python3 myprog.py -i infile.json 

zamiast błędu parsera się spodziewałem, otrzymuję:

Here's what we saw on the command line: 
args.infile [<_io.TextIOWrapper name='infile.json' mode='r' encoding='UTF-8'>] 
args.outfile <_io.TextIOWrapper name='out.json' mode='w' encoding='UTF-8'> 
fetching both infile and outfile 

... co sugeruje, że nawet jeśli nie było "-o" na linii poleceń, działało to tak, jakby było.

Answer 1

Podałeś domyślny argument dla pliku wyjściowego.

parser.add_argument('-o', '--outfile', nargs='?', type=argparse.FileType('w'), default='out.json', help='output file, in JSON format') 

Jeśli opcja -o nie jest określony w linii poleceń, parser arg wstawia domyślny argument.

Zmień to:

parser.add_argument('-o', '--outfile', nargs='?', type=argparse.FileType('w'), help='output file, in JSON format') 

i wszystko powinno działać zgodnie z oczekiwaniami.

---

Jeśli chcesz być w stanie określić -o, bez nazwy pliku, prawdopodobnie chcesz coś takiego:

out_file = args.out if args.out is not None else 'json.out'

Nie jestem pewien, czy istotnym parametrem będzie None lub '' (tj. pusty ciąg), jeśli podasz -o bez parametru - podejrzewam, że jest to None, ale nie wiem na pewno. Musisz to przetestować, żeby się upewnić.

Nie wiem jak to zrobić bez dodatkowej logiki z argparse.

Answer 2

Jako dodatek do wybranego odpowiedź:

możliwości uruchomienia -o bez określenia pliku, można to zrobić za pomocą const połączeniu z nargs='?'.

Od docs:

Kiedy add_argument() jest wywoływana z ciągów opcji (jak -f lub --foo) i nargs = '?'. Spowoduje to utworzenie opcjonalnego argumentu, który można wykonać za pomocą zerowego lub jednego argumentu wiersza polecenia: .Podczas analizowania wiersza polecenia, , jeśli napotkano łańcuch opcji, bez następującego po nim argumentu wiersza poleceń, zostanie przyjęta wartość const. Zapoznaj się z opisami na temat nargs z .