default-branch.py

#!/usr/bin/env python3
# Usage: $0 [gh-user] [gh-repo] [gh-token]

import os
import subprocess
import sys


def get_branch(rev):
    try:
        revparse = ["git", "rev-parse", "--abbrev-ref", rev]
        v = subprocess.check_output(revparse, stderr=open(os.devnull, "wb"))

        print(v.decode("utf-8").strip(" \r\n").split("/")[-1])
        exit(0)
    except subprocess.CalledProcessError:
        pass


remote = os.environ.get("GIT_REMOTE", default="origin")
get_branch(f"{remote}/HEAD")
get_branch(f"refs/remotes/{remote}/HEAD")

if len(sys.argv) < 3:
    print(f"warning: {sys.argv[0]} unable to determine default branch", file=sys.stderr)
    get_branch("HEAD")
    exit(1)

try:
    import requests
except ImportError:
    print(
        f"warning: {sys.argv[0]} need 'requests' to determine default branch",
        file=sys.stderr,
    )
    print(f"warning: 'pip3 install [--user] requests' to install", file=sys.stderr)
    get_branch("HEAD")
    exit(1)


url = f"https://api.github.com/repos/{sys.argv[1]}/{sys.argv[2]}"
headers = {}
if len(sys.argv) >= 4:
    headers["Authorization"] = f"bearer {sys.argv[3]}"
response = requests.get(url, headers=headers)
response.raise_for_status()
result = response.json()

print(result["default_branch"])