81 lines
2.0 KiB
Python
81 lines
2.0 KiB
Python
import typer
|
|
import requests
|
|
from typing import Annotated
|
|
from rich.console import Console
|
|
|
|
app = typer.Typer()
|
|
|
|
DEFAULT_HEADERS = {
|
|
"Accept": "application/vnd.github.text+json",
|
|
}
|
|
|
|
BRANCHES = ["nixos-unstable-small", "nixos-unstable", "nixos-24.05"]
|
|
|
|
|
|
def commit_in_branch(commit_sha: str, branch: str) -> bool:
|
|
url = f"https://api.github.com/repos/NixOS/nixpkgs/compare/{branch}...{commit_sha}"
|
|
commit_response = requests.get(url, headers=DEFAULT_HEADERS)
|
|
commit_response.raise_for_status()
|
|
status = commit_response.json().get("status")
|
|
|
|
if status in ["identical", "behind"]:
|
|
return True
|
|
return False
|
|
|
|
|
|
@app.command()
|
|
def pr(
|
|
pr: str,
|
|
branch: Annotated[str | None, typer.Option(help="Check specific branch")] = None,
|
|
):
|
|
"""Get status of pull request"""
|
|
url = f"https://api.github.com/repos/NixOS/nixpkgs/pulls/{pr}"
|
|
typer.echo(f"branch {branch}")
|
|
|
|
pr_response = requests.get(url, headers=DEFAULT_HEADERS)
|
|
pr_response.raise_for_status()
|
|
|
|
pr_data = pr_response.json()
|
|
|
|
console = Console()
|
|
|
|
merged = pr_data["merged"]
|
|
|
|
if branch and not merged:
|
|
console.print(f":x: {branch}")
|
|
return
|
|
|
|
commit_sha = pr_data.get("merge_commit_sha")
|
|
|
|
# Check only specified branch
|
|
if branch:
|
|
in_branch = commit_in_branch(commit_sha, branch)
|
|
if in_branch:
|
|
console.print(f":white_check_mark: {branch}", highlight=False)
|
|
else:
|
|
console.print(f":x: {branch}", highlight=False)
|
|
return
|
|
|
|
if not merged:
|
|
console.print(":x: master", highlight=False)
|
|
for branch in BRANCHES:
|
|
console.print(f":x: {branch}", highlight=False)
|
|
return
|
|
|
|
console.print(":white_check_mark: master", highlight=False)
|
|
|
|
for b in BRANCHES:
|
|
in_branch = commit_in_branch(commit_sha, b)
|
|
if in_branch:
|
|
console.print(f":white_check_mark: {b}", highlight=False)
|
|
else:
|
|
console.print(f":x: {b}", highlight=False)
|
|
|
|
|
|
def main():
|
|
app()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|