Merge branch 'bugfix/pickleable-ep' into 'master'

jaraco · jaraco · commit e226573031df · 2019-12-01T15:28:10.000Z
Allow entry points to be pickled. Closes #96 See merge request python-devs/importlib_metadata!99
diff --git a/importlib_metadata/__init__.py b/importlib_metadata/__init__.py
@@ -123,6 +123,12 @@ def __iter__(self):
         """
         return iter((self.name, self))
 
+    def __reduce__(self):
+        return (
+            self.__class__,
+            (self.name, self.value, self.group),
+            )
+
 
 class PackagePath(pathlib.PurePosixPath):
     """A reference to a path in a package"""
diff --git a/importlib_metadata/docs/changelog.rst b/importlib_metadata/docs/changelog.rst
@@ -6,6 +6,7 @@ v1.1.0
 ======
 
 * Dropped support for Python 3.4.
+* EntryPoints are now pickleable. Closes #96.
 
 v1.0.0
 ======
diff --git a/importlib_metadata/tests/test_main.py b/importlib_metadata/tests/test_main.py
@@ -2,6 +2,7 @@
 from __future__ import unicode_literals
 
 import re
+import pickle
 import textwrap
 import unittest
 import importlib
@@ -189,3 +190,10 @@ def test_egg(self):
         with self.add_sys_path(egg):
             with self.assertRaises(PackageNotFoundError):
                 version('foo')
+
+
+class TestEntryPoints(unittest.TestCase):
+    def test_entry_point_pickleable(self):
+        ep = importlib_metadata.EntryPoint('name', 'value', 'group')
+        revived = pickle.loads(pickle.dumps(ep))
+        assert revived == ep
