2

我们有一个简单的 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?

最初,我对操作(检查、构建、部署)使用 argparse 子解析,并为类型使用另一个参数。

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 的方法?

干杯,维克多

4

1 回答 1

2

只需将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)
于 2011-08-20T22:26:11.910 回答