Merge the two bionicbb services into one.
Change-Id: I6490da1ec96b2e24b330296950be84424e11bd35
diff --git a/tools/bionicbb/bionicbb.py b/tools/bionicbb/bionicbb.py
new file mode 100644
index 0000000..20d6460
--- /dev/null
+++ b/tools/bionicbb/bionicbb.py
@@ -0,0 +1,130 @@
+#!/usr/bin/env python2
+#
+# Copyright (C) 2015 The Android Open Source Project
+#
+# Licensed under the Apache License, Version 2.0 (the 'License');
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an 'AS IS' BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+import json
+import logging
+import os
+
+from apscheduler.schedulers.background import BackgroundScheduler
+from flask import Flask, request
+import requests
+
+import gerrit
+import tasks
+
+app = Flask(__name__)
+
+
+def gerrit_url(endpoint):
+ gerrit_base_url = 'https://android-review.googlesource.com'
+ return gerrit_base_url + endpoint
+
+
+@app.route('/', methods=['POST'])
+def handle_build_message():
+ result = json.loads(request.data)
+
+ name = result['name']
+ number = result['build']['number']
+ status = result['build']['status']
+ go_url = 'http://go/bionicbb/' + result['build']['url']
+ full_url = result['build']['full_url']
+ params = result['build']['parameters']
+ change_id = params['CHANGE_ID']
+ ref = params['REF']
+ patch_set = ref.split('/')[-1]
+
+ logging.debug('%s #%s %s: %s', name, number, status, full_url)
+
+ # bionic-lint is always broken, so we don't want to reject changes for
+ # those failures until we clean things up.
+ if name == 'bionic-presubmit':
+ message_lines = ['{} #{} checkbuild {}: {}'.format(
+ name, number, status, go_url)]
+ if status == 'FAILURE':
+ message_lines += ['If you believe this Verified-1 was in error, '
+ '+1 the change and bionicbb will remove the -1 '
+ 'shortly.']
+
+ request_data = {
+ 'message': '\n'.join(message_lines)
+ }
+
+ label = 'Verified'
+ if status == 'FAILURE':
+ request_data['labels'] = {label: -1}
+ elif status == 'SUCCESS':
+ request_data['labels'] = {label: +1}
+
+ url = gerrit_url('/a/changes/{}/revisions/{}/review'.format(change_id,
+ patch_set))
+
+ headers = {'Content-Type': 'application/json;charset=UTF-8'}
+ logging.debug('POST %s: %s', url, request_data)
+ requests.post(url, headers=headers, json=request_data)
+ elif name == 'clean-bionic-presubmit':
+ request_data = {'message': 'out/ directory removed'}
+ url = gerrit_url('/a/changes/{}/revisions/{}/review'.format(change_id,
+ patch_set))
+ headers = {'Content-Type': 'application/json;charset=UTF-8'}
+ logging.debug('POST %s: %s', url, request_data)
+ requests.post(url, headers=headers, json=request_data)
+ elif name == 'bionic-lint':
+ logging.warning('Result for bionic-lint ignored')
+ else:
+ logging.error('Unknown project: %s', name)
+ return ''
+
+
+@app.route('/drop-rejection', methods=['POST'])
+def drop_rejection():
+ revision_info = json.loads(request.data)
+
+ change_id = revision_info['changeid']
+ patch_set = revision_info['patchset']
+
+ bb_email = 'bionicbb@android.com'
+ labels = gerrit.get_labels(change_id, patch_set)
+ if bb_email in labels['Verified']:
+ bb_review = labels['Verified'][bb_email]
+ else:
+ bb_review = 0
+
+ if bb_review >= 0:
+ logging.info('No rejection to drop: %s %s', change_id, patch_set)
+ return ''
+
+ logging.info('Dropping rejection: %s %s', change_id, patch_set)
+
+ request_data = {'labels': {'Verified': 0}}
+ url = gerrit_url('/a/changes/{}/revisions/{}/review'.format(change_id,
+ patch_set))
+ headers = {'Content-Type': 'application/json;charset=UTF-8'}
+ logging.debug('POST %s: %s', url, request_data)
+ requests.post(url, headers=headers, json=request_data)
+ return ''
+
+
+if __name__ == "__main__":
+ logging.basicConfig(level=logging.INFO)
+
+ # Prevent the job from being rescheduled by the reloader.
+ if os.environ.get('WERKZEUG_RUN_MAIN') == 'true':
+ scheduler = BackgroundScheduler()
+ scheduler.start()
+ scheduler.add_job(tasks.get_and_process_jobs, 'interval', minutes=5)
+
+ app.run(host='0.0.0.0', debug=True)