Allow Wheel to work when no version is specified in filename

python-poetry · Nov 24, 2020 · 8976f1e · 8976f1e
1 parent bf30ca6
commit 8976f1e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/poetry/installation/chooser.py b/poetry/installation/chooser.py
@@ -24,7 +24,9 @@ def __init__(self, filename):  # type: (str) -> None
 
         self.filename = filename
         self.name = wheel_info.group("name").replace("_", "-")
-        self.version = wheel_info.group("ver").replace("_", "-")
+        self.version = wheel_info.group("ver")
+        if self.version:
+            self.version = self.version.replace("_", "-")
         self.build_tag = wheel_info.group("build")
         self.pyversions = wheel_info.group("pyver").split(".")
         self.abis = wheel_info.group("abi").split(".")

diff --git a/tests/installation/test_chooser.py b/tests/installation/test_chooser.py
@@ -7,7 +7,7 @@
 from packaging.tags import Tag
 
 from poetry.core.packages.package import Package
-from poetry.installation.chooser import Chooser
+from poetry.installation.chooser import Chooser, Wheel
 from poetry.repositories.legacy_repository import LegacyRepository
 from poetry.repositories.pool import Pool
 from poetry.repositories.pypi_repository import PyPiRepository
@@ -196,3 +196,7 @@ def test_chooser_chooses_distributions_that_match_the_package_hashes(
     link = chooser.choose_for(package)
 
     assert "isort-4.3.4.tar.gz" == link.filename
+
+
+def test_wheel_handles_no_version(env, pool):
+    Wheel("tiamat-6-py3-none-any.whl")