Current File : //proc/thread-self/root/opt/cloudlinux/venv/lib/python3.11/site-packages/clwpos/scoped_cache.py
"""
This is a simple wrapper around lru_cache which allows
us to enable caching only within some scope defined by contextmanager
It makes it easier to cache some functions e.g. only inside wpos-user get
scope and still be able to get fresh data in wpos-user enable
where cached commands might return different results because of outside
"""
from contextlib import contextmanager
from functools import lru_cache
_CACHED_FUNCTIONS = {}
CACHING_ENABLED = False
@contextmanager
def enable_caching():
"""
This context manager sets global marker
which enables caching
"""
global CACHING_ENABLED
if CACHING_ENABLED:
raise RuntimeError('nested scopes are not supported')
try:
CACHING_ENABLED = True
yield
finally:
_clear_cache()
CACHING_ENABLED = False
def _clear_cache():
"""
Clear cache of all registered functions
"""
for func in _CACHED_FUNCTIONS.values():
func.cache_clear()
def cached_in_scope(func):
"""
Primary code which decides should we return cached
result or fresh
"""
def _wrapper(*args, **kwargs):
if CACHING_ENABLED:
return _CACHED_FUNCTIONS[func.__name__](*args, **kwargs)
return func(*args, **kwargs)
cached_func = lru_cache()(func)
_CACHED_FUNCTIONS[func.__name__] = cached_func
return _wrapper
Mini Shell