Home | History | Annotate | Download | only in bionicbb 
      1 #!/usr/bin/env python2
      2 #
      3 # Copyright (C) 2015 The Android Open Source Project
      4 #
      5 # Licensed under the Apache License, Version 2.0 (the 'License');
      6 # you may not use this file except in compliance with the License.
      7 # You may obtain a copy of the License at
      8 #
      9 #      http://www.apache.org/licenses/LICENSE-2.0
     10 #
     11 # Unless required by applicable law or agreed to in writing, software
     12 # distributed under the License is distributed on an 'AS IS' BASIS,
     13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     14 # See the License for the specific language governing permissions and
     15 # limitations under the License.
     16 #
     17 import json
     18 import logging
     19 import os
     20 
     21 from apscheduler.schedulers.background import BackgroundScheduler
     22 from flask import Flask, request
     23 import requests
     24 
     25 import gerrit
     26 import tasks
     27 
     28 app = Flask(__name__)
     29 
     30 
     31 def gerrit_url(endpoint):
     32     gerrit_base_url = 'https://android-review.googlesource.com'
     33     return gerrit_base_url + endpoint
     34 
     35 
     36 @app.route('/', methods=['POST'])
     37 def handle_build_message():
     38     result = json.loads(request.data)
     39 
     40     name = result['name']
     41     number = result['build']['number']
     42     status = result['build']['status']
     43     go_url = 'http://go/bionicbb/' + result['build']['url']
     44     full_url = result['build']['full_url']
     45     params = result['build']['parameters']
     46     change_id = params['CHANGE_ID']
     47     ref = params['REF']
     48     patch_set = ref.split('/')[-1]
     49 
     50     logging.debug('%s #%s %s: %s', name, number, status, full_url)
     51 
     52     # bionic-lint is always broken, so we don't want to reject changes for
     53     # those failures until we clean things up.
     54     if name == 'bionic-presubmit':
     55         message_lines = ['{} #{} checkbuild {}: {}'.format(
     56             name, number, status, go_url)]
     57         if status == 'FAILURE':
     58             message_lines += ['If you believe this Verified-1 was in error, '
     59                               '+1 the change and bionicbb will remove the -1 '
     60                               'shortly.']
     61 
     62         request_data = {
     63             'message': '\n'.join(message_lines)
     64         }
     65 
     66         label = 'Verified'
     67         if status == 'FAILURE':
     68             request_data['labels'] = {label: -1}
     69         elif status == 'SUCCESS':
     70             request_data['labels'] = {label: +1}
     71 
     72         url = gerrit_url('/a/changes/{}/revisions/{}/review'.format(change_id,
     73                                                                     patch_set))
     74 
     75         headers = {'Content-Type': 'application/json;charset=UTF-8'}
     76         logging.debug('POST %s: %s', url, request_data)
     77         requests.post(url, headers=headers, json=request_data)
     78     elif name == 'clean-bionic-presubmit':
     79         request_data = {'message': 'out/ directory removed'}
     80         url = gerrit_url('/a/changes/{}/revisions/{}/review'.format(change_id,
     81                                                                     patch_set))
     82         headers = {'Content-Type': 'application/json;charset=UTF-8'}
     83         logging.debug('POST %s: %s', url, request_data)
     84         requests.post(url, headers=headers, json=request_data)
     85     elif name == 'bionic-lint':
     86         logging.warning('Result for bionic-lint ignored')
     87     else:
     88         logging.error('Unknown project: %s', name)
     89     return ''
     90 
     91 
     92 @app.route('/drop-rejection', methods=['POST'])
     93 def drop_rejection():
     94     revision_info = json.loads(request.data)
     95 
     96     change_id = revision_info['changeid']
     97     patch_set = revision_info['patchset']
     98 
     99     bb_email = 'bionicbb (at] android.com'
    100     labels = gerrit.get_labels(change_id, patch_set)
    101     if bb_email in labels['Verified']:
    102         bb_review = labels['Verified'][bb_email]
    103     else:
    104         bb_review = 0
    105 
    106     if bb_review >= 0:
    107         logging.info('No rejection to drop: %s %s', change_id, patch_set)
    108         return ''
    109 
    110     logging.info('Dropping rejection: %s %s', change_id, patch_set)
    111 
    112     request_data = {'labels': {'Verified': 0}}
    113     url = gerrit_url('/a/changes/{}/revisions/{}/review'.format(change_id,
    114                                                                 patch_set))
    115     headers = {'Content-Type': 'application/json;charset=UTF-8'}
    116     logging.debug('POST %s: %s', url, request_data)
    117     requests.post(url, headers=headers, json=request_data)
    118     return ''
    119 
    120 
    121 if __name__ == "__main__":
    122     logging.basicConfig(level=logging.INFO)
    123     logger = logging.getLogger()
    124     fh = logging.FileHandler('bionicbb.log')
    125     fh.setLevel(logging.INFO)
    126     logger.addHandler(fh)
    127 
    128     # Prevent the job from being rescheduled by the reloader.
    129     if os.environ.get('WERKZEUG_RUN_MAIN') == 'true':
    130         scheduler = BackgroundScheduler()
    131         scheduler.start()
    132         scheduler.add_job(tasks.get_and_process_jobs, 'interval', minutes=5)
    133 
    134     app.run(host='0.0.0.0', debug=True)
    135