| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970 |
- from cython import Py_ssize_t
- from cpython.dict cimport (
- PyDict_Contains,
- PyDict_GetItem,
- PyDict_SetItem,
- )
- cdef class CachedProperty:
- cdef readonly:
- object func, name, __doc__
- def __init__(self, func):
- self.func = func
- self.name = func.__name__
- self.__doc__ = getattr(func, '__doc__', None)
- def __get__(self, obj, typ):
- if obj is None:
- # accessed on the class, not the instance
- return self
- # Get the cache or set a default one if needed
- cache = getattr(obj, '_cache', None)
- if cache is None:
- try:
- cache = obj._cache = {}
- except (AttributeError):
- return self
- if PyDict_Contains(cache, self.name):
- # not necessary to Py_INCREF
- val = <object>PyDict_GetItem(cache, self.name)
- else:
- val = self.func(obj)
- PyDict_SetItem(cache, self.name, val)
- return val
- def __set__(self, obj, value):
- raise AttributeError("Can't set attribute")
- cache_readonly = CachedProperty
- cdef class AxisProperty:
- cdef readonly:
- Py_ssize_t axis
- object __doc__
- def __init__(self, axis=0, doc=""):
- self.axis = axis
- self.__doc__ = doc
- def __get__(self, obj, type):
- cdef:
- list axes
- if obj is None:
- # Only instances have _mgr, not classes
- return self
- else:
- axes = obj._mgr.axes
- return axes[self.axis]
- def __set__(self, obj, value):
- obj._set_axis(self.axis, value)
|
