8.20 通过字符串调用对象方法

问题

你有一个字符串形式的方法名称,想通过它调用某个对象的对应方法。

解决方案

最简单的情况,可以使用 getattr()

  1. import math
  2.  
  3. class Point:
  4. def __init__(self, x, y):
  5. self.x = x
  6. self.y = y
  7.  
  8. def __repr__(self):
  9. return 'Point({!r:},{!r:})'.format(self.x, self.y)
  10.  
  11. def distance(self, x, y):
  12. return math.hypot(self.x - x, self.y - y)
  13.  
  14.  
  15. p = Point(2, 3)
  16. d = getattr(p, 'distance')(0, 0) # Calls p.distance(0, 0)

另外一种方法是使用 operator.methodcaller() ,例如:

  1. import operator
  2. operator.methodcaller('distance', 0, 0)(p)

当你需要通过相同的参数多次调用某个方法时,使用 operator.methodcaller 就很方便了。比如你需要排序一系列的点,就可以这样做:

  1. points = [
  2. Point(1, 2),
  3. Point(3, 0),
  4. Point(10, -3),
  5. Point(-5, -7),
  6. Point(-1, 8),
  7. Point(3, 2)
  8. ]
  9. # Sort by distance from origin (0, 0)
  10. points.sort(key=operator.methodcaller('distance', 0, 0))

讨论

调用一个方法实际上是两部独立操作,第一步是查找属性,第二步是函数调用。因此,为了调用某个方法,你可以首先通过 getattr() 来查找到这个属性,然后再去以函数方式调用它即可。

operator.methodcaller() 创建一个可调用对象,并同时提供所有必要参数,然后调用的时候只需要将实例对象传递给它即可,比如:

  1. >>> p = Point(3, 4)
  2. >>> d = operator.methodcaller('distance', 0, 0)
  3. >>> d(p)
  4. 5.0
  5. >>>

通过方法名称字符串来调用方法通常出现在需要模拟 case 语句或实现访问者模式的时候。参考下一小节获取更多高级例子。

原文:

http://python3-cookbook.readthedocs.io/zh_CN/latest/c08/p20_call_method_on_object_by_string_name.html