contextcache

Cache a python function only in certain contexts.

Usage

Here's an example:

cat example.py     

import contextcache

# Define a private CacheContextVar to store the cached values.
# Don't touch this CacheContextVar from anywhere else!
# You need to define a separate CacheContextVar for every function for which
# you want to enable caching. Use `None` as the default.
_double_cache = contextcache.CacheContextVar("double_cache", default=None)


# Use the `enable_caching` decorator to enable context caching for `double`.
@contextcache.enable_caching(_double_cache)
def double(n: int) -> int:
    print(f"Doubling {n}, working...")
    return n * 2


# Without caching.
print("Without caching")
print(double(1))
print(double(1))

# With caching.
with contextcache.use_caching(double):
    print("\nWith caching")
    print(double(1))
    print(double(1))

# Without caching, again.
print("\nWithout caching, again")
print(double(1))
print(double(1))

Here's the output:

python example.py

Without caching
Doubling 1, working...
2
Doubling 1, working...
2

With caching
Doubling 1, working...
2
2

Without caching, again
Doubling 1, working...
2
Doubling 1, working...
2

See the tests for further examples.

Caveats

• Function arguments must be hashable.