Update to fix pkg_resources deprecation while maintaining back compat

moj-analytical-services · Dec 16, 2024 · cfc724c · cfc724c
1 parent ba74248
commit cfc724c
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 6 deletions.
diff --git a/iam_builder/iam_schema.py b/iam_builder/iam_schema.py
@@ -1,11 +1,24 @@
 import jsonschema
 import json
-import pkg_resources
+import importlib
 from iam_builder.exceptions import IAMValidationError
 
-IAM_SCHEMA = json.load(
-    pkg_resources.resource_stream(__name__, "schemas/iam_schema.json")
-)
+try:
+    import pkg_resources
+except ImportError:
+    import importlib
+
+
+try:
+    IAM_SCHEMA = json.load(
+        pkg_resources.resource_stream(__name__, "schemas/iam_schema.json")
+    )
+except ImportError:
+    ref = importlib_resources.files(__name__).joinpath("schemas/iam_schema.json")
+    with ref.open('rb') as fp:
+        IAM_SCHEMA = json.load(
+            fp.read()
+        )
 
 
 def validate_iam(config: dict):

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -11,6 +11,7 @@ python = "^3.7"
 PyYAML = "^6.0"
 parameterized = "0.7.*"
 jsonschema = "^4.10.0"
+importlib = "^1.0.4"
 
 [tool.poetry.scripts]
 iam_builder = "iam_builder.command_line:main"