raw.txt
No One
Actions

Authored By

	awight
	Aug 1 2019, 2:01 PM

Size

1 KB

Referenced Files

None

Subscribers

None

raw.txt
View Options

	#!/usr/bin/env python3
	"""
	Usage:
	swat-summary <gerrit-url>
	swat-summary -h \| --help
	"""
	import docopt
	import json
	import re
	import requests

	GERRIT_CHANGE_API = 'https://gerrit.wikimedia.org/r/changes/{gerrit_number}?o=CURRENT_REVISION&o=CURRENT_COMMIT'
	GERRIT_RE = 'https://gerrit.wikimedia.org/r/#/c/[-+_/a-zA-Z0-9]+/(?P<gerrit_number>[0-9]+)/'
	PHAB_HEADER_RE = '(?m)^Bug: (?P<phab_task>T[0-9]+)$'
	SUMMARY_MSG = 'SWAT: [[gerrit:{gerrit_number}\|{commit_message} ({phabricator_task})]]'


	def build_summary(gerrit_url):
	matches = re.match(GERRIT_RE, gerrit_url)
	if matches:
	gerrit_number = matches.group('gerrit_number')
	else:
	raise RuntimeError("Couldn't extract gerrit change number from string " + gerrit_url)

	change = get_gerrit(GERRIT_CHANGE_API, gerrit_number=gerrit_number)
	revision_id = change['current_revision']
	commit = change['revisions'][revision_id]['commit']
	bugs = re.findall(PHAB_HEADER_RE, commit['message'])

	return SUMMARY_MSG.format(
	gerrit_number=gerrit_number,
	commit_message=commit['subject'],
	phabricator_task=', '.join(bugs))


	def get_gerrit(api_format, **params):
	url = api_format.format(**params)
	r = requests.get(url)
	if r.status_code != 200:
	raise RuntimeError("Bad HTTP response from " + url)
	return json.loads(
	r.text.lstrip(")]}'"))


	if __name__ == '__main__':
	arguments = docopt.docopt(__doc__)
	print(build_summary(arguments['<gerrit-url>']))