| |
| """Wait for every Python release artifact to become visible on PyPI.""" |
|
|
| import argparse |
| import http.client |
| import json |
| import time |
| import urllib.request |
|
|
| from packaging.version import Version |
|
|
| RUNTIME_PLATFORM_TAGS = { |
| "macosx_10_9_x86_64", |
| "macosx_11_0_arm64", |
| "manylinux_2_17_aarch64", |
| "manylinux_2_17_x86_64", |
| "musllinux_1_1_aarch64", |
| "musllinux_1_1_x86_64", |
| "win_amd64", |
| "win_arm64", |
| } |
|
|
|
|
| def verify_release(package: str, version: str) -> None: |
| version = str(Version(version)) |
| name = package.replace("-", "_") |
| if package == "openai-codex-cli-bin": |
| expected = { |
| f"{name}-{version}-py3-none-{tag}.whl" for tag in RUNTIME_PLATFORM_TAGS |
| } |
| else: |
| expected = {f"{name}-{version}-py3-none-any.whl", f"{name}-{version}.tar.gz"} |
|
|
| for attempt in range(30): |
| try: |
| with urllib.request.urlopen( |
| f"https://pypi.org/pypi/{package}/{version}/json", timeout=30 |
| ) as response: |
| data = json.load(response) |
| if not isinstance(data, dict) or not isinstance(data.get("urls"), list): |
| raise ValueError("Expected a PyPI response with a urls list") |
| if any( |
| not isinstance(file, dict) |
| or not isinstance(file.get("filename"), str) |
| for file in data["urls"] |
| ): |
| raise ValueError("Expected each PyPI file to have a filename") |
| actual = {file["filename"] for file in data["urls"]} |
| except ( |
| OSError, |
| http.client.HTTPException, |
| ValueError, |
| ) as error: |
| print(f"Could not read {package} {version} from PyPI: {error}.") |
| else: |
| if actual == expected: |
| print(f"All {package} {version} files are available on PyPI.") |
| return |
| print(f"Missing files: {sorted(expected - actual)}") |
| print(f"Unexpected files: {sorted(actual - expected)}") |
| if attempt < 29: |
| time.sleep(10) |
|
|
| raise SystemExit(f"{package} {version} files did not become available on PyPI.") |
|
|
|
|
| def main() -> None: |
| parser = argparse.ArgumentParser(description=__doc__) |
| parser.add_argument("package", choices=["openai-codex-cli-bin", "openai-codex"]) |
| parser.add_argument("version") |
| args = parser.parse_args() |
| verify_release(args.package, args.version) |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
