cibuildwheel/bin/make_dependency_update_pr.py

#!/usr/bin/env python3

from __future__ import annotations

import os
import sys
import textwrap
import time
from pathlib import Path
from subprocess import run

import click


def shell(cmd, *, check: bool, **kwargs):
    return run([cmd], shell=True, check=check, **kwargs)


def git_repo_has_changes():
    unstaged_changes = shell("git diff-index --quiet HEAD --", check=False).returncode != 0
    staged_changes = shell("git diff-index --quiet --cached HEAD --", check=False).returncode != 0
    return unstaged_changes or staged_changes


@click.command()
def main():
    project_root = Path(__file__).parent / ".."
    os.chdir(project_root)

    if git_repo_has_changes():
        print("Your git repo has uncommitted changes. Commit or stash before continuing.")
        sys.exit(1)

    previous_branch = shell(
        "git rev-parse --abbrev-ref HEAD", check=True, capture_output=True, encoding="utf8"
    ).stdout.strip()

    shell("git fetch origin", check=True)

    timestamp = time.strftime("%Y-%m-%dT%H-%M-%S", time.gmtime())
    branch_name = f"update-constraints-{timestamp}"

    shell(f"git checkout -b {branch_name} origin/main", check=True)

    try:
        shell("bin/update_dependencies.py", check=True)

        if not git_repo_has_changes():
            print("Done: no constraint updates required.")
            return

        shell('git commit -a -m "Update dependencies"', check=True)
        body = textwrap.dedent(
            f"""
            Update the versions of our dependencies.

            PR generated by `{os.path.basename(__file__)}`.
            """
        )
        run(
            [
                "gh",
                "pr",
                "create",
                "--repo=pypa/cibuildwheel",
                "--base=main",
                "--title=Update dependencies",
                f"--body='{body}'",
            ],
            check=True,
        )

        print("Done.")
    finally:
        # remove any local changes
        shell("git checkout -- .", check=True)
        shell(f"git checkout {previous_branch}", check=True)
        shell(f"git branch -D --force {branch_name}", check=True)


if __name__ == "__main__":
    main.main(standalone_mode=True)