A decorator to limit the number of calls to a function

limit_calls.py

"""For my friends in OTA.

--Towel, 2019
"""
from functools import wraps


class LimitCalls:
    """A decorator to limit the number of calls to a function.

    When the number of executions has exceeded the call limit, calling
    the wrapped function will have no effect. If no call limit is specified
    the function is limited to one call.
    """
    __slots__ = 'executions_left'

    DEFAULT_LIMIT = 1

    def __init__(self, call_limit=DEFAULT_LIMIT):
        self.executions_left = call_limit

    def __call__(self, fcn):
        @wraps(fcn)
        def wrapped(*args, **kwargs):
            if self.executions_left > 0:
                self.executions_left -= 1
                return fcn(*args, **kwargs)
        return wrapped

Example Usage:

@LimitCalls(2)
def example():
    print('[+] Example Executed.')


example()
example()
example()  # This will have no effect