masque/masque/utils/decorators.py

from collections.abc import Callable
from functools import wraps

from ..error import OneShotError


def oneshot(func: Callable) -> Callable:
    """
    Raises a OneShotError if the decorated function is called more than once
    """
    expired = False

    @wraps(func)
    def wrapper(*args, **kwargs):       # noqa: ANN202
        nonlocal expired
        if expired:
            raise OneShotError(func.__name__)
        expired = True
        return func(*args, **kwargs)

    return wrapper