Adapt python-requires versions

Author: amol-

.vscode/settings.json

@@ -24,4 +24,7 @@
         "build/**": true,
         "venv/**": true,
     },
+    "python.analysis.exclude": [
+        "tests"
+    ],
 }

rsconnect/pyproject.py

@@ -5,16 +5,20 @@
 but not from setup.py due to its dynamic nature.
 """
 
+import configparser
 import pathlib
+import re
 import typing
-import configparser
 
 try:
     import tomllib
 except ImportError:
     # Python 3.11+ has tomllib in the standard library
     import toml as tomllib  # type: ignore[no-redef]
 
+PEP440_OPERATORS_REGEX = r"(===|==|!=|<=|>=|<|>|~=)"
+VALID_VERSION_REQ_REGEX = rf"^({PEP440_OPERATORS_REGEX}?\d+(\.[\d\*]+)*)+$"
+
 
 def detect_python_version_requirement(directory: typing.Union[str, pathlib.Path]) -> typing.Optional[str]:
     """Detect the python version requirement for a project.
@@ -103,5 +107,42 @@ def parse_pyversion_python_requires(pyversion_file: pathlib.Path) -> typing.Opti
 
     Returns None if the field is not found.
     """
-    content = pyversion_file.read_text()
-    return content.strip()
+    return adapt_python_requires(pyversion_file.read_text().strip())
+
+
+def adapt_python_requires(
+    python_requires: str,
+) -> str:
+    """Convert a literal python version to a PEP440 constraint.
+
+    Connect expects a PEP440 format, but the .python-version file can contain
+    plain version numbers and other formats.
+
+    We should convert them to the constraints that connect expects.
+    """
+    current_contraints = python_requires.split(",")
+
+    def _adapt_contraint(constraints: list[str]) -> typing.Generator[str, None, None]:
+        for constraint in constraints:
+            constraint = constraint.strip()
+            if "@" in constraint or "-" in constraint or "/" in constraint:
+                raise ValueError(
+                    f"Invalid python version, python specific implementations are not supported: {constraint}"
+                )
+
+            if "b" in constraint or "rc" in constraint or "a" in constraint:
+                raise ValueError(f"Invalid python version, pre-release versions are not supported: {constraint}")
+
+            if re.match(VALID_VERSION_REQ_REGEX, constraint) is None:
+                raise ValueError(f"Invalid python version: {constraint}")
+
+            if re.search(PEP440_OPERATORS_REGEX, constraint):
+                yield constraint
+            else:
+                # Convert to PEP440 format
+                if "*" in constraint:
+                    yield f"=={constraint}"
+                else:
+                    yield f"~={constraint.rstrip('0').rstrip('.')}"  # Remove trailing zeros and dots
+
+    return ",".join(_adapt_contraint(current_contraints))

tests/test_pyproject.py

@@ -1,17 +1,18 @@
 import os
 import pathlib
+import tempfile
+
+import pytest
 
 from rsconnect.pyproject import (
+    detect_python_version_requirement,
+    get_python_version_requirement_parser,
     lookup_metadata_file,
     parse_pyproject_python_requires,
-    parse_setupcfg_python_requires,
     parse_pyversion_python_requires,
-    get_python_version_requirement_parser,
-    detect_python_version_requirement,
+    parse_setupcfg_python_requires,
 )
 
-import pytest
-
 HERE = os.path.dirname(__file__)
 PROJECTS_DIRECTORY = os.path.abspath(os.path.join(HERE, "testdata", "python-project"))
 
@@ -142,3 +143,52 @@ def test_detect_python_version_requirement():
     assert detect_python_version_requirement(project_dir) == ">=3.8, <3.12"
 
     assert detect_python_version_requirement(os.path.join(PROJECTS_DIRECTORY, "empty")) is None
+
+
+@pytest.mark.parametrize(  # type: ignore
+    ["content", "expected"],
+    [
+        ("3.8", "~=3.8"),
+        ("3.8.0", "~=3.8"),
+        ("3.8.0b1", ValueError("Invalid python version, pre-release versions are not supported: 3.8.0b1")),
+        ("3.8.0rc1", ValueError("Invalid python version, pre-release versions are not supported: 3.8.0rc1")),
+        ("3.8.0a1", ValueError("Invalid python version, pre-release versions are not supported: 3.8.0a1")),
+        ("3.8.*", "==3.8.*"),
+        ("3.*", "==3.*"),
+        ("*", ValueError("Invalid python version: *")),
+        # This is not perfect, but the added regex complexity doesn't seem worth it.
+        ("invalid", ValueError("Invalid python version, pre-release versions are not supported: invalid")),
+        ("[email protected]", ValueError("Invalid python version, python specific implementations are not supported: [email protected]")),
+        (
+            "cpython-3.12.3-macos-aarch64-none",
+            ValueError(
+                "Invalid python version, python specific implementations are not supported: "
+                "cpython-3.12.3-macos-aarch64-none"
+            ),
+        ),
+        (
+            "/usr/bin/python3.8",
+            ValueError("Invalid python version, python specific implementations are not supported: /usr/bin/python3.8"),
+        ),
+    ],
+)
+def test_python_version_file_adapt(content, expected):
+    """Test that the python version is correctly converted to a PEP440 format.
+
+    Connect expects a PEP440 format, but the .python-version file can contain
+    plain version numbers and other formats.
+
+    We should convert them to the constraints that connect expects.
+    """
+    with tempfile.NamedTemporaryFile(mode="w+") as tmpfile:
+        tmpfile.write(content)
+        tmpfile.flush()
+
+        versionfile = pathlib.Path(tmpfile.name)
+
+        if isinstance(expected, Exception):
+            with pytest.raises(expected.__class__) as excinfo:
+                parse_pyversion_python_requires(versionfile)
+            assert str(excinfo.value) == expected.args[0]
+        else:
+            assert parse_pyversion_python_requires(versionfile) == expected