Print off pull requests associated with a branch (Github only)

.gitconfig

[alias]
	pr = "!f() { BRANCH=$(git branch --show-current); REPOURL=$(git config --get remote.origin.url); python ~/Code/work_HK/getpr.py $REPOURL $BRANCH; }; f"

conf.sh

git config --global alias.pr '!f() { BRANCH=$(git branch --show-current); REPOURL=$(git config --get remote.origin.url); python ~/Code/work_HK/getpr.py $REPOURL $BRANCH; }; f'

getpr.py

import json
import urllib.request
import sys

# get repo and branch from sys.argv
if len(sys.argv) < 3:
    print("Usage: python getpr.py <repo> <branch>")
    sys.exit(1)

REPO = sys.argv[1].strip()
BRANCH = sys.argv[2].strip()

repo = REPO.replace(".git", "").replace("https://github.com/", "")
print("\nrepo:", repo, "\nbranch:", BRANCH, "\n")

found = False
for ix, ux in enumerate(
    [
        "\t" + dx["html_url"] + ' -- ' + dx["title"]
        for dx in json.loads(
            urllib.request.urlopen(
                urllib.request.Request(f"https://api.github.com/repos/{repo}/pulls")
            )
            .read()
            .decode("utf-8")
        )
        if dx["head"]["ref"] == BRANCH
    ]
):
    print("\t", f"{ix+1}.", ux, "\n")
    found = True

if not found:
    print("No pull request found for branch", BRANCH, "in", REPO)
    sys.exit(1)