cachebox 5.0.4__cp313-cp313-musllinux_1_2_i686.whl → 5.1.0__cp313-cp313-musllinux_1_2_i686.whl

cachebox/utils.py

@@ -267,13 +267,14 @@ EVENT_HIT = 2
 
 def _cached_wrapper(
     func,
-    cache: BaseCacheImpl,
+    cache: typing.Union[BaseCacheImpl, typing.Callable],
     key_maker: typing.Callable[[tuple, dict], typing.Hashable],
     clear_reuse: bool,
     callback: typing.Optional[typing.Callable[[int, typing.Any, typing.Any], typing.Any]],
     copy_level: int,
     is_method: bool,
 ):
+    is_method = cache_is_function = inspect.isfunction(cache)
     _key_maker = (lambda args, kwds: key_maker(args[1:], kwds)) if is_method else key_maker
 
     hits = 0
@@ -287,11 +288,12 @@ def _cached_wrapper(
         if kwds.pop("cachebox__ignore", False):
             return func(*args, **kwds)
 
+        _cache = cache(args[0]) if cache_is_function else cache
         key = _key_maker(args, kwds)
 
         # try to get result from cache
         try:
-            result = cache[key]
+            result = _cache[key]
         except KeyError:
             pass
         else:
@@ -310,7 +312,7 @@ def _cached_wrapper(
                 raise cached_error
 
             try:
-                result = cache[key]
+                result = _cache[key]
                 hits += 1
                 event = EVENT_HIT
             except KeyError:
@@ -323,7 +325,7 @@ def _cached_wrapper(
                     raise e
 
                 else:
-                    cache[key] = result
+                    _cache[key] = result
                     misses += 1
                     event = EVENT_MISS
 
@@ -332,34 +334,39 @@ def _cached_wrapper(
 
         return _copy_if_need(result, level=copy_level)
 
-    _wrapped.cache = cache
+    if not cache_is_function:
+        _wrapped.cache = cache
+        _wrapped.cache_info = lambda: CacheInfo(
+            hits, misses, cache.maxsize, len(cache), cache.capacity()
+        )
+
     _wrapped.callback = callback
-    _wrapped.cache_info = lambda: CacheInfo(
-        hits, misses, cache.maxsize, len(cache), cache.capacity()
-    )
 
-    def cache_clear() -> None:
-        nonlocal misses, hits, locks, exceptions
-        cache.clear(reuse=clear_reuse)
-        misses = 0
-        hits = 0
-        locks.clear()
-        exceptions.clear()
+    if not cache_is_function:
+
+        def cache_clear() -> None:
+            nonlocal misses, hits, locks, exceptions
+            cache.clear(reuse=clear_reuse)
+            misses = 0
+            hits = 0
+            locks.clear()
+            exceptions.clear()
 
-    _wrapped.cache_clear = cache_clear
+        _wrapped.cache_clear = cache_clear
 
     return _wrapped
 
 
 def _async_cached_wrapper(
     func,
-    cache: BaseCacheImpl,
+    cache: typing.Union[BaseCacheImpl, typing.Callable],
     key_maker: typing.Callable[[tuple, dict], typing.Hashable],
     clear_reuse: bool,
     callback: typing.Optional[typing.Callable[[int, typing.Any, typing.Any], typing.Any]],
     copy_level: int,
     is_method: bool,
 ):
+    is_method = cache_is_function = inspect.isfunction(cache)
     _key_maker = (lambda args, kwds: key_maker(args[1:], kwds)) if is_method else key_maker
 
     hits = 0
@@ -375,11 +382,12 @@ def _async_cached_wrapper(
         if kwds.pop("cachebox__ignore", False):
             return await func(*args, **kwds)
 
+        _cache = cache(args[0]) if cache_is_function else cache
         key = _key_maker(args, kwds)
 
         # try to get result from cache
         try:
-            result = cache[key]
+            result = _cache[key]
         except KeyError:
             pass
         else:
@@ -400,7 +408,7 @@ def _async_cached_wrapper(
                 raise cached_error
 
             try:
-                result = cache[key]
+                result = _cache[key]
                 hits += 1
                 event = EVENT_HIT
             except KeyError:
@@ -413,7 +421,7 @@ def _async_cached_wrapper(
                     raise e
 
                 else:
-                    cache[key] = result
+                    _cache[key] = result
                     misses += 1
                     event = EVENT_MISS
 
@@ -424,21 +432,25 @@ def _async_cached_wrapper(
 
         return _copy_if_need(result, level=copy_level)
 
-    _wrapped.cache = cache
+    if not cache_is_function:
+        _wrapped.cache = cache
+        _wrapped.cache_info = lambda: CacheInfo(
+            hits, misses, cache.maxsize, len(cache), cache.capacity()
+        )
+
     _wrapped.callback = callback
-    _wrapped.cache_info = lambda: CacheInfo(
-        hits, misses, cache.maxsize, len(cache), cache.capacity()
-    )
 
-    def cache_clear() -> None:
-        nonlocal misses, hits, locks, exceptions
-        cache.clear(reuse=clear_reuse)
-        misses = 0
-        hits = 0
-        locks.clear()
-        exceptions.clear()
+    if not cache_is_function:
 
-    _wrapped.cache_clear = cache_clear
+        def cache_clear() -> None:
+            nonlocal misses, hits, locks, exceptions
+            cache.clear(reuse=clear_reuse)
+            misses = 0
+            hits = 0
+            locks.clear()
+            exceptions.clear()
+
+        _wrapped.cache_clear = cache_clear
 
     return _wrapped
 
@@ -456,7 +468,8 @@ def cached(
     Wraps a function to automatically cache and retrieve its results based on input parameters.
 
     Args:
-        cache (BaseCacheImpl, dict, optional): Cache implementation to store results. Defaults to FIFOCache.
+        cache (BaseCacheImpl, dict, callable): Cache implementation to store results. Defaults to FIFOCache.
+                                               Can be a function that got `self` and should return cache.
         key_maker (Callable, optional): Function to generate cache keys from function arguments. Defaults to make_key.
         clear_reuse (bool, optional): Whether to reuse cache during clearing. Defaults to False.
         callback (Callable, optional): Function called on cache hit/miss events. Defaults to None.
@@ -465,7 +478,7 @@ def cached(
     Returns:
         Callable: Decorated function with caching capabilities.
 
-    Example::
+    Example for functions::
 
         @cachebox.cached(cachebox.LRUCache(128))
         def sum_as_string(a, b):
@@ -476,6 +489,20 @@ def cached(
         assert len(sum_as_string.cache) == 1
         sum_as_string.cache_clear()
         assert len(sum_as_string.cache) == 0
+
+    Example for methods::
+
+        class A:
+            def __init__(self, num):
+                self.num = num
+                self._cache = cachebox.FIFOCache(0)
+
+            @cachebox.cached(lambda self: self._cache)
+            def method(self, n):
+                return self.num * n
+
+        instance = A(10)
+        assert A.method(2) == 20
     """
     if cache is None:
         cache = FIFOCache(0)
@@ -483,8 +510,8 @@ def cached(
     if type(cache) is dict:
         cache = FIFOCache(0, cache)
 
-    if not isinstance(cache, BaseCacheImpl):
-        raise TypeError("we expected cachebox caches, got %r" % (cache,))
+    if not isinstance(cache, BaseCacheImpl) and not inspect.isfunction(cache):
+        raise TypeError("we expected cachebox caches or function, got %r" % (cache,))
 
     def decorator(func: FT) -> FT:
         if inspect.iscoroutinefunction(func):
@@ -509,6 +536,9 @@ def cachedmethod(
     copy_level: int = 1,
 ) -> typing.Callable[[FT], FT]:
     """
+    **This function is deperecated due to issue [#35](https://github.com/awolverp/cachebox/issues/35)**.
+    Use `cached` method instead.
+
     Decorator to create a method-specific memoized cache for function results.
 
     Similar to `cached()`, but ignores `self` parameter when generating cache keys.
@@ -523,6 +553,14 @@ def cachedmethod(
     Returns:
         Callable: Decorated method with method-specific caching capabilities.
     """
+    import warnings
+
+    warnings.warn(
+        "cachedmethod is deprecated, use cached instead. see issue https://github.com/awolverp/cachebox/issues/35",
+        DeprecationWarning,
+        stacklevel=2,
+    )
+
     if cache is None:
         cache = FIFOCache(0)
 

{cachebox-5.0.4.dist-info → cachebox-5.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cachebox
-Version: 5.0.4
+Version: 5.1.0
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Classifier: Programming Language :: Python :: 3

{cachebox-5.0.4.dist-info → cachebox-5.1.0.dist-info}/RECORD

@@ -1,11 +1,11 @@
-cachebox-5.0.4.dist-info/METADATA,sha256=vx5tWzBq0cdk0_0alFRm_R_1KquIHOixh3vzMgi00LY,27190
-cachebox-5.0.4.dist-info/WHEEL,sha256=hnEjxG8MMZmmlhJat41pJlHVa7PMz9kEhUjTj-jdJr4,105
-cachebox-5.0.4.dist-info/licenses/LICENSE,sha256=4oH8nU-rNZ9h5tbh6MoPVPFgtYDqsPXlAr2QnhnxQ4Q,1064
+cachebox-5.1.0.dist-info/METADATA,sha256=APyzN3otl7JZGklA6mcBETronY87eKlRyHHRbhWyFsE,27190
+cachebox-5.1.0.dist-info/WHEEL,sha256=hnEjxG8MMZmmlhJat41pJlHVa7PMz9kEhUjTj-jdJr4,105
+cachebox-5.1.0.dist-info/licenses/LICENSE,sha256=4oH8nU-rNZ9h5tbh6MoPVPFgtYDqsPXlAr2QnhnxQ4Q,1064
 cachebox.libs/libgcc_s-27e5a392.so.1,sha256=x5sO63liVwXxrjGGP371wB0RyQe1KEnIynYm82T0G0M,449745
 cachebox/__init__.py,sha256=I93y-hLQKXsEIlmd0KAZ6CVfDqKqcWFvyiJdDsMUcvI,647
 cachebox/_cachebox.py,sha256=hCZLh1DRMHvD5Crqcrsffy0pFRmc9fToTxHmUJDFK5Q,70522
-cachebox/_core.cpython-313-i386-linux-musl.so,sha256=YohxIVKfWlypxRAl98oXfPEVYekxdt8j2cg0x1z949U,801617
+cachebox/_core.cpython-313-i386-linux-musl.so,sha256=jWLNWMHU43OeB70BbFN31tDASU_Hytn6zn8AuhaZe2s,793425
 cachebox/_core.pyi,sha256=qVl1rXUfzAzxy5FrH35Bv7RTp4gvXbaoZ-oMTDe4zTY,2842
 cachebox/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cachebox/utils.py,sha256=tHl21N5dW1HDpscwfkvre3-LeBJo8GiZ9ixYBn9KkGk,16804
-cachebox-5.0.4.dist-info/RECORD,,
+cachebox/utils.py,sha256=Lshu3ZMoEXSB535Ucxmo09EV5eFy24uJMf2cRWhcl-U,18211
+cachebox-5.1.0.dist-info/RECORD,,