Getattr和methodcaller剖析

2021年07月25日约 778 字预计阅读 2 分钟

一、getattr解析

getattr() 函数获取某个类实例对象中指定属性的值

1
getattr(obj, name[, default])

obj 表示指定的类实例对象
name 表示指定的属性名
default 是可选参数，用于设定该函数的默认返回值
- 即当函数查找失败时，如果不指定 default 参数，则程序将直接报 AttributeError 错误
- 反之该函数将返回 default 指定的值。

1
2
3
4
class KeyWord(object):
    driver = "chrome驱动"
    def openBrowser(self, browser):
        print(f"openBrowser is running {browser}")

1
2
3
4
5
kw = Keywork()
driver = getattr(kw, "driver")
print(driver)
# ----------------------
chrome驱动

1
2
3
4
name = getattr(kw, "name")
print(name)
# -------------------
AttributeError: 'Keywork' object has no attribute 'name'

1
2
3
4
name = getattr(kw, "name","Desire")
print(name)
# -------------------
Desire

1
2
openBrowser = getattr(kw, "openBrowser") # 获取类方法
openBrowser(browser="Chrome") # 调用类方法

Python内部提供的一个可以通过字符串调用类方法的一个包，在operator库中。

1
from operator import methodcaller

1
2
3
method = methodcaller("openBrowser", browser="Chrome")
keyword = KeyWord()
method(keyword)

1
2
# --------------------------------
openBrowser is running Chrome

1
2
3
4
def methodcaller(name, /, *args, **kwargs):
    def caller(obj):
        return getattr(obj, name)(*args, **kwargs)
    return caller

采用了闭包的形式
调用methodcaller函数时，返回的是methodcaller内部的函数
- 也就是说调用后返回的还是一个函数
- 调用methodcaller函数的返回值（method），其实就是caller(obj)
method == caller，所以我们可以通过method(obj)执行methodcaller内部方法caller中的函数体了
caller函数操作：通过getattr() 函数从指定的对象返回指定属性的值，从而实现了通过字符串调用类方法

从上面的methodcaller源码和解读中可以看到，methodcaller底层还是使用的getattr，对于字符串调用类方法，这两种方式都是可行的，自行选择使用。