Skip to main content

A comprehensive Redis toolkit for Python with caching, memoization, and utilities

Project description

rediskit

A Python toolkit that provides Redis-backed performance and concurrency primitives for applications. It enables developers to add caching, distributed coordination, and data protection to their Python applications with minimal effort.

Still work in progress

Many features are still under development, there will be many breaking changes. Please use at your own risk.

Features

  • Function Result Caching: Use the @RedisMemoize decorator to cache expensive function calls with automatic serialization, compression, and encryption
  • Distributed Coordination: Redis-based distributed locks and semaphores for coordinating access across multiple processes/machines
  • Data Protection: Multi-version encryption keys with automatic key rotation for sensitive cached data
  • Async Support: Full support for both synchronous and asynchronous applications
  • Flexible Storage: Choose between string or hash-based Redis storage patterns
  • Modern Type Hints: Full type safety with Python 3.12+ syntax

Installation

uv add rediskit
# or
poetry add rediskit

Quick Start

Basic Setup

from rediskit import redis_memoize, init_redis_connection_pool

# Initialize Redis connection pool (call once at app startup)
init_redis_connection_pool()


# Cache expensive function results
@redis_memoize(memoize_key="expensive_calc", ttl=300)
def expensive_calculation(tenantId: str, value: int) -> dict:
    # Simulate expensive computation
    import time
    time.sleep(2)
    return {"result": value * 42}


# Usage
result = expensive_calculation("tenant1", 10)  # Takes 2 seconds
result = expensive_calculation("tenant1", 10)  # Returns instantly from cache

Custom Redis Connection

import redis
from rediskit import redis_memoize

# Use your own Redis connection
my_redis = redis.Redis(host='my-redis-host', port=6379, db=1)


@redis_memoize(
    memoize_key="custom_calc",
    ttl=600,
    connection=my_redis
)
def my_function(tenantId: str, data: dict) -> dict:
    return {"processed": data}

Advanced Caching Options

from rediskit import redis_memoize


# Hash-based storage with encryption
@redis_memoize(
    memoize_key=lambda tenantId, user_id: f"user_profile:{tenantId}:{user_id}",
    ttl=3600,
    storage_type="hash",  # Store in Redis hash for efficient field access
    enable_encryption=True,  # Encrypt sensitive data
    cache_type="zipJson"  # JSON serialization with compression
)
def get_user_profile(tenantId: str, user_id: str) -> dict:
    # Fetch user data from database
    return {"user_id": user_id, "name": "John Doe", "email": "john@example.com"}


# Dynamic TTL and cache bypass
@redis_memoize(
    memoize_key="dynamic_data",
    ttl=lambda tenantId, priority: 3600 if priority == "high" else 300,
    bypass_cache=lambda tenantId, force_refresh: force_refresh
)
def get_dynamic_data(tenantId: str, priority: str, force_refresh: bool = False) -> dict:
    return {"data": "fresh_data", "priority": priority}

Async Support

import asyncio
from rediskit import a_redis_memoize, init_async_redis_connection_pool


# Use a_redis_memoize for async functions
@a_redis_memoize(memoize_key="async_calc", ttl=300)
async def async_expensive_function(tenantId: str, value: int) -> dict:
    await asyncio.sleep(1)  # Simulate async work
    return {"async_result": value * 100}


async def main() -> None:
    # Initialize async Redis connection pool (once per event loop)
    await init_async_redis_connection_pool()
    result = await async_expensive_function("tenant1", 5)


asyncio.run(main())

High Availability with Redis Sentinel (async)

When your Redis runs behind Sentinel (1 master + replicas, with automatic failover), a plain client will keep writing to a node that gets demoted to a read-only replica during a failover — surfacing as ReadOnlyError: You can't write against a read only replica. Enable Sentinel so the async client discovers the current master on every reconnect and retries transparently through a failover:

