Fix and improve wheel installation

poetry/console/commands/source/add.py

@@ -11,7 +11,6 @@
 from poetry.config.source import Source
 from poetry.console.commands.command import Command
 from poetry.factory import Factory
-from poetry.repositories import Pool
 
 
 class SourceAddCommand(Command):
@@ -89,13 +88,14 @@ def handle(self) -> Optional[int]:
             self.line(f"Adding source with name <c1>{name}</c1>.")
             sources.append(self.source_to_table(new_source))
 
+        self.poetry.config.merge(
+            {"sources": {source["name"]: source for source in sources}}
+        )
+
         # ensure new source is valid. eg: invalid name etc.
-        self.poetry._pool = Pool()
         try:
-            Factory.configure_sources(
-                self.poetry, sources, self.poetry.config, NullIO()
-            )
-            self.poetry.pool.repository(name)
+            pool = Factory.create_pool(self.poetry.config, NullIO())
+            pool.repository(name)
         except ValueError as e:
             self.line_error(
                 f"<error>Failed to validate addition of <c1>{name}</c1>: {e}</error>"

poetry/installation/executor.py

@@ -488,7 +488,7 @@ def _execute_uninstall(self, operation: Uninstall) -> int:
         )
         self._write(operation, message)
 
-        return self._remove(operation)
+        return self._remove(operation.package)
 
     def _install(self, operation: Union[Install, Update]) -> int:
         package = operation.package
@@ -520,9 +520,7 @@ def _install(self, operation: Union[Install, Update]) -> int:
     def _update(self, operation: Union[Install, Update]) -> int:
         return self._install(operation)
 
-    def _remove(self, operation: Uninstall) -> int:
-        package = operation.package
-
+    def _remove(self, package: "Package") -> int:
         # If we have a VCS package, remove its source directory
         if package.source_type == "git":
             src_dir = self._env.path / "src" / package.name
@@ -597,9 +595,12 @@ def _install_directory(self, operation: Union[Install, Update]) -> int:
         archive = self._prepare_archive(operation)
 
         try:
-            return self.pip_install(
-                str(archive), upgrade=operation.job_type == "update"
-            )
+            if operation.job_type == "update":
+                # Uninstall first
+                # TODO: Make an uninstaller and find a way to rollback in case the new package can't be installed
+                self._remove(operation.initial_package)
+
+            self._wheel_installer.install(archive)
         finally:
             archive.unlink()
 

poetry/installation/wheel_installer.py

@@ -3,9 +3,14 @@
 
 from pathlib import Path
 from typing import TYPE_CHECKING
+from typing import Iterator
+from typing import Tuple
 from typing import Union
+from typing import cast
 
 from installer.destinations import SchemeDictionaryDestination as BaseDestination
+from installer.records import parse_record_file
+from installer.sources import WheelFile as BaseWheelFile
 
 from poetry import __version__
 from poetry.utils._compat import WINDOWS
@@ -20,6 +25,29 @@
     from poetry.utils.env import Env
 
 
