Merge pull request #109 from jaraco/feature/functools

Re-use functionality from jaraco.functools

newsfragments/+275f3051.feature.rst

@@ -0,0 +1 @@
+Rely on save_method_args to save method args.

zipp/__init__.py

@@ -20,6 +20,9 @@
 from .compat.py310 import text_encoding
 from .glob import Translator
 
+from ._functools import save_method_args
+
+
 __all__ = ['Path']
 
 
@@ -86,13 +89,12 @@ class InitializedState:
     Mix-in to save the initialization state for pickling.
     """
 
+    @save_method_args
     def __init__(self, *args, **kwargs):
-        self.__args = args
-        self.__kwargs = kwargs
         super().__init__(*args, **kwargs)
 
     def __getstate__(self):
-        return self.__args, self.__kwargs
+        return self._saved___init__.args, self._saved___init__.kwargs
 
     def __setstate__(self, state):
         args, kwargs = state

zipp/_functools.py

@@ -0,0 +1,20 @@
+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')
+
+    @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