Как argparse с nargs + и подкомандами

Question

Я пытаюсь создать команду, как

prog [-h] [-i ID [ID ...]] | -x [SOMETHING] 
    {cmd1,cmd2,cmd3}... 

Так в основном на верхнем уровне у меня есть синтаксический анализатор, который имеет взаимную EXLUSIVE группу для -i и -x , а затем следуя этим (и, возможно, другим) параметрам, у меня есть команда, которую я хочу запустить. Каждая команда имеет свой собственный набор опций, которые они используют. Я могу заставить команды работать нормально с add_subparsers(), но проблема, с которой я сталкиваюсь, - это когда я пытаюсь добавить аргумент в корневой парсер с nargs='+'. Когда я это делаю, он перекрывает все аргументы для -i, полагая, что команда является аргументом, а не идентификатором.

Есть ли способ обойти это? Похоже, что ему придется просмотреть аргументы в -i, ища командное слово, а затем сообщить argparse, что он должен возобновить разбор в этой точке.

Answer 1

я должен был прочитать ваше описание в несколько раз, но я думаю, что это проблема:

prog -i id1 id2 cmd1 -foo 3 .... 

и это дает какое-то предупреждение о не найдя {cmd1,cmd2,cmd3}. Точная ошибка может отличаться, поскольку в некоторых версиях subparsers на самом деле не требуется.

В любом случае, аргументы -i - это ['id1','id2','cmd1'], все до следующего -. Для основного анализатора аргумент подпараметров - это еще один номер positional (с choices). При распределении строк до -i он не проверяет соответствие строки одной из cmds. Он просто смотрит, начинается ли это с - или нет.

Единственный способ, которым вы можете использовать nargs='+' (или '*') в контексте, - включить в него другой флаг, например,

prog -i id1 id2 -x 3 cmd1 --foo ... 

Я понимаю, что идет против вашей группы mutually_exclusive.

Основной момент - это не-флаги, которые выделяются в зависимости от позиции, а не значения. Для переменной nargs у вас должен быть какой-то явный ограничитель списка.

На боковой панели

Argparse nargs="+" is eating positional argument

Это аналогично, за исключением, что ваш следующий позиционная является subparsers CMD.

==============

Позиционная с '+' будет работать прямо перед subparsers CMD

usage: prog [-h] foo [foo ...] {cmd1,cmd2} ... 

In [160]: p1.parse_args('1 22 3 cmd1'.split()) 
Out[160]: Namespace(cmd='cmd1', foo=['1', '22', '3']) 

Но это потому, что строки для foo и cmd выделяются одним шаблоном regex.

В

usage: prog [-h] [--bar BAR [BAR ...]] {cmd1,cmd2} ... 

строки выделяются bar без ссылки на потребности следующих позиционной, cmd. Как показано в предложенных исправлениях для http://bugs.python.org/issue9338, изменение этого поведения не является тривиальным изменением. Для этого требуется дополнительный цикл проб и ошибок.