Include type checker versions in long description

Cf. python/typeshed#6843

Author: srittau

stub_uploader/build_wheel.py

@@ -23,6 +23,7 @@
 import argparse
 import os
 import os.path
+from pathlib import Path
 import shutil
 import subprocess
 import sys
@@ -37,6 +38,7 @@
     THIRD_PARTY_NAMESPACE,
 )
 from stub_uploader.metadata import Metadata, read_metadata
+from stub_uploader.ts_data import TypeshedData, read_typeshed_data
 
 CHANGELOG = "CHANGELOG.md"
 
@@ -108,7 +110,9 @@
 
 DESCRIPTION_OUTRO_TEMPLATE = """
 See https://github.com/python/typeshed/blob/main/README.md for more details.
-This package was generated from typeshed commit `{commit}`.
+This package was generated from typeshed commit `{commit}` and was tested
+with mypy {ts_data.mypy_version}, pyright {ts_data.pyright_version}, and
+pytype {ts_data.pytype_version}.
 """.strip()
 
 
@@ -234,7 +238,11 @@ def add_partial_marker(package_data: dict[str, list[str]], stub_dir: str) -> Non
 
 
 def generate_setup_file(
-    build_data: BuildData, metadata: Metadata, version: str, commit: str
+    build_data: BuildData,
+    ts_data: TypeshedData,
+    metadata: Metadata,
+    version: str,
+    commit: str,
 ) -> str:
     """Auto-generate a setup.py file for given distribution using a template."""
     all_requirements = [
@@ -247,7 +255,7 @@ def generate_setup_file(
         distribution=build_data.distribution,
         stub_distribution=metadata.stub_distribution,
         long_description=generate_long_description(
-            build_data.distribution, commit, metadata
+            build_data.distribution, commit, ts_data, metadata
         ),
         version=version,
         requires=all_requirements,
@@ -257,7 +265,7 @@ def generate_setup_file(
 
 
 def generate_long_description(
-    distribution: str, commit: str, metadata: Metadata
+    distribution: str, commit: str, ts_data: TypeshedData, metadata: Metadata
 ) -> str:
     extra_description = metadata.extra_description.strip()
     parts: list[str] = []
@@ -280,7 +288,7 @@ def generate_long_description(
         )
     if metadata.partial:
         parts.append(PARTIAL_STUBS_DESCRIPTION)
-    parts.append(DESCRIPTION_OUTRO_TEMPLATE.format(commit=commit))
+    parts.append(DESCRIPTION_OUTRO_TEMPLATE.format(commit=commit, ts_data=ts_data))
     return "\n\n".join(parts)
 
 
@@ -294,6 +302,7 @@ def main(
     Note: the caller should clean the temporary directory where wheel is
     created after uploading it.
     """
+    ts_data = read_typeshed_data(Path(typeshed_dir))
     build_data = BuildData(typeshed_dir, distribution)
     if build_dir:
         tmpdir = build_dir
@@ -307,7 +316,7 @@ def main(
     ).stdout.strip()
     metadata = read_metadata(typeshed_dir, distribution)
     with open(os.path.join(tmpdir, "setup.py"), "w") as f:
-        f.write(generate_setup_file(build_data, metadata, version, commit))
+        f.write(generate_setup_file(build_data, ts_data, metadata, version, commit))
     copy_stubs(build_data.stub_dir, tmpdir)
     copy_changelog(distribution, tmpdir)
     current_dir = os.getcwd()

stub_uploader/ts_data.py

@@ -0,0 +1,63 @@
+"""
+Information about typeshed.
+"""
+
+from __future__ import annotations
+import argparse
+from dataclasses import dataclass, fields
+from pathlib import Path
+from tomllib import load as toml_load
+from typing import Any
+
+
+REQUIREMENTS = "requirements-tests.txt"
+PYPROJECT = "pyproject.toml"
+
+
+@dataclass
+class TypeshedData:
+    mypy_version: str
+    pyright_version: str
+    pytype_version: str
+
+
+def read_typeshed_data(typeshed_dir: Path) -> TypeshedData:
+    pyproject = _read_pyproject_toml(typeshed_dir / PYPROJECT)
+    requirements = _read_requirements(typeshed_dir / REQUIREMENTS)
+    return TypeshedData(
+        mypy_version=requirements["mypy"],
+        pyright_version=pyproject["tool"]["typeshed"]["pyright_version"],
+        pytype_version=requirements["pytype"],
+    )
+
+
+def _read_pyproject_toml(path: Path) -> dict[str, Any]:
+    with path.open("rb") as f:
+        return toml_load(f)
+
+
+def _read_requirements(path: Path) -> dict[str, str]:
+    requirements = {}
+    with path.open() as f:
+        for line in f:
+            line = line.strip()
+            line = line.split("#")[0]  # strip comments
+            line = line.split(";")[0]  # strip extras
+            if not line.strip():  # skip empty lines
+                continue
+            if "==" in line:
+                name, version = line.split("==")
+            else:
+                name = line
+                version = ""
+            requirements[name.strip()] = version.strip()
+    return requirements
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("typeshed_dir", help="Path to typeshed checkout directory")
+    args = parser.parse_args()
+    data = read_typeshed_data(Path(args.typeshed_dir))
+    for field in fields(data):
+        print(f"{field.name}: {getattr(data, field.name)}")