Argparse Subparsers и ссылки на классы - PullRequest
Новая
       29

Argparse Subparsers и ссылки на классы

2 голосов
/ 18 июля 2011

У нас есть простая программа на Python для управления различными типами внутренних серверов с использованием argparse: 

manage_servers.py <operation> <type_of_server>

Операции - это проверка, сборка, развертывание, настройка, проверка и т. Д.

Типы серверов - это просто разные типы внутренних серверов, которые мы используем.

У нас есть общий класс сервера, а затем определенные типы, которые наследуются от него: 

class Server
    def configure_logging(self, loggin_file):
        ...
    def check(self):
        ...
    def deploy(self):
        ...
    def configure(self):
        ...
    def __init__(self, hostname):
        self.hostname = hostname
        logging = self.configure_logging(LOG_FILENAME)
class SpamServer(Server):
    def check(self):
        ...
class HamServer(Server):
    def deploy(self):
        ...

У меня вопрос, как связать все это с argparse?

Первоначально я использовал arparse subparses для операций (check, build, deploy) и другой аргумент для типа. 

subparsers = parser.add_subparsers(help='The operation that you want to run on the server.')
parser_check = subparsers.add_parser('check', help='Check that the server has been setup correctly.')
parser_build = subparsers.add_parser('build', help='Download and build a copy of the execution stack.')
parser_build.add_argument('-r', '--revision', help='SVN revision to build from.')
...
parser.add_argument('type_of_server', action='store', choices=types_of_servers,
                    help='The type of server you wish to create.')

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

Любые идеи о том, как я мог бы достичь выше? Возможно, другой шаблон дизайна для серверов? Или способ использовать argparse как есть?

Ура, Victor

1 Ответ

2 голосов
/ 21 августа 2011

Просто используйте аргумент parser.add_subparsers(dest=... с отображением type_of_server в классы: 

subparsers = parser.add_subparsers(dest='operation', help='The operation that you want to run on the server.')
...
server_types = dict(spam=SpamServer, ham=HamServer)
args = parser.parse_args()
server = server_types[args.type_of_server]()
getattr(server, args.operation)(args)
...