Добавить аргументы в argparse в зависимости от значения начальных аргументов

Question

Я знаю, что подобные вопросы задавали раньше, но я не мог найти точного (также я не мог понять концепцию «nargs» или «subparse» и как это было бы обратиться к моему делу)

Я хочу что-то вроде:

parser = argparse.ArgumentParser() 
parser.add_argument('-mode', choices=['download', 'upload'], required=True) 
parser.add_argument('-d', required=True) 
args = parser.parse_args() 
if args.mode == 'download': 
    parser.add_argument('-f', required=True) 

args = parser.parse_args() 

так требуется -f аргумент только тогда, когда -mode является download иначе оно не должно быть разобрано

Answer 1

Что об этом?

import argparse 

parser = argparse.ArgumentParser() 
parser.add_argument('-mode', choices=['download', 'upload'], required=True) 
parser.add_argument('-d', required=True) 
parser.add_argument('-f') 

args = parser.parse_args() 
if args.mode == 'download' and not args.f: 
    parser.error('-f argument is required in "download" mode.') 

DEMO:

$ python test.py -mode=upload -d 10 
$ python test.py -mode=download -d 10 
usage: test.py [-h] -mode {download,upload} -d D [-f F] 
test.py: error: -f argument is required in "download" mode. 
$ python test.py -mode=download -d 10 -f 10 

Answer 2

Как правило, вы должны избегать --options требуется. В этом случае я бы предложил следующее:

1. Заменить --mode с подкомандой.
2. Заменить -d с позиционным аргументом
3. Заменить -f с 2-позиционным аргументом в пользу только download команды.

Если вы действительно этого хотите, вы можете оставить -d и -f вместо того, чтобы делать им позиционные аргументы.

Ваши звонки будут выглядеть следующим образом:

./script.py upload d-argument 
./script.py download d-argument f-argument 

или

./script.py upload -d foo 
./script.py download -d foo -f bar 

Код для реализации немного утомительно, но довольно просто.

parser = argparse.ArgumentParser() 
subparsers = parser.add_subparsers() 
upload_parser = subparsers.add_parser('upload') 
download_parser = subparsers.add_parser('download') 
upload_parser.set_defaults(cmd='upload') 
download_parser.set_defaults(cmd='download') 
for p in [ upload_parser, download_parser ]: 
    p.add_argument("d") 
    # p.add_argument("-d", required=True) 
download_parser.add_argument("f") 
# download_parser.add_argument("-f", required=True) 


args = parser.parse_args() 
if args.cmd == 'upload': 
    # handle upload with args.d 
elif args.cmd == 'download': 
    # handle download with args.d and args.f 

Обратите внимание на вызовы set_defaults, используемые для обозначения в конечных проанализированных аргументов, команда была использована.