/
utils.py
46 lines (42 loc) · 2 KB
/
utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
try:
from functools import cached_property
except ImportError:
from threading import RLock
_NOT_FOUND = object()
class cached_property:
def __init__(self, func):
self.func = func
self.attrname = None
self.__doc__ = func.__doc__
self.lock = RLock()
def __set_name__(self, owner, name):
if self.attrname is None:
self.attrname = name
elif name != self.attrname:
raise TypeError('Cannot assign the same cached_property to two different names ' f'({self.attrname!r} and {name!r}).')
def __get__(self, instance, owner=None):
if instance is None:
return self
if self.attrname is None:
raise TypeError('Cannot use cached_property instance without calling __set_name__ on it.')
try:
cache = instance.__dict__
except AttributeError: # not all objects have __dict__ (e.g. class defines slots)
msg = f"No '__dict__' attribute on {type(instance).__name__!r} " f'instance to cache {self.attrname!r} property.'
raise TypeError(msg) from None
val = cache.get(self.attrname, _NOT_FOUND)
if val is _NOT_FOUND:
with self.lock:
# check if another thread filled cache while we awaited lock
val = cache.get(self.attrname, _NOT_FOUND)
if val is _NOT_FOUND:
val = self.func(instance)
try:
cache[self.attrname] = val
except TypeError:
msg = (
f"The '__dict__' attribute on {type(instance).__name__!r} instance "
f'does not support item assignment for caching {self.attrname!r} property.'
)
raise TypeError(msg) from None
return val