Python, argparse: разные параметры с различным количеством аргументов

Question

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

sync set_version <build> <version_number> 
sync get_version <build> 
sync has_started <build_1> <build_2> ... <build_n> 

Я попытался с помощью subparsers argparse «s для каждого режима работы:

import argparse 

parser = argparse.ArgumentParser(description='Build synchronization mechanism') 
subparsers = parser.add_subparsers() 

parser_get_version = subparsers.add_parser('get_version') 
parser_get_version.add_argument('build') 

parser_update_version = subparsers.add_parser('update_version') 
parser_update_version.add_argument('build') 
parser_update_version.add_argument('version') 

args = parser.parse_args() 
print args 

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

usage: sync.py [-h] {get_version,update_version} ... 

Build synchronization mechanism 

positional arguments: 
    {get_version,update_version} 

optional arguments: 
    -h, --help   show this help message and exit 

Как заставить argparse отобразить полное сообщение справки, со всеми параметрами subparsers?

Answer 1

У каждой подкоманды есть своя собственная помощь; попробуйте sync.py get_version --help, например.

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

class HelpAction(argparse._HelpAction): 
    def __call__(self, parser, namespace, values, option_string=None): 
     parser.print_help() 

     for group in parser._subparsers._group_actions: 
      group.choices.values()[0].print_help() 

     parser.exit() 

parser = argparse.ArgumentParser(description='Build synchronization mechanism', 
           add_help=False) 
parser.add_argument('-h', '--help', action=HelpAction, default=argparse.SUPPRESS, 
    help=argparse._('show this help message and exit')) 

Возможно, вы захотите еще немного настроить выход.