gh-email-comments.py

#!/usr/bin/python
# encoding=utf8

#
# Cron script to validate patches in open pull requests.
# 1. Validation skipped if "checkpatch" label is set.
# 2. Run both checkpatch.pl and odp agreement checks.
# Result is status on odp pull request github web  page and
# label "checkpatch" set.

from github import Github
import github
import os
import re
import glob
import sys
import pickle
import imaplib
import email
import smtplib
from email.mime.text import MIMEText
reload(sys)  
sys.setdefaultencoding('utf8')

configfile = '~/gscripts_config.py'
sys.path.append(os.path.dirname(os.path.expanduser(configfile)))
import gscripts_config as gcfg


gh_login = gcfg.gcfg['gh']['login']
gh_password = gcfg.gcfg['gh']['pass']

g = Github(gh_login, password=gh_password)

for repo in g.get_user().get_repos():
	if repo.full_name == "Linaro/odp":
		break
if not repo:
	exit(1)

# get Message-Id to reply with correct In-Reply-To
def get_email_id_by_title(title):
	ret_msg_id = None

	imap_login = gcfg.gcfg['imap']['login']
	imap_password = gcfg.gcfg['imap']['pass']

	mail = imaplib.IMAP4_SSL('imap.yandex.ru')
	mail.login(imap_login, imap_password)
	mail.list()

	# Out: list of "folders" aka labels in gmail.
	mail.select("inbox") # connect to inbox.
	mail.select(readonly=1) # Select inbox or default namespace

	# imap truncates subject search
	m = re.search(r'\[(PATCH.*v[0-9]+)\](.*)', title[0:52])
	if not m:
		mail.logout()
		return None

	prefix = m.group(1).lstrip() + " 0/" #look for PATCH v1 0/X
	subject = m.group(2).lstrip()

	(retcode, messages) = mail.uid('search', '(SUBJECT "%s")' % subject)
	for num in messages[0].split():
		typ, data = mail.fetch(num,'(RFC822)')
		msg_str = email.message_from_string(data[0][1])
		m = re.search(prefix,  msg_str['Subject'])
		if not m:
			continue
		ret_msg_id = msg_str['Message-Id']
		break
	mail.logout()
	return ret_msg_id

def smtp_send_comment(pull, rvc, id):
	if rvc.user.name:
		user = "%s(%s)" % (rvc.user.name, rvc.user.login)
	else:
		user = "%s" % (rvc.user.login)

	text = "%s replied on github web page:\n\n" % (user)

	text += "%s\n" % (rvc.path)
	if rvc.position:
		text += "line %s\n" % (rvc.position)
	text += "%s\n" % rvc.diff_hunk
	text += "\n\n"
	text += "Comment:\n"
	text += rvc.body
	text += "\n\n"

	text += "%s\n" % rvc.html_url
	text += "updated_at %s\n" % rvc.updated_at
	msg = MIMEText(text.encode("utf-8"))

	msg['Subject'] = "Re: %s" % pull.title
	msg['From'] =  "Github ODP bot  <odpbot@yandex.ru>"
	msg['To'] = "lng-odp@lists.linaro.org"
	msg['In-Reply-To'] = id
	#msg['To'] = "maxim.uvarov@linaro.org"

	smtp_server = gcfg.gcfg['smtp']['server']
	smtp_login = gcfg.gcfg['smtp']['login']
	smtp_password = gcfg.gcfg['smtp']['pass']

	s = smtplib.SMTP_SSL(smtp_server)
	s.login(smtp_login, smtp_password)
	s.sendmail(msg['From'], msg['To'], msg.as_string())
	s.quit()

def email_new_comment(pull, rvc):
	id = get_email_id_by_title(pull.title)
	smtp_send_comment(pull, rvc, id)

### MAIN

os.system("mkdir -p gh-ec-cache")


for pull in repo.get_pulls():
	try:
		issue = repo.get_issue(pull.number)
	except:
		# timeout ssl http error can happen. Process this pr
		# on next run.
		continue

	pull_cache = {}
	filename = "gh-ec-cache/%d.cache" % pull.number
	try:
		f = open(filename, "rb" );
		pull_cache = pickle.load(f)
		f.close()
	except:
		print "return at 1"
		break

	try:
		for rvc in pull.get_review_comments():
			if rvc.id not in pull_cache:
				email_new_comment(pull, rvc)
				pull_cache[rvc.id] = { rvc.body }
	except:
		print "return at 2"
		break

	f = open(filename, 'w')
	pickle.dump(pull_cache, f)
	f.close()