Henry Schreiner
b98602705f
* chore: improve mypy Signed-off-by: Henry Schreiner <henryschreineriii@gmail.com> * chore(types): type functions in tests Signed-off-by: Henry Schreiner <henryschreineriii@gmail.com> * chore(types): No partial types in tests Signed-off-by: Henry Schreiner <henryschreineriii@gmail.com> --------- Signed-off-by: Henry Schreiner <henryschreineriii@gmail.com>
85 lines
2.3 KiB
Python
Executable File
85 lines
2.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import textwrap
|
|
import time
|
|
from pathlib import Path
|
|
|
|
import click
|
|
|
|
|
|
def shell(cmd: str, *, check: bool, **kwargs: object) -> subprocess.CompletedProcess[str]:
|
|
return subprocess.run([cmd], shell=True, check=check, **kwargs) # type: ignore[call-overload, no-any-return]
|
|
|
|
|
|
def git_repo_has_changes() -> bool:
|
|
unstaged_changes: bool = shell("git diff-index --quiet HEAD --", check=False).returncode != 0
|
|
staged_changes: bool = (
|
|
shell("git diff-index --quiet --cached HEAD --", check=False).returncode != 0
|
|
)
|
|
return unstaged_changes or staged_changes
|
|
|
|
|
|
@click.command()
|
|
def main() -> None:
|
|
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__)}`.
|
|
"""
|
|
)
|
|
subprocess.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)
|