python - 如何动态选择方法调用？

Question

我有一个类似这样的代码:

if command == "print":
     foo_obj.print()

if command == "install":
     foo_obj.install()

if command == "remove":
     foo_obj.remove()

command 是一个字符串(我通过解析命令行参数来定义它，但这超出了重点)。有没有办法用类似的代码替换上面的代码块？

foo_obj.function(command)

对于记录我使用的是 Python 2.7

Answer 1

使用getattr 并调用其结果:

getattr(foo_obj, command)()

读作:

method = getattr(foo_obj, command)
method()

当然，从用户输入中获取 command 字符串时要小心。你最好用类似的东西检查命令是否被允许

command in {'print', 'install', 'remove'}