ez-a-sync 0.32.29__cp310-cp310-win_amd64.whl

a_sync/async_property/cached.pxd

@@ -0,0 +1,10 @@
+ctypedef str FieldName
+ctypedef object CacheValue
+
+cdef public str ASYNC_PROPERTY_ATTR = "__async_property__"
+
+cdef class AsyncCachedPropertyInstanceState:
+    cdef readonly dict[FieldName, CacheValue] cache
+    cdef readonly object locks
+    cdef object get_lock(self, str field_name)
+    cdef object get_cache_value(self, str field_name)

a_sync/async_property/cached.pyi

@@ -0,0 +1,45 @@
+from asyncio import Lock
+from typing import Any, Callable, Dict, Generic, Optional, TypeVar, Union
+
+from a_sync.async_property.proxy import AwaitableOnly, AwaitableProxy
+
+ProxyType = Union[AwaitableOnly[__T], AwaitableProxy[__T]]
+
+FieldName = str
+
+__I = TypeVar("__I")
+__T = TypeVar("__T")
+
+def async_cached_property(
+    func: Callable[[__I], __T], *args, **kwargs
+) -> "AsyncCachedPropertyDescriptor[__I, __T]": ...
+
+class AsyncCachedPropertyInstanceState(Generic[__T]):
+    cache: Dict[FieldName, __T]
+    locks: Dict[FieldName, Lock]
+
+class AsyncCachedPropertyDescriptor(Generic[__I, __T]):
+    field_name: FieldName
+    _load_value: Callable[[__I], __T]
+    _fget: Callable[[__I], __T]
+    _fset: Optional[Callable]
+    _fdel: Optional[Callable]
+    def __init__(
+        self, _fget: Callable[[__I], __T], _fset=None, _fdel=None, field_name=None
+    ) -> None: ...
+    def __set_name__(self, owner, name) -> None: ...
+    def __get__(self, instance: __I, owner) -> ProxyType[__T]: ...
+    def __set__(self, instance: __I, value): ...
+    def __delete__(self, instance: __I) -> None: ...
+    def setter(self, method): ...
+    def deleter(self, method): ...
+    def _check_method_name(self, method, method_type) -> None: ...
+    def _check_method_sync(self, method, method_type) -> None: ...
+    def get_instance_state(self, instance) -> AsyncCachedPropertyInstanceState[__T]: ...
+    def get_lock(self, instance: __I) -> Lock: ...
+    def get_cache(self, instance: __I) -> Dict[field_name, Any]: ...
+    def has_cache_value(self, instance: __I) -> bool: ...
+    def get_cache_value(self, instance: __I) -> __T: ...
+    def set_cache_value(self, instance: __I, value: __T) -> None: ...
+    def del_cache_value(self, instance: __I) -> None: ...
+    def get_loader(self, instance: __I) -> Callable[[], __T]: ...

a_sync/async_property/cached.pyx

