pocketpy/python/functools.py

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)