cachu 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl

cachu/__init__.py

@@ -1,19 +1,16 @@
 """Flexible caching library with support for memory, file, and Redis backends.
 """
-__version__ = '0.2.3'
+__version__ = '0.2.4'
 
-from .async_decorator import async_cache, clear_async_backends
-from .async_decorator import get_async_backend, get_async_cache_info
-from .async_operations import async_cache_clear, async_cache_delete
-from .async_operations import async_cache_get, async_cache_info
-from .async_operations import async_cache_set
 from .backends import AsyncBackend, Backend
 from .backends.redis import get_redis_client
 from .config import configure, disable, enable, get_all_configs, get_config
 from .config import is_disabled
-from .decorator import cache, get_backend
-from .operations import cache_clear, cache_delete, cache_get, cache_info
-from .operations import cache_set
+from .decorator import async_cache, cache, clear_async_backends
+from .decorator import get_async_backend, get_async_cache_info, get_backend
+from .operations import async_cache_clear, async_cache_delete, async_cache_get
+from .operations import async_cache_info, async_cache_set, cache_clear
+from .operations import cache_delete, cache_get, cache_info, cache_set
 
 __all__ = [
     'configure',

cachu/backends/__init__.py

@@ -1,7 +1,7 @@
 """Cache backend implementations.
 """
 from abc import ABC, abstractmethod
-from collections.abc import Iterator
+from collections.abc import AsyncIterator, Iterator
 from typing import Any
 
 NO_VALUE = object()
@@ -47,6 +47,49 @@ class Backend(ABC):
         """
 
 
-from .async_base import AsyncBackend
+class AsyncBackend(ABC):
+    """Abstract base class for async cache backends.
+    """
+
+    @abstractmethod
+    async def get(self, key: str) -> Any:
+        """Get value by key. Returns NO_VALUE if not found.
+        """
+
+    @abstractmethod
+    async def get_with_metadata(self, key: str) -> tuple[Any, float | None]:
+        """Get value and creation timestamp. Returns (NO_VALUE, None) if not found.
+        """
+
+    @abstractmethod
+    async def set(self, key: str, value: Any, ttl: int) -> None:
+        """Set value with TTL in seconds.
+        """
+
+    @abstractmethod
+    async def delete(self, key: str) -> None:
+        """Delete value by key.
+        """
+
+    @abstractmethod
+    async def clear(self, pattern: str | None = None) -> int:
+        """Clear entries matching pattern. Returns count of cleared entries.
+        """
+
+    @abstractmethod
+    async def keys(self, pattern: str | None = None) -> AsyncIterator[str]:
+        """Iterate over keys matching pattern.
+        """
+
+    @abstractmethod
+    async def count(self, pattern: str | None = None) -> int:
+        """Count keys matching pattern.
+        """
+
+    @abstractmethod
+    async def close(self) -> None:
+        """Close the backend and release resources.
+        """
+
 
 __all__ = ['Backend', 'AsyncBackend', 'NO_VALUE']

cachu/backends/memory.py

@@ -1,13 +1,14 @@
 """Memory cache backend implementation.
 """
+import asyncio
 import fnmatch
 import pickle
 import threading
 import time
-from collections.abc import Iterator
+from collections.abc import AsyncIterator, Iterator
 from typing import Any
 
-from . import NO_VALUE, Backend
+from . import NO_VALUE, AsyncBackend, Backend
 
 
 class MemoryBackend(Backend):
@@ -100,3 +101,102 @@ class MemoryBackend(Backend):
         """Count keys matching pattern.
         """
         return sum(1 for _ in self.keys(pattern))
+
+
+class AsyncMemoryBackend(AsyncBackend):
+    """Async in-memory cache backend using asyncio.Lock.
+    """
+
+    def __init__(self) -> None:
+        self._cache: dict[str, tuple[bytes, float, float]] = {}
+        self._lock = asyncio.Lock()
+
+    async def get(self, key: str) -> Any:
+        """Get value by key. Returns NO_VALUE if not found or expired.
+        """
+        async with self._lock:
+            entry = self._cache.get(key)
+            if entry is None:
+                return NO_VALUE
+
+            pickled_value, created_at, expires_at = entry
+            if time.time() > expires_at:
+                del self._cache[key]
+                return NO_VALUE
+
+            return pickle.loads(pickled_value)
+
+    async def get_with_metadata(self, key: str) -> tuple[Any, float | None]:
+        """Get value and creation timestamp. Returns (NO_VALUE, None) if not found.
+        """
+        async with self._lock:
+            entry = self._cache.get(key)
+            if entry is None:
+                return NO_VALUE, None
+
+            pickled_value, created_at, expires_at = entry
+            if time.time() > expires_at:
+                del self._cache[key]
+                return NO_VALUE, None
+
+            return pickle.loads(pickled_value), created_at
+
+    async def set(self, key: str, value: Any, ttl: int) -> None:
+        """Set value with TTL in seconds.
+        """
+        now = time.time()
+        pickled_value = pickle.dumps(value)
+        async with self._lock:
+            self._cache[key] = (pickled_value, now, now + ttl)
+
+    async def delete(self, key: str) -> None:
+        """Delete value by key.
+        """
+        async with self._lock:
+            self._cache.pop(key, None)
+
+    async def clear(self, pattern: str | None = None) -> int:
+        """Clear entries matching pattern. Returns count of cleared entries.
+        """
+        async with self._lock:
+            if pattern is None:
+                count = len(self._cache)
+                self._cache.clear()
+                return count
+
+            keys_to_delete = [k for k in self._cache if fnmatch.fnmatch(k, pattern)]
+            for key in keys_to_delete:
+                del self._cache[key]
+            return len(keys_to_delete)
+
+    async def keys(self, pattern: str | None = None) -> AsyncIterator[str]:
+        """Iterate over keys matching pattern.
+        """
+        now = time.time()
+        async with self._lock:
+            all_keys = list(self._cache.keys())
+
+        for key in all_keys:
+            async with self._lock:
+                entry = self._cache.get(key)
+                if entry is None:
+                    continue
+                _, _, expires_at = entry
+                if now > expires_at:
+                    del self._cache[key]
+                    continue
+
+            if pattern is None or fnmatch.fnmatch(key, pattern):
+                yield key
+
+    async def count(self, pattern: str | None = None) -> int:
+        """Count keys matching pattern.
+        """
+        count = 0
+        async for _ in self.keys(pattern):
+            count += 1
+        return count
+
+    async def close(self) -> None:
+        """Close the backend (no-op for memory backend).
+        """

cachu/backends/redis.py

@@ -3,13 +3,14 @@
 import pickle
 import struct
 import time
-from collections.abc import Iterator
+from collections.abc import AsyncIterator, Iterator
 from typing import TYPE_CHECKING, Any
 
-from . import NO_VALUE, Backend
+from . import NO_VALUE, AsyncBackend, Backend
 
 if TYPE_CHECKING:
     import redis
+    import redis.asyncio as aioredis
 
 
 _METADATA_FORMAT = 'd'
@@ -28,6 +29,29 @@ def _get_redis_module() -> Any:
         ) from e
 
 
+def _get_async_redis_module() -> Any:
+    """Import redis.asyncio module, raising helpful error if not installed.
+    """
+    try:
+        import redis.asyncio as aioredis
+        return aioredis
+    except ImportError as e:
+        raise RuntimeError(
+            "Async Redis support requires the 'redis' package (>=4.2.0). "
+            "Install with: pip install cachu[redis]"
+        ) from e
+
+
+async def get_async_redis_client(url: str) -> 'aioredis.Redis':
+    """Create an async Redis client from URL.
+
+    Args:
+        url: Redis URL (e.g., 'redis://localhost:6379/0')
+    """
+    aioredis = _get_async_redis_module()
+    return aioredis.from_url(url)
+
+
 def get_redis_client(url: str) -> 'redis.Redis':
     """Create a Redis client from URL.
 
@@ -38,13 +62,28 @@ def get_redis_client(url: str) -> 'redis.Redis':
     return redis_module.from_url(url)
 
 
+def _pack_value(value: Any, created_at: float) -> bytes:
+    """Pack value with creation timestamp.
+    """
+    metadata = struct.pack(_METADATA_FORMAT, created_at)
+    pickled = pickle.dumps(value)
+    return metadata + pickled
+
+
+def _unpack_value(data: bytes) -> tuple[Any, float]:
+    """Unpack value and creation timestamp.
+    """
+    created_at = struct.unpack(_METADATA_FORMAT, data[:_METADATA_SIZE])[0]
+    value = pickle.loads(data[_METADATA_SIZE:])
+    return value, created_at
+
+
 class RedisBackend(Backend):
     """Redis cache backend.
     """
 
-    def __init__(self, url: str, distributed_lock: bool = False) -> None:
+    def __init__(self, url: str) -> None:
         self._url = url
-        self._distributed_lock = distributed_lock
         self._client: redis.Redis | None = None
 
     @property
@@ -55,27 +94,13 @@ class RedisBackend(Backend):
             self._client = get_redis_client(self._url)
         return self._client
 
-    def _pack_value(self, value: Any, created_at: float) -> bytes:
-        """Pack value with creation timestamp.
-        """
-        metadata = struct.pack(_METADATA_FORMAT, created_at)
-        pickled = pickle.dumps(value)
-        return metadata + pickled
-
-    def _unpack_value(self, data: bytes) -> tuple[Any, float]:
-        """Unpack value and creation timestamp.
-        """
-        created_at = struct.unpack(_METADATA_FORMAT, data[:_METADATA_SIZE])[0]
-        value = pickle.loads(data[_METADATA_SIZE:])
-        return value, created_at
-
     def get(self, key: str) -> Any:
         """Get value by key. Returns NO_VALUE if not found.
         """
         data = self.client.get(key)
         if data is None:
             return NO_VALUE
-        value, _ = self._unpack_value(data)
+        value, _ = _unpack_value(data)
         return value
 
     def get_with_metadata(self, key: str) -> tuple[Any, float | None]:
@@ -84,14 +109,14 @@ class RedisBackend(Backend):
         data = self.client.get(key)
         if data is None:
             return NO_VALUE, None
-        value, created_at = self._unpack_value(data)
+        value, created_at = _unpack_value(data)
         return value, created_at
 
     def set(self, key: str, value: Any, ttl: int) -> None:
         """Set value with TTL in seconds.
         """
         now = time.time()
-        packed = self._pack_value(value, now)
+        packed = _pack_value(value, now)
         self.client.setex(key, ttl, packed)
 
     def delete(self, key: str) -> None:
@@ -129,3 +154,89 @@ class RedisBackend(Backend):
         if self._client is not None:
             self._client.close()
             self._client = None
+
+
+class AsyncRedisBackend(AsyncBackend):
+    """Async Redis cache backend using redis.asyncio.
+    """
+
+    def __init__(self, url: str) -> None:
+        self._url = url
+        self._client: aioredis.Redis | None = None
+
+    async def _get_client(self) -> 'aioredis.Redis':
+        """Lazy-load async Redis client.
+        """
+        if self._client is None:
+            self._client = await get_async_redis_client(self._url)
+        return self._client
+
+    async def get(self, key: str) -> Any:
+        """Get value by key. Returns NO_VALUE if not found.
+        """
+        client = await self._get_client()
+        data = await client.get(key)
+        if data is None:
+            return NO_VALUE
+        value, _ = _unpack_value(data)
+        return value
+
+    async def get_with_metadata(self, key: str) -> tuple[Any, float | None]:
+        """Get value and creation timestamp. Returns (NO_VALUE, None) if not found.
+        """
+        client = await self._get_client()
+        data = await client.get(key)
+        if data is None:
+            return NO_VALUE, None
+        value, created_at = _unpack_value(data)
+        return value, created_at
+
+    async def set(self, key: str, value: Any, ttl: int) -> None:
+        """Set value with TTL in seconds.
+        """
+        client = await self._get_client()
+        now = time.time()
+        packed = _pack_value(value, now)
+        await client.setex(key, ttl, packed)
+
+    async def delete(self, key: str) -> None:
+        """Delete value by key.
+        """
+        client = await self._get_client()
+        await client.delete(key)
+
+    async def clear(self, pattern: str | None = None) -> int:
+        """Clear entries matching pattern. Returns count of cleared entries.
+        """
+        client = await self._get_client()
+        if pattern is None:
+            pattern = '*'
+
+        count = 0
+        async for key in client.scan_iter(match=pattern):
+            await client.delete(key)
+            count += 1
+        return count
+
+    async def keys(self, pattern: str | None = None) -> AsyncIterator[str]:
+        """Iterate over keys matching pattern.
+        """
+        client = await self._get_client()
+        redis_pattern = pattern or '*'
+        async for key in client.scan_iter(match=redis_pattern):
+            yield key.decode() if isinstance(key, bytes) else key
+
+    async def count(self, pattern: str | None = None) -> int:
+        """Count keys matching pattern.
+        """
+        count = 0
+        async for _ in self.keys(pattern):
+            count += 1
+        return count
+
+    async def close(self) -> None:
+        """Close the Redis connection.
+        """
+        if self._client is not None:
+            await self._client.close()
+            self._client = None