mirror of
https://github.com/pocketpy/pocketpy
synced 2025-10-31 00:40:16 +00:00
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
class cache:
|
|
def __init__(self, f):
|
|
self.f = f
|
|
self.cache = {}
|
|
|
|
def __call__(self, *args):
|
|
if args not in self.cache:
|
|
self.cache[args] = self.f(*args)
|
|
return self.cache[args]
|
|
|
|
class lru_cache:
|
|
def __init__(self, maxsize=128):
|
|
self.maxsize = maxsize
|
|
self.cache = {}
|
|
|
|
def __call__(self, f):
|
|
def wrapped(*args):
|
|
if args in self.cache:
|
|
res = self.cache.pop(args)
|
|
self.cache[args] = res
|
|
return res
|
|
|
|
res = f(*args)
|
|
if len(self.cache) >= self.maxsize:
|
|
first_key = next(iter(self.cache))
|
|
self.cache.pop(first_key)
|
|
self.cache[args] = res
|
|
return res
|
|
return wrapped
|
|
|
|
def reduce(function, sequence, initial=...):
|
|
it = iter(sequence)
|
|
if initial is ...:
|
|
try:
|
|
value = next(it)
|
|
except StopIteration:
|
|
raise TypeError("reduce() of empty sequence with no initial value")
|
|
else:
|
|
value = initial
|
|
for element in it:
|
|
value = function(value, element)
|
|
return value
|
|
|
|
class partial:
|
|
def __init__(self, f, *args, **kwargs):
|
|
self.f = f
|
|
if not callable(f):
|
|
raise TypeError("the first argument must be callable")
|
|
self.args = args
|
|
self.kwargs = kwargs
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
kwargs.update(self.kwargs)
|
|
return self.f(*self.args, *args, **kwargs)
|
|
|