fix-backport-labels.py

#!/usr/bin/env python
import re
from github import Github
from os.path import expanduser
from optparse import OptionParser
from socket import setdefaulttimeout
from github_utils import api_rate_limits
from process_pr import get_backported_pr
from cms_static import ISSUE_SEEN_MSG, CMSBUILD_GH_USER
setdefaulttimeout(120)

parser = OptionParser(usage="%prog <pull-request-id>")
parser.add_option("-n", "--dry-run",    dest="dryRun",     action="store_true", help="Do not modify Github", default=False)
parser.add_option("-r", "--repository", dest="repository", help="Github Repositoy name e.g. cms-sw/cmssw.", type=str, default="cms-sw/cmssw")
opts, args = parser.parse_args()

if len(args) != 0: parser.error("Too many/few arguments")
  
gh = Github(login_or_token=open(expanduser("~/.github-token")).read().strip())
repo = gh.get_repo(opts.repository)
label = [ repo.get_label("backport") ]
issues = repo.get_issues(state="open", sort="updated", labels=label)
  
for issue in issues:
  if not issue.pull_request: continue
  api_rate_limits(gh)
  backport_pr=None
  issue_body = issue.body.encode("ascii", "ignore")
  if (issue.user.login == CMSBUILD_GH_USER) and re.match(ISSUE_SEEN_MSG,issue_body.split("\n",1)[0].strip()):
     backport_pr=get_backported_pr(issue_body)
  else:
    for comment in issue.get_comments():
      commenter = comment.user.login
      comment_msg = comment.body.encode("ascii", "ignore")
      # The first line is an invariant.
      comment_lines = [ l.strip() for l in comment_msg.split("\n") if l.strip() ]
      first_line = comment_lines[0:1]
      if not first_line: continue
      first_line = first_line[0]
      if (commenter == CMSBUILD_GH_USER) and re.match(ISSUE_SEEN_MSG, first_line):
        backport_pr=get_backported_pr(comment_msg)
        break
  if backport_pr and re.match("^[1-9][0-9]+$",backport_pr):
    print issue.number, backport_pr
    try:
      pr   = repo.get_pull(int(backport_pr))
      print "  Backported PR merged:",pr.merged
      if pr.merged:
        labels = list(set([x.name for x in issue.labels if x.name!="backport"]+["backport-ok"]))
        if not opts.dryRun: issue.edit(labels=labels)
        print issue.number,"New Labels:",labels
    except Exception, e:
      print e