oss-fuzz/infra/gcb/wait_for_build.py

72 lines
1.5 KiB
Python
Executable File

#!/usr/bin/python2
"""Waits for project build on Google Cloud Builder.
Usage: wait_for_build.py <build_id>
"""
import argparse
import sys
import time
import datetime
from googleapiclient.discovery import build
from oauth2client.client import GoogleCredentials
POLL_INTERVAL = 15
cloudbuild = None
def get_build(build_id, cloudbuild, project):
return cloudbuild.projects().builds().get(
projectId=project, id=build_id).execute()
def wait_for_build(build_id, project):
DONE_STATUSES = [
'SUCCESS',
'FAILURE',
'INTERNAL_ERROR',
'CANCELLED',
]
status = None
while True:
build_info = get_build(build_id, cloudbuild, project)
current_status = build_info['status']
if current_status != status:
print datetime.datetime.now(), current_status
sys.stdout.flush()
status = current_status
if status in DONE_STATUSES:
return status == 'SUCCESS'
time.sleep(POLL_INTERVAL)
def main():
global cloudbuild
parser = argparse.ArgumentParser(description='Wait for build to complete')
parser.add_argument(
'-p',
'--project',
help='Cloud Project',
default='clusterfuzz-external')
parser.add_argument('build_id', help='The Container Builder build ID.')
args = parser.parse_args()
credentials = GoogleCredentials.get_application_default()
cloudbuild = build('cloudbuild', 'v1', credentials=credentials)
success = wait_for_build(args.build_id, args.project)
if not success:
sys.exit(1)
if __name__ == '__main__':
main()