Support for parsin setup.cfg and python-version, detection of parser

Author: amol-

rsconnect/pyproject.py

@@ -7,6 +7,7 @@
 
 import pathlib
 import typing
+import configparser
 
 try:
     import tomllib
@@ -24,14 +25,31 @@ def lookup_metadata_file(directory: typing.Union[str, pathlib.Path]) -> typing.L
     directory = pathlib.Path(directory)
 
     def _generate():
-        for filename in ("pyproject.toml", "setup.cfg", ".python-version"):
+        for filename in (".python-version", "pyproject.toml", "setup.cfg"):
             path = directory / filename
             if path.is_file():
                 yield (filename, path)
 
     return list(_generate())
 
 
+def get_python_requires_parser(
+    metadata_file: pathlib.Path,
+) -> typing.Optional[typing.Callable[[pathlib.Path], typing.Optional[str]]]:
+    """Given the metadata file, return the appropriate parser function.
+
+    The returned function takes a pathlib.Path and returns the parsed value.
+    """
+    if metadata_file.name == "pyproject.toml":
+        return parse_pyproject_python_requires
+    elif metadata_file.name == "setup.cfg":
+        return parse_setupcfg_python_requires
+    elif metadata_file.name == ".python-version":
+        return parse_pyversion_python_requires
+    else:
+        return None
+
+
 def parse_pyproject_python_requires(pyproject_file: pathlib.Path) -> typing.Optional[str]:
     """Parse the project.requires-python field from a pyproject.toml file.
 
@@ -43,3 +61,27 @@ def parse_pyproject_python_requires(pyproject_file: pathlib.Path) -> typing.Opti
     pyproject = tomllib.loads(content)
 
     return pyproject.get("project", {}).get("requires-python", None)
+
+
+def parse_setupcfg_python_requires(setupcfg_file: pathlib.Path) -> typing.Optional[str]:
+    """Parse the python_requires field from a setup.cfg file.
+
+    Assumes that the setup.cfg file exists, is accessible and well formatted.
+
+    Returns None if the field is not found.
+    """
+    config = configparser.ConfigParser()
+    config.read(setupcfg_file)
+
+    return config.get("options", "python_requires", fallback=None)
+
+
+def parse_pyversion_python_requires(pyversion_file: pathlib.Path) -> typing.Optional[str]:
+    """Parse the python version from a .python-version file.
+
+    Assumes that the .python-version file exists, is accessible and well formatted.
+
+    Returns None if the field is not found.
+    """
+    content = pyversion_file.read_text()
+    return content.strip()

tests/test_pyproject.py

@@ -1,7 +1,14 @@
 import os
 import pathlib
+import tempfile
 
-from rsconnect.pyproject import lookup_metadata_file, parse_pyproject_python_requires
+from rsconnect.pyproject import (
+    lookup_metadata_file,
+    parse_pyproject_python_requires,
+    parse_setupcfg_python_requires,
+    parse_pyversion_python_requires,
+    get_python_requires_parser,
+)
 
 import pytest
 
@@ -17,11 +24,12 @@
         (
             os.path.join(PROJECTS_DIRECTORY, "using_pyversion"),
             (
-                "pyproject.toml",
                 ".python-version",
+                "pyproject.toml",
+                "setup.cfg",
             ),
         ),
-        (os.path.join(PROJECTS_DIRECTORY, "allofthem"), ("pyproject.toml", "setup.cfg", ".python-version")),
+        (os.path.join(PROJECTS_DIRECTORY, "allofthem"), (".python-version", "pyproject.toml", "setup.cfg")),
     ],
     ids=["pyproject.toml", "setup.cfg", ".python-version", "allofthem"],
 )
@@ -30,6 +38,22 @@ def test_python_project_metadata_detect(project_dir, expected):
     assert lookup_metadata_file(project_dir) == expectation
 
 
+@pytest.mark.parametrize(
+    "filename, expected_parser",
+    [
+        ("pyproject.toml", parse_pyproject_python_requires),
+        ("setup.cfg", parse_setupcfg_python_requires),
+        (".python-version", parse_pyversion_python_requires),
+        ("invalid.txt", None),
+    ],
+    ids=["pyproject.toml", "setup.cfg", ".python-version", "invalid"],
+)
+def test_get_python_requires_parser(filename, expected_parser):
+    metadata_file = pathlib.Path(PROJECTS_DIRECTORY) / filename
+    parser = get_python_requires_parser(metadata_file)
+    assert parser == expected_parser
+
+
 @pytest.mark.parametrize(
     "project_dir",
     [
@@ -53,3 +77,30 @@ def test_python_project_metadata_missing(project_dir):
 def test_pyprojecttoml_python_requires(project_dir, expected):
     pyproject_file = pathlib.Path(project_dir) / "pyproject.toml"
     assert parse_pyproject_python_requires(pyproject_file) == expected
+
+
+@pytest.mark.parametrize(
+    "project_dir, expected",
+    [
+        (os.path.join(PROJECTS_DIRECTORY, "using_setupcfg"), ">=3.8"),
+        (os.path.join(PROJECTS_DIRECTORY, "using_pyversion"), None),
+    ],
+    ids=["option-exists", "option-missing"],
+)
+def test_setupcfg_python_requires(tmp_path, project_dir, expected):
+    setupcfg_file = pathlib.Path(project_dir) / "setup.cfg"
+    assert parse_setupcfg_python_requires(setupcfg_file) == expected
+
+
+@pytest.mark.parametrize(
+    "project_dir, expected",
+    [
+        (os.path.join(PROJECTS_DIRECTORY, "using_pyversion"), ">=3.8, <3.12"),
+        # There is no case (option-missing) where the .python-version file is empty,
+        # so we don't test that.
+    ],
+    ids=["option-exists"],
+)
+def test_pyversion_python_requires(tmp_path, project_dir, expected):
+    versionfile = pathlib.Path(project_dir) / ".python-version"
+    assert parse_pyversion_python_requires(versionfile) == expected

tests/testdata/python-project/using_pyversion/setup.cfg

@@ -0,0 +1,4 @@
+[metadata]
+name = python-project
+version = 0.1.0
+description = Add your description here