export REDIS_SENTINEL_ENABLED="true"
export REDIS_SENTINEL_HOSTS="sentinel-a:26379,sentinel-b:26379,sentinel-c:26379"
export REDIS_SENTINEL_MASTER_NAME="myMaster"   # Sentinel's monitored master name
export REDIS_PASSWORD="<data-node password>"   # authenticates the master/replicas
# export REDIS_SENTINEL_PASSWORD="..."         # only if your sentinels require auth

No code change is required — init_async_redis_connection_pool() and every capability built on it (locks, pub/sub, memoize, semaphores, counters, all ops) automatically route through the Sentinel-managed master:

from rediskit import init_async_redis_connection_pool, get_async_redis_mutex_lock

await init_async_redis_connection_pool()   # builds a Sentinel-managed client when enabled
async with get_async_redis_mutex_lock("critical_section", expire=30):
    ...  # writes always land on the current master, even after a failover

A single host:port that resolves to all sentinels (e.g. a Kubernetes Service DNS name) is also valid for REDIS_SENTINEL_HOSTS. The sync client is unaffected by these settings.

Testing against a local Sentinel

docker-compose.sentinel.yml brings up a real 1-master + 1-replica + 1-sentinel topology and a test runner that executes the suite inside the docker network — required because Sentinel hands clients the redis / redis-replica hostnames it announces, which only resolve there (not from your host shell). Because every async op funnels through one client, this runs the whole async suite through the Sentinel-managed master:

# Run the Sentinel integration suite (every capability + broadcast, through Sentinel):
PYTEST_TARGET=tests/test_sentinel_integration.py \
  docker compose -f docker-compose.sentinel.yml run --rm test

# Or the entire suite through Sentinel (default PYTEST_TARGET=tests/):
docker compose -f docker-compose.sentinel.yml run --rm test

# Transparent-failover test — then, in another shell, kill the master:
#   docker compose -f docker-compose.sentinel.yml stop redis
REDISKIT_SENTINEL_FAILOVER=1 PYTEST_TARGET=tests/test_sentinel_integration.py \
  docker compose -f docker-compose.sentinel.yml run --rm test

docker compose -f docker-compose.sentinel.yml down -v   # tear down

tests/test_sentinel_integration.py skips unless REDIS_SENTINEL_ENABLED=true, so the default pytest tests/ (single Redis, no Sentinel) is unaffected.

Distributed Locking

from rediskit import get_redis_mutex_lock, get_async_redis_mutex_lock

# Synchronous distributed lock
with get_redis_mutex_lock("critical_section", expire=30) as lock:
    # Only one process can execute this block at a time
    perform_critical_operation()

# Async distributed lock
async with get_async_redis_mutex_lock("async_critical_section", expire=30) as lock:
    await perform_async_critical_operation()

Encryption Management

from rediskit import Encrypter

# Generate new encryption keys
encrypter = Encrypter()
new_key = encrypter.generate_new_hex_key()

# Encrypt/decrypt data manually
encrypted = encrypter.encrypt("sensitive data", useZstd=True)
decrypted = encrypter.decrypt(encrypted)

Configuration

Configure rediskit using environment variables (loaded from .env / private.env if present):

# Redis connection settings
export REDIS_HOST="localhost"
export REDIS_PORT="6379"
export REDIS_PASSWORD=""

# Sentinel (async high-availability master discovery; sync client unaffected)
export REDIS_SENTINEL_ENABLED="false"          # set "true" to route the async client via Sentinel
export REDIS_SENTINEL_HOSTS=""                 # "host-a:26379,host-b:26379" (bare host uses REDIS_SENTINEL_PORT)
export REDIS_SENTINEL_PORT="26379"             # default port for bare hosts above
export REDIS_SENTINEL_MASTER_NAME="myMaster"   # Sentinel's monitored master name
export REDIS_SENTINEL_PASSWORD=""              # only if your sentinels themselves require auth
export REDIS_KIT_RETRY_ATTEMPTS="10"           # transparent retries per command (covers the failover window)