+class WheelFile(BaseWheelFile):
+    def get_contents(
+        self,
+    ) -> Iterator[Tuple[Tuple[Union[Path, str], str, str], "BinaryIO"]]:
+        record_lines = self.read_dist_info("RECORD").splitlines()
+        records = parse_record_file(record_lines)
+        record_mapping = {record[0]: record for record in records}
+
+        for item in self._zipfile.infolist():
+            if item.is_dir():
+                continue
+
+            record = record_mapping.pop(item.filename)
+            assert record is not None, "In {}, {} is not mentioned in RECORD".format(
+                self._zipfile.filename,
+                item.filename,
+            )  # should not happen for valid wheels
+
+            with self._zipfile.open(item) as stream:
+                stream_casted = cast("BinaryIO", stream)
+                yield record, stream_casted
+
+
 class WheelDestination(BaseDestination):
     def write_to_fs(
         self, scheme: "Scheme", path: Union[Path, str], stream: "BinaryIO"
@@ -62,14 +90,18 @@ def __init__(self, env: "Env") -> None:
 
     def install(self, wheel: Path) -> None:
         from installer import install
-        from installer.sources import WheelFile
 
         with WheelFile.open(wheel.as_posix()) as source:
-            install(
-                source=source,
-                destination=self._destination,
-                # Additional metadata that is generated by the installation tool.
-                additional_metadata={
-                    "INSTALLER": f"Poetry {__version__}".encode(),
-                },
-            )
+            try:
+                install(
+                    source=source,
+                    destination=self._destination,
+                    # Additional metadata that is generated by the installation tool.
+                    additional_metadata={
+                        "INSTALLER": f"Poetry {__version__}".encode(),
+                    },
+                )
+            except Exception as e:
+                print(e)
+
+                raise

tests/installation/test_executor.py

@@ -6,6 +6,7 @@
 import shutil
 
 from pathlib import Path
+from urllib.parse import urlparse
 
 import pytest
 
@@ -20,6 +21,7 @@
 from poetry.installation.operations import Install
 from poetry.installation.operations import Uninstall
 from poetry.installation.operations import Update
+from poetry.installation.wheel_installer import WheelInstaller
 from poetry.repositories.pool import Pool
 from poetry.utils.env import MockEnv
 from tests.repositories.test_pypi_repository import MockRepository
@@ -95,9 +97,15 @@ def pool():
 @pytest.fixture()
 def mock_file_downloads(http):
     def callback(request, uri, headers):
+        name = Path(urlparse(uri).path).name
+
         fixture = Path(__file__).parent.parent.joinpath(
-            "fixtures/distributions/demo-0.1.0-py2.py3-none-any.whl"
+            "repositories/fixtures/pypi.org/dists/" + name
         )
+        if not fixture.exists():
+            fixture = Path(__file__).parent.parent.joinpath(
+                "fixtures/distributions/demo-0.1.0-py2.py3-none-any.whl"
+            )
 
         with fixture.open("rb") as f:
             return [200, headers, f.read()]
@@ -129,6 +137,7 @@ def test_execute_executes_a_batch_of_operations(
     pip_editable_install = mocker.patch(
         "poetry.installation.executor.pip_editable_install", unsafe=not PY36
     )
+    wheel_install = mocker.patch.object(WheelInstaller, "install")
 
     config = Config()
     config.merge({"cache-dir": tmp_dir})
@@ -199,9 +208,11 @@ def test_execute_executes_a_batch_of_operations(
     expected = set(expected.splitlines())
     output = set(io.fetch_output().splitlines())
     assert expected == output
-    assert 5 == len(env.executed)
+    # One pip uninstall command
+    assert 1 == len(env.executed)
     assert 0 == return_code
     pip_editable_install.assert_called_once()
+    assert wheel_install.call_count == 4
 
 
 def test_execute_shows_skipped_operations_if_verbose(
@@ -248,30 +259,25 @@ def test_execute_should_show_errors(config, mocker, io, env):
 
 
 def test_execute_works_with_ansi_output(
-    mocker, config, pool, io_decorated, tmp_dir, mock_file_downloads, env
+    config, pool, io_decorated, tmp_dir, mock_file_downloads, env, wheel
 ):
     config = Config()
     config.merge({"cache-dir": tmp_dir})
 
     executor = Executor(env, pool, config, io_decorated)
 
-    install_output = (
-        "some string that does not contain a keyb0ard !nterrupt or cance11ed by u$er"
-    )
-    mocker.patch.object(env, "_run", return_value=install_output)
     return_code = executor.execute(
         [
-            Install(Package("pytest", "3.5.2")),
+            Install(Package("clikit", "0.2.4")),
         ]
     )
-    env._run.assert_called_once()
 
     expected = [
         "\x1b[39;1mPackage operations\x1b[39;22m: \x1b[34m1\x1b[39m install, \x1b[34m0\x1b[39m updates, \x1b[34m0\x1b[39m removals",
-        "\x1b[34;1m•\x1b[39;22m \x1b[39mInstalling \x1b[39m\x1b[36mpytest\x1b[39m\x1b[39m (\x1b[39m\x1b[39;1m3.5.2\x1b[39;22m\x1b[39m)\x1b[39m: \x1b[34mPending...\x1b[39m",
-        "\x1b[34;1m•\x1b[39;22m \x1b[39mInstalling \x1b[39m\x1b[36mpytest\x1b[39m\x1b[39m (\x1b[39m\x1b[39;1m3.5.2\x1b[39;22m\x1b[39m)\x1b[39m: \x1b[34mDownloading...\x1b[39m",
-        "\x1b[34;1m•\x1b[39;22m \x1b[39mInstalling \x1b[39m\x1b[36mpytest\x1b[39m\x1b[39m (\x1b[39m\x1b[39;1m3.5.2\x1b[39;22m\x1b[39m)\x1b[39m: \x1b[34mInstalling...\x1b[39m",
-        "\x1b[32;1m•\x1b[39;22m \x1b[39mInstalling \x1b[39m\x1b[36mpytest\x1b[39m\x1b[39m (\x1b[39m\x1b[32m3.5.2\x1b[39m\x1b[39m)\x1b[39m",  # finished
+        "\x1b[34;1m•\x1b[39;22m \x1b[39mInstalling \x1b[39m\x1b[36mclikit\x1b[39m\x1b[39m (\x1b[39m\x1b[39;1m0.2.4\x1b[39;22m\x1b[39m)\x1b[39m: \x1b[34mPending...\x1b[39m",
+        "\x1b[34;1m•\x1b[39;22m \x1b[39mInstalling \x1b[39m\x1b[36mclikit\x1b[39m\x1b[39m (\x1b[39m\x1b[39;1m0.2.4\x1b[39;22m\x1b[39m)\x1b[39m: \x1b[34mDownloading...\x1b[39m",
+        "\x1b[34;1m•\x1b[39;22m \x1b[39mInstalling \x1b[39m\x1b[36mclikit\x1b[39m\x1b[39m (\x1b[39m\x1b[39;1m0.2.4\x1b[39;22m\x1b[39m)\x1b[39m: \x1b[34mInstalling...\x1b[39m",
+        "\x1b[32;1m•\x1b[39;22m \x1b[39mInstalling \x1b[39m\x1b[36mclikit\x1b[39m\x1b[39m (\x1b[39m\x1b[32m0.2.4\x1b[39m\x1b[39m)\x1b[39m",  # finished
     ]
     output = io_decorated.fetch_output()
     # hint: use print(repr(output)) if you need to debug this
@@ -282,28 +288,23 @@ def test_execute_works_with_ansi_output(
 
 
 def test_execute_works_with_no_ansi_output(
-    mocker, config, pool, io_not_decorated, tmp_dir, mock_file_downloads, env
+    config, pool, io_not_decorated, tmp_dir, mock_file_downloads, env
 ):
     config = Config()
     config.merge({"cache-dir": tmp_dir})
 
     executor = Executor(env, pool, config, io_not_decorated)
 
-    install_output = (
-        "some string that does not contain a keyb0ard !nterrupt or cance11ed by u$er"
-    )
-    mocker.patch.object(env, "_run", return_value=install_output)
     return_code = executor.execute(
         [
-            Install(Package("pytest", "3.5.2")),
+            Install(Package("clikit", "0.2.4")),
         ]
     )
-    env._run.assert_called_once()
 
     expected = """
 Package operations: 1 install, 0 updates, 0 removals
 
-  • Installing pytest (3.5.2)
+  • Installing clikit (0.2.4)
 """
     expected = set(expected.splitlines())
     output = set(io_not_decorated.fetch_output().splitlines())