function cache

cache.py

from functools import wraps
def cached(fn):
  name=fn.__name__
  mycache={}
  if not name in mycache:
    mycache[name]={}
  @wraps(fn)
  def cachefn(*args,**kwargs):
    key=(args,frozenset(kwargs))
    if key in mycache[name]:
      return mycache[name][key]
    else:
      mycache[name][key]=fn(*args,**kwargs)
      return mycache[name][key]
  return cachefn
def counted(fn):
  @wraps(fn)
  def myfn(*args,**kwargs):
    myfn.count+=1
    return fn(*args,**kwargs)
  myfn.count=0
  return myfn

@counted
def fib(n):
  if n==0 or n==1:
    return 1
  return fib(n-1)+fib(n-2)

@counted
@cached
def cached_fib(n):
  if n==0 or n==1:
    return 1
  return cached_fib(n-1)+cached_fib(n-2)

print('fib(10)=',fib(10))
print('fib.count=',fib.count)
print('cached_fib(10)=',cached_fib(10))
print('cached_fib.count=',cached_fib.count)