# Encryption keys (base64-encoded JSON, e.g. produced by Encrypter.encode_keys_dict_to_base64)
export REDIS_KIT_ENCRYPTION_SECRET="eyJfX2VuY192MSI6ICI0MGViODJlNWJhNTJiNmQ4..."

# Cache settings
export REDIS_TOP_NODE="my_app_cache"      # key prefix, default "redis_kit_node"
export REDIS_SKIP_CACHING="false"         # short-circuit cache reads
export REDIS_SCAN_COUNT="10000"           # SCAN batch size hint

API Reference

Core Decorators

@redis_memoize / @a_redis_memoize

Cache function results in Redis with configurable options. Use redis_memoize for sync functions and a_redis_memoize for async functions.

Parameters:

  • memoize_key: Cache key (string or callable)
  • ttl: Time to live in seconds (int, callable, or None)
  • bypass_cache: Skip cache lookup (bool or callable)
  • cache_type: Serialization method ("zipJson" or "zipPickled")
  • reset_ttl_upon_read: Refresh TTL when reading from cache
  • enable_encryption: Encrypt cached data
  • storage_type: Redis storage pattern ("string" or "hash")
  • connection: Custom Redis connection (optional)
  • lock_sleep (a_redis_memoize only): Seconds between mutex acquisition attempts

Connection Management

  • init_redis_connection_pool(): Initialize sync Redis connection pool
  • init_async_redis_connection_pool(): Initialize async Redis connection pool (per event loop)
  • get_redis_connection(): Get sync Redis connection
  • get_async_redis_connection(): Get async Redis connection
  • close() / async_connection_close(): Tear down the sync / async pool

Distributed Coordination

  • get_redis_mutex_lock(lock_name, expire, auto_renewal, id): Get sync distributed lock
  • get_async_redis_mutex_lock(lock_name, expire, sleep, blocking, blocking_timeout, ...): Get async distributed lock
  • Semaphore / AsyncSemaphore: Distributed semaphore limiting concurrent holders across processes

Encryption

  • Encrypter(keyHexDict): Encryption/decryption with key versioning

Requirements

  • Python 3.12+
  • Redis server (RedisJSON module required for the JSON cache helpers, hash-field TTLs require Redis 7.4+)
  • Dependencies: redis, python-redis-lock, pynacl, zstd, httpx, python-dotenv

License

Apache-2.0 license

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

rediskit-0.0.43.tar.gz (75.2 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

rediskit-0.0.43-py3-none-any.whl (52.7 kB view details)

Uploaded Python 3

File details

Details for the file rediskit-0.0.43.tar.gz.

File metadata

  • Download URL: rediskit-0.0.43.tar.gz
  • Upload date:
  • Size: 75.2 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for rediskit-0.0.43.tar.gz
Algorithm Hash digest
SHA256 49e9fba5da29c7d601735ec39400202c6e703850b575aca1b04b70d140d25e3b
MD5 9308b743e8b36a7d3d3ba4f1bc72f170
BLAKE2b-256 c6bb4538c1db154444454ba4e8949ce1d185ad7f5324dda324584b214c352bc6

See more details on using hashes here.

Provenance

The following attestation bundles were made for rediskit-0.0.43.tar.gz:

Publisher: publish.yml on BadrElfarri/rediskit

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file rediskit-0.0.43-py3-none-any.whl.

File metadata

  • Download URL: rediskit-0.0.43-py3-none-any.whl
  • Upload date:
  • Size: 52.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for rediskit-0.0.43-py3-none-any.whl
Algorithm Hash digest
SHA256 bf852c99f63008643349284614348d4b5813d4bc98baacaf231b07f4e6610663
MD5 2671886d252ba5539b5f7166272a6f31
BLAKE2b-256 713251a71b9e22c6881ec33bf6eb3c7a68633f6a5c4a676e9f8cc2843ee9255e

See more details on using hashes here.

Provenance

The following attestation bundles were made for rediskit-0.0.43-py3-none-any.whl:

Publisher: publish.yml on BadrElfarri/rediskit

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page