Merge GitHub PRs when the web UI is being dumb

merge-pr.py

#!/usr/bin/env python

import argparse
import re
import os

from github3 import GitHub

parser = argparse.ArgumentParser()
parser.add_argument('prs', nargs='+')
parser.add_argument('--token', default=os.getenv('GITHUB_TOKEN'))
parser.add_argument('--commit-message', '-m', default=None)
parser.add_argument('--commit-title', '-t', default=None)
args = parser.parse_args()


PR_RE = re.compile(r'https://github.com/([^/]+)/([^/]+)/pull/(\d+)')

def parse_url(url):
    match = PR_RE.match(url)
    if not match:
        raise SystemExit('Invalid URL: %s' % url)
    return match.groups()

assert args.token
g = GitHub(token=args.token)

for pr_url in args.prs:
    owner, _repo, number = parse_url(pr_url)
    repo = g.repository(owner, _repo)
    pr = repo.pull_request(int(number))
    print('%r %r' % (repo, pr))
    pr.merge(merge_method='squash', commit_message=args.commit_message, commit_title=args.commit_title)