Migrate pkg_resources -> importlib_resources

coveo · Dec 7, 2023 · 2fdd972 · 2fdd972
1 parent a078e09
commit 2fdd972
Show file tree

Hide file tree

Showing 3 changed files with 43 additions and 2 deletions.
diff --git a/coveo_stew/ci/mypy_runner.py b/coveo_stew/ci/mypy_runner.py
@@ -1,3 +1,6 @@
+import atexit
+from contextlib import ExitStack
+import importlib_resources
 import re
 from pathlib import Path
 from typing import Generator, Optional, Union
@@ -28,7 +31,11 @@ def _mypy_config_path(self) -> Optional[Path]:
             return None
 
         if self.set_config is True:
-            return Path(pkg_resources.resource_filename("coveo_stew", "package_resources/mypy.ini"))
+            config_ref = importlib_resources.files("coveo_stew") / "package_resources/mypy.ini"
+            stack = ExitStack()
+            atexit.register(stack)
+            config_path = stack.enter_context(importlib_resources.as_file(config_ref))
+            return config_path
 
         assert isinstance(self.set_config, str)  # mypy
         return self._pyproject.project_path / self.set_config

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -23,6 +23,7 @@ junit-xml = "*"
 packaging = "*"
 toml = "*"
 tomlkit = "*"
+importlib-resources = "^6.1.1"
 
 
 [tool.poetry.dev-dependencies]