Timed LRU Cache

timed_lru_cache.py

# From RealPython blog.
from functools import lru_cache, wraps
from datetime import datetime, timedelta

# Note: this will clear whole cache each time expiration is reached.  Should adapt to expire items independently when they expire
def timed_lru_cache(seconds: int, maxsize: int = 128):
    def wrapper_cache(func):
        func = lru_cache(maxsize=maxsize)(func)
        func.lifetime = timedelta(seconds=seconds)
        func.expiration = datetime.utcnow() + func.lifetime
        
        @wraps(func)
        def wrapped(*args, **kwargs):
            if datetime.utcnow() >= func.expiration:
                func.cache_clear()
                func.expiration = datetime.utcnow() + func.lifetime
            wrapped.cache_info = func.cache_info()
            return func(*args, **kwargs)
        
        return wrapped
    return wrapper_cache