call_limit.py

call_limit.py

class CallLimitException(Exception):
    pass


def call_limit(limit=10):
    def deco(wrapped):
        nonlocal limit
        i = 0

        def wrapper(*args, **kwargs):
            nonlocal i
            i += 1
            if i > limit:
                raise CallLimitException(i)

            return wrapped(*args, **kwargs)

        return wrapper

    return deco


@call_limit(2)
def f(x):
    print(f'f({x})')
    return x


for j in range(1, 20):
    print(f'{j}: {f(j)}')