@@ -0,0 +1,178 @@
+import asyncio
+import collections
+import functools
+import typing
+
+from a_sync._smart cimport shield
+from a_sync.async_property.proxy import AwaitableProxy
+from a_sync.async_property.proxy cimport AwaitableOnly
+from a_sync.functools cimport update_wrapper
+
+# cdef asyncio
+cdef object iscoroutinefunction = asyncio.iscoroutinefunction
+cdef object Lock = asyncio.Lock
+del asyncio
+
+# cdef collections
+cdef object defaultdict = collections.defaultdict
+del collections
+
+# cdef functools
+cdef object wraps = functools.wraps
+del functools
+
+# cdef typing
+cdef object Any = typing.Any
+cdef object DefaultDict = typing.DefaultDict
+cdef object Dict = typing.Dict
+del typing
+
+
+cdef object _AwaitableProxy = AwaitableProxy
+
+
+def async_cached_property(func, *args, **kwargs) -> "AsyncCachedPropertyDescriptor":
+    assert iscoroutinefunction(func), "Can only use with async def"
+    return __AsyncCachedPropertyDescriptor(func, *args, **kwargs)
+
+
+cdef class AsyncCachedPropertyInstanceState:
+    def __cinit__(self) -> None:
+        self.cache: Dict[FieldName, Any] = {}
+        self.locks: DefaultDict[FieldName, Lock] = defaultdict(Lock)
+    
+    cdef object get_lock(self, str field_name):
+        return self.locks[field_name]
+    
+    cdef object get_cache_value(self, str field_name):
+        return self.cache[field_name]
+
+
+class AsyncCachedPropertyDescriptor:
+    _load_value = None
+
+    def __init__(self, _fget, _fset=None, _fdel=None, field_name=None) -> None:
+        self._fget = _fget
+        self._fset = _fset
+        self._fdel = _fdel
+        self.field_name = field_name or _fget.__name__
+
+        update_wrapper(self, _fget)
+        self._check_method_sync(_fset, "setter")
+        self._check_method_sync(_fdel, "deleter")
+
+    def __set_name__(self, owner, name):
+        self.field_name = name
+
+    def __get__(self, instance, owner):
+        if instance is None:
+            return self
+        cache = (<AsyncCachedPropertyInstanceState>self.get_instance_state(instance)).cache
+        if self.field_name in cache:
+            return _AwaitableProxy(cache[self.field_name])
+        return AwaitableOnly(self.get_loader(instance))
+
+    def __set__(self, instance, value):
+        if self._fset is not None:
+            self._fset(instance, value)
+        self.set_cache_value(instance, value)
+
+    def __delete__(self, instance):
+        if self._fdel is not None:
+            self._fdel(instance)
+        self.del_cache_value(instance)
+
+    def setter(self, method):
+        self._check_method_name(method, "setter")
+        return type(self)(self._fget, method, self._fdel, self.field_name)
+
+    def deleter(self, method):
+        self._check_method_name(method, "deleter")
+        return type(self)(self._fget, self._fset, method, self.field_name)
+
+    def _check_method_name(self, method, method_type):
+        if method.__name__ != self.field_name:
+            raise AssertionError(f"@{self.field_name}.{method_type} name must match property name")
+
+    def _check_method_sync(self, method, method_type):
+        if method and iscoroutinefunction(method):
+            raise AssertionError(f"@{self.field_name}.{method_type} must be synchronous")
+
+    def get_instance_state(self, instance):
+        try:
+            return instance.__async_property__
+        except AttributeError:
+            state = AsyncCachedPropertyInstanceState()
+            instance.__async_property__ = state
+            return state
+
+    def get_lock(self, instance):
+        return (<AsyncCachedPropertyInstanceState>self.get_instance_state(instance)).get_lock(self.field_name)
+
+    def get_cache(self, instance):
+        return (<AsyncCachedPropertyInstanceState>self.get_instance_state(instance)).cache
+
+    def has_cache_value(self, instance):
+        return self.field_name in (<AsyncCachedPropertyInstanceState>self.get_instance_state(instance)).cache
+
+    def get_cache_value(self, instance):
+        return (<AsyncCachedPropertyInstanceState>self.get_instance_state(instance)).get_cache_value(self.field_name)
+
+    def set_cache_value(self, instance, value):
+        (<AsyncCachedPropertyInstanceState>self.get_instance_state(instance)).cache[self.field_name] = value
+
+    def del_cache_value(self, instance):
+        del (<AsyncCachedPropertyInstanceState>self.get_instance_state(instance)).cache[self.field_name]
+
+    def get_loader(self, instance):
+        cdef str field_name
+
+        loader = self._load_value
+        if loader is None:
+
+            field_name = self.field_name
+            _fget = self._fget
+            get_instance_state = self.get_instance_state
+
+            if self._fset is None:
+                @wraps(_fget)
+                async def loader(instance):
+                    cdef AsyncCachedPropertyInstanceState instance_state
+                    cdef dict[str, object] cache
+
+                    instance_state = get_instance_state(instance)
+                    locks: defaultdict = instance_state.locks
+                    async with locks[field_name]:
+                        cache = instance_state.cache
+                        if field_name in cache:
+                            return cache[field_name]
+                        value = await _fget(instance)
+                        cache[field_name] = value
+                        dict.pop(locks, field_name)
+                        return value
+            else:
+                _fset = self._fset
+
+                @wraps(_fget)
+                async def loader(instance):
+                    cdef AsyncCachedPropertyInstanceState instance_state
+                    cdef dict[str, object] cache
+
+                    instance_state = get_instance_state(instance)
+                    cache = instance_state.cache
+                    locks: defaultdict = instance_state.locks
+                    async with locks[field_name]:
+                        if field_name in cache:
+                            return cache[field_name]
+                        value = await _fget(instance)
+                        _fset(instance, value)
+                        cache[field_name] = value
+                        dict.pop(locks, field_name)
+                        return value
+
+            self._load_value = loader
+
+        return lambda: shield(loader(instance))
+
+
+cdef object __AsyncCachedPropertyDescriptor = AsyncCachedPropertyDescriptor