| 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 | |
| @functools.wraps(method) | |
| 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 |