37. 通过实例方法名字的字符串调用方法

某项目中,代码使用了三个不同库中的图形类:CircleTriangleRectangle,它们都有一个获取图形面积的接口(方法),但方法名字不同。

要求:实现一个统一的获取面积的函数,使用各种方法名进行尝试,调用相应类的接口。

解决方案:

  1. 使用内置函数getattr(),通过名字获取方法对象,然后调用。

  2. 使用标准库operator下的methodcaller()函数调用。


  • 对于内置函数getattr()
getattr(object, name[, default])

返回对象命名属性的值。name必须是字符串。如果该字符串是对象的属性之一,则返回该属性的值。例如getattr(x, 'foobar') 等同于x.foobar。如果指定的属性不存在,且提供了default值,则返回它,否则触发AttributeError。

>>> s = 'abc123'>>> s.find('123')3>>> getattr(s, 'find', None)('123')             #等同于s.find('123')3

  • 对于methodcaller()函数:
operator.methodcaller(name[, args...])

methodcaller()函数会返回一个可调用对象,该对象在其操作数上调用方法名。如果提供了额外的参数或关键字参数,它们也将被提供给方法。

>>> from operator import methodcaller>>> s = 'abc123abc456'>>> s.find('abc', 3)6>>> methodcaller('find', 'abc', 3)(s)               #等价于s.find('abc', 3)6


lib1.py

class Triangle:
    def __init__(self, a, b, c):
        self.a, self.b, self.c = a,b,c    
    def get_area(self):
        a,b,c = self.a, self.b, self.c
        p = (a+b+c) / 2
        return (p * (p-a) * (p-b) * (p-c)) ** 0.5

lib2.py

class Rectangle:
    def __init__(self, a, b):
        self.a, self.b = a,b    def getarea(self):
        return self.a * self.b

lib3.py

import mathclass Circle:
    def __init__(self, r):
        self.r = r    def area(self):
        return round(self.r ** 2 * math.pi, 1)

  • 方案1示例:
from lib1 import Trianglefrom lib2 import Rectanglefrom lib3 import Circledef get_area(shape, method_name = ['get_area', 'getarea', 'area']):
    for name in method_name:
        f = getattr(shape, name, None)
        if f: return f()shape1 = Triangle(3, 4, 5)shape2 = Rectangle(4, 6)shape3 = Circle(2)shape_list = [shape1, shape2, shape3]area_list = list(map(get_area, shape_list))print(area_list)[6.0, 24, 12.6]             #结果

  • 方案2示例:
from lib1 import Trianglefrom lib2 import Rectanglefrom lib3 import Circlefrom operator import methodcallerdef get_area(shape, method_name = ['get_area', 'getarea', 'area']):
    for name in method_name:
        if hasattr(shape, name):
            return methodcaller(name)(shape)shape1 = Triangle(3, 4, 5)shape2 = Rectangle(4, 6)shape3 = Circle(2)shape_list = [shape1, shape2, shape3]area_list = list(map(get_area, shape_list))print(area_list)[6.0, 24, 12.6]             #结果


上一篇:个人*投资备案(2)


下一篇:spring-boot开发:使用内嵌容器进行快速开发及测试