108 lines
3.6 KiB
Python
108 lines
3.6 KiB
Python
#
|
|
# 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 httplib
|
|
import httplib2
|
|
import logging
|
|
import re
|
|
import socket
|
|
|
|
import apiclient.errors
|
|
|
|
import gerrit
|
|
import gmail
|
|
import presubmit
|
|
|
|
|
|
def get_gerrit_info(body):
|
|
info = {}
|
|
gerrit_pattern = r'^Gerrit-(\S+): (.+)$'
|
|
for match in re.finditer(gerrit_pattern, body, flags=re.MULTILINE):
|
|
info[match.group(1)] = match.group(2).strip()
|
|
return info
|
|
|
|
|
|
def process_message(msg, dry_run):
|
|
try:
|
|
body = gmail.get_body(msg)
|
|
gerrit_info = get_gerrit_info(body)
|
|
if not gerrit_info:
|
|
logging.fatal('No Gerrit info found: %s', msg.subject)
|
|
msg_type = gerrit_info['MessageType']
|
|
handlers = {
|
|
'comment': presubmit.handle_comment,
|
|
'newchange': presubmit.handle_change,
|
|
'newpatchset': presubmit.handle_change,
|
|
|
|
'abandon': presubmit.skip_handler,
|
|
'merge-failed': presubmit.skip_handler,
|
|
'merged': presubmit.skip_handler,
|
|
'restore': presubmit.skip_handler,
|
|
'revert': presubmit.skip_handler,
|
|
}
|
|
|
|
message_type = gerrit_info['MessageType']
|
|
if message_type in handlers:
|
|
return handlers[message_type](gerrit_info, body, dry_run)
|
|
else:
|
|
logging.warning('MessageType %s unhandled.', msg_type)
|
|
return False
|
|
except NotImplementedError as ex:
|
|
logging.error("%s", ex)
|
|
return False
|
|
except gerrit.GerritError as ex:
|
|
change_id = gerrit_info['Change-Id']
|
|
logging.error('Gerrit error (%d): %s %s', ex.code, change_id, ex.url)
|
|
return ex.code == 404
|
|
|
|
|
|
def get_and_process_jobs():
|
|
dry_run = False
|
|
|
|
gmail_service = gmail.build_service()
|
|
msg_service = gmail_service.users().messages()
|
|
|
|
# We run in a loop because some of the exceptions thrown here mean we just
|
|
# need to retry. For errors where we should back off (typically any gmail
|
|
# API exceptions), process_changes catches the error and returns normally.
|
|
while True:
|
|
try:
|
|
process_changes(gmail_service, msg_service, dry_run)
|
|
return
|
|
except httplib.BadStatusLine:
|
|
pass
|
|
except httplib2.ServerNotFoundError:
|
|
pass
|
|
except socket.error:
|
|
pass
|
|
|
|
|
|
def process_changes(gmail_service, msg_service, dry_run):
|
|
try:
|
|
labels = gmail_service.users().labels().list(userId='me').execute()
|
|
if not labels['labels']:
|
|
logging.error('Could not retrieve Gmail labels')
|
|
return
|
|
label_id = gmail.get_gerrit_label(labels['labels'])
|
|
if not label_id:
|
|
logging.error('Could not find gerrit label')
|
|
return
|
|
|
|
for msg in gmail.get_all_messages(gmail_service, label_id):
|
|
msg = msg_service.get(userId='me', id=msg['id']).execute()
|
|
if process_message(msg, dry_run) and not dry_run:
|
|
msg_service.trash(userId='me', id=msg['id']).execute()
|
|
except apiclient.errors.HttpError as ex:
|
|
logging.error('API Client HTTP error: %s', ex)
|