Как проверить общее количество переданных аргументов?

Question

При использовании argparse, мне нужно убедиться, что пользователь вводит две строки. Однако есть два возможных варианта для ввода строк в, например. варианты -a и -b. Они должны быть в состоянии сделать любое из следующих действий:

python runscript.py -a str1 str2 
python runscript.py -a str1 -b str2 
python runscript.py -b str1 str2 

Ключ должен быть только два входа, но они могут быть в любой из доступных вариантов.

Моя попытка в основном устанавливает аргументы как этот

parser.add_argument('-a', nargs='*') 
parser.add_argument('-b', nargs='*') 

Тогда я должен сделать это:

if (args.a): num_a = len(args.a) # I need to check if the -a option was even used 
else: num_a = 0 

if (args.b): num_b = len(args.b) # same as above but for -b option 
else: num_b = 0 

# Check that number of arguments is correct 
if (num_a + num_b) != 2: 
    <error code> 

Это кажется немного постороннее. Есть ли более питонический способ сделать это?

Answer 1

Дайте оба аргумента по умолчанию пустое значение списка:

parser.add_argument('-a', nargs='*', default=[]) 
parser.add_argument('-b', nargs='*', default=[]) 

Теперь значение является всегда будет список, упрощая проверку ошибок:

if len(args.a) + len(args.b) != 2: 
    parser.error('....')