Script to check for outdated dependencies (GSI 371)

.static_files

@@ -25,6 +25,7 @@ scripts/update_openapi_docs.py
 scripts/update_readme.py
 scripts/update_lock.py
 scripts/update_hook_revs.py
+scripts/list_outdated_dependencies.py
 scripts/README.md
 
 .github/workflows/check_config_docs.yaml

scripts/list_outdated_dependencies.py

@@ -0,0 +1,269 @@
+#!/usr/bin/env python3
+
+# Copyright 2021 - 2023 Universität Tübingen, DKFZ, EMBL, and Universität zu Köln
+# for the German Human Genome-Phenome Archive (GHGA)
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+"""Check capped dependencies for newer versions."""
+import re
+import sys
+from copy import deepcopy
+from pathlib import Path
+from typing import Any
+
+import httpx
+import stringcase
+import tomli
+from packaging.requirements import Requirement
+
+from script_utils import cli
+
+REPO_ROOT_DIR = Path(__file__).parent.parent.resolve()
+PYPROJECT_TOML_PATH = REPO_ROOT_DIR / "pyproject.toml"
+DEV_DEPS_PATH = REPO_ROOT_DIR / "requirements-dev.in"
+LOCK_FILE_PATH = REPO_ROOT_DIR / "requirements-dev.txt"
+
+
+def get_transitive_deps(exclude: set[str]) -> list[str]:
+    """Inspect the lock file to get the transitive dependencies"""
+    dependency_pattern = re.compile(r"([^=\s]+==[^\s]*?)\s")
+
+    # Get the set of dependencies from the requirements-dev.txt lock file
+    with open(LOCK_FILE_PATH, encoding="utf-8") as lock_file:
+        lines = lock_file.readlines()
+
+    dependencies = []
+    for line in lines:
+        if match := re.match(dependency_pattern, line):
+            dependency = match.group(1)
+            requirement = Requirement(dependency)
+            if requirement not in exclude:
+                dependencies.append(dependency)
+    return dependencies
+
+
+def exclude_from_dependency_list(*, package_name: str, dependencies: list) -> list:
+    """Exclude the specified package from the provided dependency list."""
+
+    return [
+        dependency
+        for dependency in dependencies
+        if not dependency.startswith(package_name)
+    ]
+
+
+def remove_self_dependencies(pyproject: dict) -> dict:
+    """Filter out self dependencies (dependencies of the package on it self) from the
+    dependencies and optional-dependencies in the provided pyproject metadata."""
+
+    if "project" not in pyproject:
+        return pyproject
+
+    modified_pyproject = deepcopy(pyproject)
+
+    project_metadata = modified_pyproject["project"]
+
+    package_name = stringcase.spinalcase(project_metadata.get("name"))
+
+    if not package_name:
+        raise ValueError("The provided project metadata does not contain a name.")
+
+    if "dependencies" in project_metadata:
+        project_metadata["dependencies"] = exclude_from_dependency_list(
+            package_name=package_name, dependencies=project_metadata["dependencies"]
+        )
+
+    if "optional-dependencies" in project_metadata:
+        for group in project_metadata["optional-dependencies"]:
+            project_metadata["optional-dependencies"][
+                group
+            ] = exclude_from_dependency_list(
+                package_name=package_name,
+                dependencies=project_metadata["optional-dependencies"][group],
+            )
+
+    return modified_pyproject
+
+
+def get_main_deps_pyproject(modified_pyproject: dict[str, Any]) -> list[str]:
+    """Get a list of the dependencies from pyproject.toml"""
+
+    dependencies: list[str] = []
+    if "dependencies" in modified_pyproject["project"]:
+        dependencies = modified_pyproject["project"]["dependencies"]
+
+    return dependencies
+
+
+def get_optional_deps_pyproject(modified_pyproject: dict[str, Any]) -> list[str]:
+    """Get a list of the optional dependencies from pyproject.toml"""
+
+    dependencies: list[str] = []
+
+    if "optional-dependencies" in modified_pyproject["project"]:
+        for optional_dependency_list in modified_pyproject["project"][
+            "optional-dependencies"
+        ]:
+            dependencies.extend(
+                modified_pyproject["project"]["optional-dependencies"][
+                    optional_dependency_list
+                ]
+            )
+
+    return dependencies
+
+
+def get_modified_pyproject() -> dict[str, Any]:
+    """Get a copy of pyproject.toml with any self-referencing dependencies removed."""
+    with open(PYPROJECT_TOML_PATH, "rb") as pyproject_toml:
+        pyproject = tomli.load(pyproject_toml)
+
+    modified_pyproject = remove_self_dependencies(pyproject)
+    return modified_pyproject
+
+
+def get_deps_dev() -> list[str]:
+    """Get a list of raw dependency strings from requirements-dev.in"""
+    with open(DEV_DEPS_PATH, encoding="utf-8") as dev_deps:
+        dependencies = [
+            line
+            for line in (line.strip() for line in dev_deps)
+            if line  # skip empty lines
+            and not line.startswith("#")  # skip comments
+            and "requirements-dev-common.in" not in line  # skip inclusion line
+        ]
+
+    return dependencies
+
+
+def get_version_from_pypi(package_name: str, client: httpx.Client) -> str:
+    """Make a call to PyPI to get the version information about `package_name`."""
+    try:
+        response = client.get(f"https://pypi.org/pypi/{package_name}/json")
+        body = response.json()
+        version = body["info"]["version"]
+    except (httpx.RequestError, KeyError):
+        cli.echo_failure(f"Unable to retrieve information for package '{package_name}'")
+        sys.exit(1)
+
+    return version
+
+
+def get_outdated_deps(
+    dependencies: list[str], strip: bool = False
+) -> list[tuple[str, ...]]:
+    """Determine which packages have updates available outside of pinned ranges."""
+    outdated: list[tuple[str, ...]] = []
+    with httpx.Client(timeout=10) as client:
+        for dependency in dependencies:
+            # Convert string into Requirement object so we can reference its parts
+            requirement = Requirement(dependency)
+
+            pypi_version = get_version_from_pypi(requirement.name, client)
+
+            specified = str(requirement.specifier)
+
+            # Strip the specifier symbols from the front of the string if desired
+            if strip:
+                specified = specified.lstrip("<=>!~")
+
+            # append package name, specified version, and latest available version
+            if not requirement.specifier.contains(pypi_version):
+                outdated.append((requirement.name, specified, pypi_version))
+    outdated.sort()
+    return outdated
+
+
+def print_table(
+    rows: list[tuple[str, ...]],
+    headers: list[str],
+    delimiter: str = " | ",
+):
+    """
+    List outdated dependencies in a formatted table.
+
+    Args:
+        `outdated`: A list of lists containing strings.
+        `headers`: A list containing the header strings for the table columns.
+    """
+    header_lengths = [len(header) for header in headers]
+
+    # Find the maximum length of each column
+    col_widths = [max(len(str(cell)) for cell in col) for col in zip(*rows)]
+
+    # Create a row format based on the maximum column widths
+    row_format = delimiter.join(
+        f"{{:<{max(width, header_len)}}}"
+        for width, header_len in zip(col_widths, header_lengths)
+    )
+
+    print("  " + row_format.format(*headers))
+    for dependency in rows:
+        print("  " + row_format.format(*dependency))
+
+
+def main(transitive: bool = False):
+    """Check capped dependencies for newer versions.
+
+    Examine `pyproject.toml` and `requirements-dev.in` for capped dependencies.
+    Make a call to PyPI to see if any newer versions exist.
+
+    Use `transitive` to show outdated transitive dependencies.
+    """
+    modified_pyproject: dict[str, Any] = get_modified_pyproject()
+    main_dependencies = get_main_deps_pyproject(modified_pyproject)
+    optional_dependencies = get_optional_deps_pyproject(modified_pyproject)
+    dev_dependencies = get_deps_dev()
+
+    outdated_main = get_outdated_deps(main_dependencies)
+    outdated_optional = get_outdated_deps(optional_dependencies)
+    outdated_dev = get_outdated_deps(dev_dependencies)
+
+    found_outdated = any([outdated_main, outdated_optional, outdated_dev])
+    headers = ["PACKAGE", "SPECIFIED", "AVAILABLE"]
+    if outdated_main:
+        location = PYPROJECT_TOML_PATH.name + " - dependencies"
+        cli.echo_failure(f"Outdated dependencies from {location}:")
+        print_table(outdated_main, headers)
+    if outdated_optional:
+        location = PYPROJECT_TOML_PATH.name + " - optional-dependencies"
+        cli.echo_failure(f"Outdated dependencies from {location}:")
+        print_table(outdated_optional, headers)
+    if outdated_dev:
+        cli.echo_failure(f"Outdated dependencies from {DEV_DEPS_PATH.name}:")
+        print_table(outdated_dev, headers)
+
+    if not found_outdated:
+        cli.echo_success("All top-level dependencies up to date.")
+
+    if transitive:
+        top_level: set[str] = {
+            item[0] for item in outdated_main + outdated_optional + outdated_dev
+        }
+
+        print("\nRetrieving transitive dependency information...")
+        transitive_dependencies = get_transitive_deps(exclude=top_level)
+        outdated_transitive = get_outdated_deps(transitive_dependencies, strip=True)
+
+        if outdated_transitive:
+            headers[1] = "PINNED"
+
+            cli.echo_failure("Outdated transitive dependencies:")
+            print_table(outdated_transitive, headers)
+        else:
+            cli.echo_success("All transitive dependencies up to date.")
+
+
+if __name__ == "__main__":
+    cli.run(main)