ensure compatibility with Poetry 2.0 (#128)

Author: radoering

poetry.lock

@@ -16,7 +16,7 @@ include = [
 
 [tool.poetry.dependencies]
 python = "^3.9"
-poetry = "^1.5.0"
+poetry = ">=1.5.0,<3.0.0"
 
 [tool.poetry.group.dev.dependencies]
 pre-commit = ">=2.6"

pyproject.toml

@@ -55,7 +55,16 @@ def bundle(self, poetry: Poetry, io: IO) -> bool:
 
         from cleo.io.null_io import NullIO
         from poetry.core.masonry.builders.wheel import WheelBuilder
-        from poetry.core.masonry.utils.module import ModuleOrPackageNotFound
+
+        try:
+            from poetry.core.masonry.utils.module import (  # type: ignore[attr-defined]
+                ModuleOrPackageNotFoundError,
+            )
+        except ImportError:
+            # poetry-core < 2.0
+            from poetry.core.masonry.utils.module import (
+                ModuleOrPackageNotFound as ModuleOrPackageNotFoundError,
+            )
         from poetry.core.packages.package import Package
         from poetry.installation.installer import Installer
         from poetry.installation.operations.install import Install
@@ -140,7 +149,12 @@ def locked_repository(self) -> LockfileRepository:
                     package.develop = False
                 return repo
 
-        custom_locker = CustomLocker(poetry.locker.lock, poetry.local_config)
+        try:
+            locker_data = poetry.locker._pyproject_data  # type: ignore[attr-defined]
+        except AttributeError:
+            # poetry < 2.0
+            locker_data = poetry.locker._local_config
+        custom_locker = CustomLocker(poetry.locker.lock, locker_data)
 
         installer = Installer(
             NullIO() if not io.is_debug() else io,
@@ -192,7 +206,7 @@ def locked_repository(self) -> LockfileRepository:
                         source_url=str(wheel),
                     )
                     installer.executor.execute([Install(package)])
-                except ModuleOrPackageNotFound:
+                except ModuleOrPackageNotFoundError:
                     warnings.append(
                         "The root package was not installed because no matching module or"
                         " package was found."

src/poetry_plugin_bundle/bundlers/venv_bundler.py

@@ -27,7 +27,12 @@ def poetry(project_directory: str, config: Config) -> Poetry:
     p = Factory().create_poetry(
         Path(__file__).parent.parent / "fixtures" / project_directory
     )
-    p.set_locker(TestLocker(p.locker.lock, p.locker._local_config))
+    try:
+        locker_data = p.locker._pyproject_data  # type: ignore[attr-defined]
+    except AttributeError:
+        # poetry < 2.0
+        locker_data = p.locker._local_config
+    p.set_locker(TestLocker(p.locker.lock, locker_data))
 
     return p