Python中的lazy property
引言
我们都知道,在Python的类中,__
dict
__
保存了一个对象所有的属性,如下面的例子,我们建立了一个Circle的对象,对象的字典中保存着半径radius
这个k-v值:
class Circle(object):
def __init__(self,radius):
self.radius = radius
c = Circle(2.5)
print(c.__dict__)
#{'radius': 2.5}
对于类中的方法,我们有时候希望它可以像属性一样被调用,这时候我们通常给类的方法添加@property修饰符:
class Circle(object):
def __init__(self,radius):
self.radius = radius
@property
def area(self):
print("calculate area")
return 3.14 * self.radius * 2
circle = Circle(4)
print(circle.area)
#calculate area
#25.12
但是这么做,虽然area可以当作一个属性访问,但是并不是真正的变成了一个属性,同时,我们每次调用circle.area,都会从头到尾执行一遍函数
,我们来看下面的例子:
class Circle(object):
def __init__(self,radius):
self.radius = radius
@property
def area(self):
print("calculate area")
return 3.14 * self.radius * 2
c = Circle(4)
print(c.__dict__)
print(c.area)
print(c.area)
print(c.__dict__)
输出为:
{'radius': 4}
calculate area
25.12
calculate area
25.12
{'radius': 4}
可以发现:
每次执行circle.area,整个函数流程都会被执行一次
同时area也没有真正变成对象的属性,dict中并没有area
深入
那么我们有没有办法把一个类中的函数真正变成对象的属性,同时只有在第一次调用时进行一次计算,而之后每次调用不会重复计算呢?这就是Python中的lazy property。本文介绍两种方法。一种是使用python描述符,另一种是使用python修饰符。
1、使用python描述符
class lazy(object):
def __init__(self, func):
self.func = func
def __get__(self, instance, cls):
val = self.func(instance)
setattr(instance, self.func.__name__, val)
return val
class Circle(object):
def __init__(self, radius):
self.radius = radius
@lazy
def area(self):
print("calculate area")
return 3.14 * self.radius ** 2
c = Circle(4)
print('before calculate area')
print(c.__dict__)
print(c.area)
print(c.area)
print('after calculate area')
print(c.__dict__)
c.radius = 5
print(c.__dict__)
print(c.area)
输出为:
before calculate area
{'radius': 4}
calculate area
50.24
50.24
after calculate area
{'radius': 4, 'area': 50.24}
{'radius': 5, 'area': 50.24}
50.24
{'radius': 5, 'area': 50.24}
可以看到,area只在第一次调用时计算了一次,同时在调用以后area变成了对象的一个属性,同时值并不随半径的变化而变化。
在这里,lazy类有__get__方法,说明是个描述器,第一次执行c.area的时候,因为顺序问题,先去c.__dict__中找,没找到,就去类空间找,在类Circle中,有area()方法,于是就被__get__拦截。
在__get__中,调用实例的area()方法算出结果,并动态给实例添加个同名属性把结果赋给它,即加到c.__dict__中去。
再次执行c.area的时候,先去c.__dict__找,因为此时已经有了,就不会经过area()方法和__get__了。
2、使用python修饰符
def lazy_property(func):
attr_name = "_lazy_" + func.__name__
@property
def _lazy_property(instance):
if not hasattr(instance, attr_name):
setattr(instance, attr_name, func(instance))
return getattr(instance, attr_name)
return _lazy_property
class Circle(object):
def __init__(self, radius):
self.radius = radius
@lazy_property
def area(self):
return 3.14 * self.radius ** 2
c = Circle(4)
print('before calculate area')
print(c.__dict__)
print(c.area)
print(c.area)
print('after calculate area')
print(c.__dict__)
c.radius = 5
print(c.__dict__)
print(c.area)
输出为:
before calculate area
{'radius': 4}
50.24
50.24
after calculate area
{'radius': 4, '_lazy_area': 50.24}
{'radius': 5, '_lazy_area': 50.24}
50.24
可以看到,area只在第一次调用时计算了一次,同时在调用以后area变成了对象的一个属性,同时值并不随半径的变化而变化。
很疑问:为什么装饰器中的 def _lazy_property(instance)需要传入instance,其实这个instance就是Circle类的对象实例c。