Python function cache

# Example use - Decorator code below -----------------------------
import time

@cache(ttl=timedelta(minutes=3), max_entries=300)
def add(a, b):
    time.sleep(2)
    return a + b

@cache()
def substract(a, b):
    time.sleep(2)
    return a - b
  
  
# Decorator code -----------------------------
from datetime import datetime, timedelta

def cache(**kwargs):
  def decorator(function):
    # static function variable for cache, lazy initialization
    try: function.cache
    except: function.cache = {}
    def wrapper(*args):
        # if nothing valid in cache, insert something
        if not args in function.cache or datetime.now() > function.cache[args]['expiry']:
            if 'max_entries' in kwargs:
                max_entries = kwargs['max_entries']
                if max_entries != None and len(function.cache) >= max_entries:
                    now = datetime.now()
                    function.cache = { key: function.cache[key] for key in function.cache.keys() if function.cache[key]['expiry'] > now }
                    # if nothing is expired that is deletable, delete the first
                    if len(function.cache) >= max_entries:
                        del function.cache[next(iter(function.cache))]
            function.cache[args] = {'result': function(*args), 'expiry': datetime.max if 'ttl' not in kwargs else datetime.now() + kwargs['ttl']}

        # answer from cache
        return function.cache[args]['result']
    return wrapper
  return decorator

Source: stackoverflow.com

Similar codes

0 vote248 views1 answer
Pt_core_news_sm spacy download
Python
0 vote174 views1 answer
String maniupulation in python
Python
0 vote161 views1 answer
Base conversion python
Python
0 vote203 views1 answer
Code=H18 desc="Server Request Interrupted" django
Python
0 vote176 views1 answer
Show every second xtick
Python
0 vote175 views1 answer
CMake Error at pybind11/tools/FindPythonLibsNew.cmake:131 (message): Python config failure:
Python
3 votes188 views1 answer
WKUP2 in stm32 microcontroller
Python
3 votes180 views2 answers
Scanning 2d array in python
Python
2 votes190 views1 answer
Discord.py check if user has role
Python
-1 vote186 views1 answer
Cannot import name 'httpresponse' from 'django.http'
Python