.

. BSA HACKER

Server : Apache
System : Linux nusantara.hosteko.com 4.18.0-553.16.1.lve.el8.x86_64 #1 SMP Tue Aug 13 17:45:03 UTC 2024 x86_64
User : koperas1 ( 1254)
PHP Version : 7.4.33
Disable Function : NONE
Directory : /opt/cloudlinux/venv/lib64/python3.11/site-packages/clwpos/

Upload File :

Current File : //opt/cloudlinux/venv/lib64/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