import collections | |
import functools | |
# from jaraco.functools 4.0.2 | |
def save_method_args(method): | |
""" | |
Wrap a method such that when it is called, the args and kwargs are | |
saved on the method. | |
""" | |
args_and_kwargs = collections.namedtuple('args_and_kwargs', 'args kwargs') # noqa: PYI024 | |
def wrapper(self, /, *args, **kwargs): | |
attr_name = '_saved_' + method.__name__ | |
attr = args_and_kwargs(args, kwargs) | |
setattr(self, attr_name, attr) | |
return method(self, *args, **kwargs) | |
return wrapper | |