fix: DeprecationWarning: pkg_resources is deprecated as an API

Author: targetblank

mode/utils/imports.py

@@ -368,14 +368,19 @@ def load_extension_class_names(
     ```
     """
     try:
-        from pkg_resources import iter_entry_points
+        from importlib.metadata import entry_points
     except ImportError:
         return
 
-    for ep in iter_entry_points(namespace):
-        yield RawEntrypointExtension(
-            ep.name, ":".join([ep.module_name, ep.attrs[0]])
-        )
+    eps = entry_points()
+    # For Python 3.10+, entry_points() returns an object with .select()
+    if hasattr(eps, "select"):
+        for ep in eps.select(group=namespace):
+            yield RawEntrypointExtension(ep.name, f"{ep.module}:{ep.attr}")
+    else:
+        # For Python 3.8/3.9, entry_points is a dict
+        for ep in eps.get(namespace, []):
+            yield RawEntrypointExtension(ep.name, f"{ep.module}:{ep.attr}")
 
 
 @contextmanager

tests/unit/utils/test_imports.py

@@ -92,7 +92,7 @@ def test__maybe_finalize(self, *, map):
 
     def test__finalize(self, *, map):
         map.namespaces = {"foo"}
-        with patch_iter_entry_points():
+        with patch_importlib_metadata_entry_points():
             map._finalize()
         assert map.aliases["ep1"] == "foo:a"
         assert map.aliases["ep2"] == "bar:c"
@@ -158,7 +158,7 @@ def test_smart_import():
 
 
 def test_load_extension_classes():
-    with patch_iter_entry_points():
+    with patch_importlib_metadata_entry_points():
         with patch("mode.utils.imports.symbol_by_name") as sbn:
             assert list(load_extension_classes("foo")) == [
                 EntrypointExtension("ep1", sbn.return_value),
@@ -169,33 +169,37 @@ def test_load_extension_classes():
 
 
 def test_load_extension_classes_syntax_error():
-    with patch_iter_entry_points():
+    with patch_importlib_metadata_entry_points():
         with patch("mode.utils.imports.symbol_by_name") as sbn:
             sbn.side_effect = SyntaxError()
             with pytest.warns(UserWarning):
                 assert list(load_extension_classes("foo")) == []
 
 
 def test_load_extension_class_names():
-    with patch_iter_entry_points():
+    with patch_importlib_metadata_entry_points():
         assert list(load_extension_class_names("foo")) == [
             RawEntrypointExtension("ep1", "foo:a"),
             RawEntrypointExtension("ep2", "bar:c"),
         ]
 
 
 @contextmanager
-def patch_iter_entry_points():
-    with patch("pkg_resources.iter_entry_points") as iter_entry_points:
+def patch_importlib_metadata_entry_points():
+    with patch(
+        "importlib.metadata.entry_points"
+    ) as importlib_metadata_entry_points:
         ep1 = Mock(name="ep1")
         ep1.name = "ep1"
-        ep1.module_name = "foo"
-        ep1.attrs = ["a", "b"]
+        ep1.module = "foo"
+        ep1.attr = "a"
         ep2 = Mock(name="ep2")
         ep2.name = "ep2"
-        ep2.module_name = "bar"
-        ep2.attrs = ["c", "d"]
-        iter_entry_points.return_value = [ep1, ep2]
+        ep2.module = "bar"
+        ep2.attr = "c"
+        mock_entry_points = Mock()
+        mock_entry_points.select.return_value = [ep1, ep2]
+        importlib_metadata_entry_points.return_value = mock_entry_points
         yield