Fix markers evaluation for python_version with precision < 3

python-poetry · Apr 9, 2019 · 46ad807 · 46ad807
1 parent 2515679
commit 46ad807
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/poetry/packages/__init__.py b/poetry/packages/__init__.py
@@ -1,6 +1,7 @@
 import os
 import re
 
+from poetry.semver import Version
 from poetry.version.requirements import Requirement
 
 from .dependency import Dependency
@@ -105,6 +106,22 @@ def dependency_from_pep_508(name):
                     op = ""
                 elif op == "!=":
                     version += ".*"
+                elif op in ("<=", ">"):
+                    parsed_version = Version.parse(version)
+                    if parsed_version.precision == 1:
+                        if op == "<=":
+                            op = "<"
+                            version = parsed_version.next_major.text
+                        elif op == ">":
+                            op = ">="
+                            version = parsed_version.next_major.text
+                    elif parsed_version.precision == 2:
+                        if op == "<=":
+                            op = "<"
+                            version = parsed_version.next_minor.text
+                        elif op == ">":
+                            op = ">="
+                            version = parsed_version.next_minor.text
                 elif op in ("in", "not in"):
                     versions = []
                     for v in re.split("[ ,]+", version):