Группа взаимных исключений Python argparse

88

Что мне нужно:

pro [-a xxx | [-b yyy -c zzz]]

Я пробовал это, но не работает. Может ли кто-нибудь мне помочь?

group= parser.add_argument_group('Model 2')
group_ex = group.add_mutually_exclusive_group()
group_ex.add_argument("-a", type=str, action = "store", default = "", help="test")
group_ex_2 = group_ex.add_argument_group("option 2")
group_ex_2.add_argument("-b", type=str, action = "store", default = "", help="test")
group_ex_2.add_argument("-c", type=str, action = "store", default = "", help="test")

Благодарность!

Шон
источник
Заткнул, но я хотел упомянуть свою библиотеку joffrey . Позволяет вам делать то, что требует этот вопрос, например, не заставляя вас использовать подкоманды (как в принятом ответе) или проверять все самостоятельно (как в ответе, получившем второе место по количеству голосов).
Алло

Ответы:

109

add_mutually_exclusive_groupне делает всю группу взаимоисключающей. Это делает варианты внутри группы взаимоисключающими.

Вам нужны подкоманды . Вместо проги [-a xxxx | [-b yyy -c zzz]], у вас будет:

prog 
  command 1 
    -a: ...
  command 2
    -b: ...
    -c: ...

Чтобы вызвать с первым набором аргументов:

prog command_1 -a xxxx

Чтобы вызвать со вторым набором аргументов:

prog command_2 -b yyyy -c zzzz

Вы также можете установить аргументы подкоманды как позиционные.

prog command_1 xxxx

Что-то вроде git или svn:

git commit -am
git merge develop

Рабочий пример

# create the top-level parser
parser = argparse.ArgumentParser(prog='PROG')
parser.add_argument('--foo', action='store_true', help='help for foo arg.')
subparsers = parser.add_subparsers(help='help for subcommand')

# create the parser for the "command_1" command
parser_a = subparsers.add_parser('command_1', help='command_1 help')
parser_a.add_argument('a', type=str, help='help for bar, positional')

# create the parser for the "command_2" command
parser_b = subparsers.add_parser('command_2', help='help for command_2')
parser_b.add_argument('-b', type=str, help='help for b')
parser_b.add_argument('-c', type=str, action='store', default='', help='test')

Попробуй это

>>> parser.print_help()
usage: PROG [-h] [--foo] {command_1,command_2} ...

positional arguments:
  {command_1,command_2}
                        help for subcommand
    command_1           command_1 help
    command_2           help for command_2

optional arguments:
  -h, --help            show this help message and exit
  --foo                 help for foo arg.
>>>

>>> parser.parse_args(['command_1', 'working'])
Namespace(a='working', foo=False)
>>> parser.parse_args(['command_1', 'wellness', '-b x'])
usage: PROG [-h] [--foo] {command_1,command_2} ...
PROG: error: unrecognized arguments: -b x

Удачи.

Джонатан
источник
Я уже поместил их в группу аргументов. Как я могу добавить подкоманду в этом случае? Благодарность!
Шон
1
Обновлено с помощью образца кода. Вы не будете использовать группы, а будете использовать субпарсеры.
Джонатан
6
Но как бы вы сделали то, о чем первоначально просил OP? В настоящее время у меня есть набор [[-a <val>] | [-b <val1> -c <val2>]]
подкоманд
2
Это не отвечает на вопрос, потому что не позволяет вам выполнять команды noname и выполнять то, о чем просил OP[-a xxx | [-b yyy -c zzz]]
Крестный отец
34

Хотя ответ Джонатана отлично подходит для сложных вариантов, существует очень простое решение, которое будет работать для простых случаев, например, 1 вариант исключает 2 других варианта, например, в

command [- a xxx | [ -b yyy | -c zzz ]] 

или даже как в исходном вопросе:

pro [-a xxx | [-b yyy -c zzz]]

Вот как бы я это сделал:

parser = argparse.ArgumentParser()

# group 1 
parser.add_argument("-q", "--query", help="query", required=False)
parser.add_argument("-f", "--fields", help="field names", required=False)

# group 2 
parser.add_argument("-a", "--aggregation", help="aggregation",
                    required=False)

Я использую здесь параметры, данные оболочке командной строки для запроса mongodb. collectionЭкземпляр может либо вызвать метод aggregateили метод findс для дополнительных аргументов queryи fields, следовательно , вы видите , почему первые два аргумента является совместимым и последним не является.

Итак, теперь я бегу parser.parse_args()и проверяю его содержимое:

args = parser().parse_args()

print args.aggregation
if args.aggregation and (args.query or args.fields):
    print "-a and -q|-f are mutually exclusive ..."
    sys.exit(2)

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

Oz123
источник
5
Я бы не назвал это «хаком» в данном случае, поскольку он кажется более читаемым и управляемым - спасибо, что указали на это!
sage
13
Еще лучше было бы использовать parser.error("-a and -q ..."). Таким образом, полная справка по использованию будет распечатана автоматически.
WGH
Обратите внимание, что в этом случае вам также необходимо будет проверить такие случаи, как: (1) оба qи fтребуются в первой группе - это пользователь, (2) требуется любая из групп. И это делает «простое» решение уже не таким простым. Так что я согласен, что это скорее взлом для написанного вручную сценария, но не реальное решение
Крестный отец
4

Существует патч для Python (в разработке), который позволит вам это сделать.
http://bugs.python.org/issue10984

Идея состоит в том, чтобы разрешить перекрывающиеся взаимоисключающие группы. Так usageможет выглядеть:

pro [-a xxx | -b yyy] [-a xxx | -c zzz]

Изменение кода argparse таким образом, чтобы вы могли создать две группы, как это, было несложной частью. Для изменения usageкода форматирования потребовалось написать собственный HelpFormatter.

В argparse, группы действий не влияют на синтаксический анализ. Они всего лишь helpинструмент форматирования. Во helpвзаимоисключающих группах влияет только на usageстроку. При синтаксическом анализе parserиспользует взаимоисключающие группы для создания словаря потенциальных конфликтов ( aне может возникнуть bили c, bне может возникнуть aи т.д.), а затем выдает ошибку, если возникает конфликт.

Без этого argparse патча, я думаю , ваш лучший выбор , чтобы проверить пространство имен произведенных parse_argsсамостоятельно (например , если оба aи bимеют нестандартные значения), а также повысить свою собственную ошибку. Вы даже можете использовать собственный механизм ошибок парсера.

parser.error('custom error message')
hpaulj
источник
1
Проблема с Python: bugs.python.org/issue11588 изучает способы, позволяющие вам писать собственные эксклюзивные / инклюзивные тесты.
hpaulj 07