21 lines
575 B
Python
21 lines
575 B
Python
|
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
|