python 3.11 compat

src/poetry/core/pyproject/toml.py

@@ -4,9 +4,8 @@
 from pathlib import Path
 from typing import Any
 
-import tomli
-
 from poetry.core.pyproject.tables import BuildSystem
+from poetry.core.utils._compat import tomllib
 
 
 class PyProjectTOML:
@@ -26,7 +25,7 @@ def data(self) -> dict[str, Any]:
                 self._data = {}
             else:
                 with self.file.open("rb") as f:
-                    self._data = tomli.load(f)
+                    self._data = tomllib.load(f)
 
         return self._data
 

src/poetry/core/utils/_compat.py

@@ -4,3 +4,11 @@
 
 
 WINDOWS = sys.platform == "win32"
+
+if sys.version_info < (3, 11):
+    # compatibility for python <3.11
+    import tomli as tomllib
+else:
+    import tomllib  # nopycln: import
+
+__all__ = ["tomllib"]

tests/pyproject/test_pyproject_toml.py

@@ -3,17 +3,17 @@
 from pathlib import Path
 
 import pytest
-import tomli
 
 from poetry.core.pyproject.exceptions import PyProjectException
 from poetry.core.pyproject.toml import PyProjectTOML
+from poetry.core.utils._compat import tomllib
 
 
 def test_pyproject_toml_simple(
     pyproject_toml: Path, build_system_section: str, poetry_section: str
 ) -> None:
     with pyproject_toml.open("rb") as f:
-        data = tomli.load(f)
+        data = tomllib.load(f)
     assert PyProjectTOML(pyproject_toml).data == data
 
 
@@ -35,7 +35,7 @@ def test_pyproject_toml_poetry_config(
 ) -> None:
     pyproject = PyProjectTOML(pyproject_toml)
     with pyproject_toml.open("rb") as f:
-        doc = tomli.load(f)
+        doc = tomllib.load(f)
     config = doc["tool"]["poetry"]
 
     assert pyproject.is_poetry_project()

tests/test_factory.py

@@ -6,13 +6,13 @@
 from typing import cast
 
 import pytest
-import tomli
 
 from packaging.utils import canonicalize_name
 
 from poetry.core.constraints.version import parse_constraint
 from poetry.core.factory import Factory
 from poetry.core.packages.url_dependency import URLDependency
+from poetry.core.utils._compat import tomllib
 from poetry.core.version.markers import SingleMarker
 
 
@@ -187,7 +187,7 @@ def test_create_poetry_with_multi_constraints_dependency() -> None:
 def test_validate() -> None:
     complete = fixtures_dir / "complete.toml"
     with complete.open("rb") as f:
-        doc = tomli.load(f)
+        doc = tomllib.load(f)
     content = doc["tool"]["poetry"]
 
     assert Factory.validate(content) == {"errors": [], "warnings": []}
@@ -196,7 +196,7 @@ def test_validate() -> None:
 def test_validate_fails() -> None:
     complete = fixtures_dir / "complete.toml"
     with complete.open("rb") as f:
-        doc = tomli.load(f)
+        doc = tomllib.load(f)
     content = doc["tool"]["poetry"]
     content["authors"] = "this is not a valid array"
 
@@ -210,7 +210,7 @@ def test_validate_without_strict_fails_only_non_strict() -> None:
         fixtures_dir / "project_failing_strict_validation" / "pyproject.toml"
     )
     with project_failing_strict_validation.open("rb") as f:
-        doc = tomli.load(f)
+        doc = tomllib.load(f)
     content = doc["tool"]["poetry"]
 
     assert Factory.validate(content) == {
@@ -229,7 +229,7 @@ def test_validate_strict_fails_strict_and_non_strict() -> None:
         fixtures_dir / "project_failing_strict_validation" / "pyproject.toml"
     )
     with project_failing_strict_validation.open("rb") as f:
-        doc = tomli.load(f)
+        doc = tomllib.load(f)
     content = doc["tool"]["poetry"]
 
     assert Factory.validate(content, strict=True) == {
@@ -271,7 +271,7 @@ def test_validate_strict_fails_strict_and_non_strict() -> None:
 def test_strict_validation_success_on_multiple_readme_files() -> None:
     with_readme_files = fixtures_dir / "with_readme_files" / "pyproject.toml"
     with with_readme_files.open("rb") as f:
-        doc = tomli.load(f)
+        doc = tomllib.load(f)
     content = doc["tool"]["poetry"]
 
     assert Factory.validate(content, strict=True) == {"errors": [], "warnings": []}
@@ -280,7 +280,7 @@ def test_strict_validation_success_on_multiple_readme_files() -> None:
 def test_strict_validation_fails_on_readme_files_with_unmatching_types() -> None:
     with_readme_files = fixtures_dir / "with_readme_files" / "pyproject.toml"
     with with_readme_files.open("rb") as f:
-        doc = tomli.load(f)
+        doc = tomllib.load(f)
     content = doc["tool"]["poetry"]
     content["readme"][0] = "README.md"
 

tests/testutils.py

@@ -11,9 +11,10 @@
 from typing import Any
 from typing import Generator
 
-import tomli
 import tomli_w
 
+from poetry.core.utils._compat import tomllib
+
 
 __toml_build_backend_patch__ = {
     "build-system": {
@@ -46,7 +47,7 @@ def temporary_project_directory(
         shutil.copytree(str(path), dst)
         toml = dst / "pyproject.toml"
         with toml.open("rb") as f:
-            data = tomli.load(f)
+            data = tomllib.load(f)
         data.update(toml_patch or __toml_build_backend_patch__)
         with toml.open("wb") as f:
             tomli_w.dump(data, f)