Поддерживает ли argparse (python) взаимоисключающие группы аргументов?

Если у меня есть аргументы '-a', '-b', '-c', '-d', то с помощью функции add_mutually_exclusive_group() моя программа должна будет использовать только один из них. Есть ли способ объединить это, так что программа будет принимать только '-a 999 -b 999' или '-c 999 -d 999'?

Изменить: добавление простой программы для большей ясности:

>>> parser = argparse.ArgumentParser()
>>> group = parser.add_mutually_exclusive_group()
>>> group.add_argument('-a')
>>> group.add_argument('-b')
>>> group.add_argument('-c')
>>> group.add_argument('-d')

Тогда можно вызвать только ./app.py -a | ./app.py -b | ./app.py -c | ./app.py -d. Возможно ли иметь группу argparse для групп исключения, так что называть только ./app.py -a .. -b .. | ./app.py -c .. -d ..?

Ответы

Ответ 1

ИЗМЕНИТЬ: Ничего. Потому что argparse делает ужасный выбор необходимости создания опции при вызове group.add_argument. Это был бы не мой выбор дизайна. Если вы отчаянно нуждаетесь в этой функции, вы можете попробовать сделать это с помощью ConflictsOptionParser:

# exclusivegroups.py
import conflictsparse

parser = conflictsparse.ConflictsOptionParser()
a_opt = parser.add_option('-a')
b_opt = parser.add_option('-b')
c_opt = parser.add_option('-c')
d_opt = parser.add_option('-d')

import itertools
compatible_opts1 = (a_opt, b_opt)
compatible_opts2 = (c_opt, d_opt)
exclusives = itertools.product(compatible_opts1, compatible_opts2)
for exclusive_grp in exclusives:
    parser.register_conflict(exclusive_grp)


opts, args = parser.parse_args()
print "opts: ", opts
print "args: ", args

Таким образом, когда мы вызываем его, мы видим, что мы получаем желаемый эффект.

$ python exclusivegroups.py -a 1 -b 2
opts:  {'a': '1', 'c': None, 'b': '2', 'd': None}
args:  []
$ python exclusivegroups.py -c 3 -d 2
opts:  {'a': None, 'c': '3', 'b': None, 'd': '2'}
args:  []
$ python exclusivegroups.py -a 1 -b 2 -c 3
Usage: exclusivegroups.py [options]

exclusivegroups.py: error: -b, -c are incompatible options.

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

OLDER EDIT: [Это изменение неверно, хотя это был бы просто идеальный мир, если argparse работал таким образом?] Мой предыдущий ответ на самом деле был неправильным, вы должны быть способны сделайте это с помощью argparse, указав одну группу на взаимоисключающие параметры. Мы можем даже использовать itertools для обобщения процесса. И сделайте так, что нам не нужно явно вводить все комбинации:

import itertools
compatible_opts1 = ('-a', '-b')
compatible_opts2 = ('-c', '-d')
exclusives = itertools.product(compatible_opts1, compatible_opts2)
for exclusive_grp in exclusives:
    group = parser.add_mutually_exclusive_group()
    group.add_argument(exclusive_grp[0])
    group.add_argument(exclusive_grp[1])

Ответ 2

Просто наткнулся на эту проблему сам. Из моего чтения документов argparse, похоже, нет простого способа добиться этого в argparse. Я рассмотрел использование parse_known_args, но вскоре это приведет к написанию специальной версии argparse;-)

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

    subparsers = parser.add_subparsers()
    p_ab = subparsers.add_parser('ab')
    p_ab.add_argument(...)

    p_cd = subparsers.add_parser('cd')
    p_cd.add_argument(...)

Не идеален, но, по крайней мере, он дает вам хорошее от argparse без слишком уродливого хакера. Я закончил работу с коммутаторами и просто использовал операции subparser с необходимыми субаргументами.