ofCodeStyleGuard.py

#!/usr/bin/python
"""Make sure openFrameworks Pull Requests conform to the code style"""
import styleguard
import logging
import json
import os
from flask import Flask, request

WEBLOGGER = logging.getLogger('styleguard.webserver')
WEBLOGGER.setLevel(styleguard.cfg['logging_level'])
APP = Flask(__name__)
APP.logger.setLevel(styleguard.cfg['logging_level'])
logging.getLogger('urllib3').setLevel(logging.WARNING)
logging.getLogger('requests.packages.urllib3').setLevel(logging.INFO)
# Add a file handler to the root logger if log filename is set
styleguard.add_file_logger()


@APP.route('/check')
def manual_check():
	"""Initiate manual request for checking a PR"""
	WEBLOGGER.info('Manual PR check has been requested')
	WEBLOGGER.debug('Access route: ' + str(request.access_route[:]))
	# get the number of the requested pr (i.e. URL/check?pr=number)
	try:
		pr_number = int(request.args.get('pr', 0))
	except ValueError:
		WEBLOGGER.error('Invalid PR ID! Skipping...')
		return 'Error: Invalid PR ID!'
	if pr_number:
		WEBLOGGER.info('PR number ' + str(pr_number))
		styleguard.handle_payload(pr_number)
		return ('Received request for checking PR ' + str(pr_number))
	else:
		WEBLOGGER.error('Invalid PR ID! Skipping...')
		return 'Error: Invalid PR ID!'


@APP.route('/', methods=['POST'])
def api_pr():
	""" React to a received POST request"""
	WEBLOGGER.info(60 * "#")
	WEBLOGGER.info("Received POST request.")
	WEBLOGGER.debug('Access route: ' + str(request.access_route[:]))
	origin = request.access_route[0]
	# was using request.remote_addr. access_route could possibly be spoofed
	if origin not in styleguard.cfg['github_ips']:
		WEBLOGGER.warning("Origin of request UNKNOWN: " + origin)
		return 'Error'
	else:
		WEBLOGGER.debug("Origin of request: " + origin)

	try:
		payload = json.loads(request.form['payload'])['pull_request']
	except KeyError:
		# crutch: if an invalid request arrives locally, load a json file directly
		if origin == '127.0.0.1':
			location = os.getenv('OPENSHIFT_REPO_DIR', '')
			with open(os.path.join(location, 'sample_payload.json'), 'r') as sample:
				payload = json.load(sample)
		else:
			raise
	styleguard.handle_payload(payload)
	return 'OK'


def main():
	"""Main function"""
	# Instantiate a PrHandler, which start waiting on styleguard.MY_QUEUE
	WEBLOGGER.debug('In ofCodeStyleGuard main function')
	_threaded_pr_worker = styleguard.PrHandler()
	APP.run(host='0.0.0.0', port=styleguard.cfg['local_port'])
	styleguard.MY_QUEUE.join()

if __name__ == "__main__":
	main()