You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
469 B
Python
22 lines
469 B
Python
from typing 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):
|
|
nonlocal expired
|
|
if expired:
|
|
raise OneShotError(func.__name__)
|
|
expired = True
|
|
return func(*args, **kwargs)
|
|
|
|
return wrapper
|