| 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 |
