mirror of https://github.com/google/oss-fuzz.git
73 lines
1.5 KiB
Python
Executable File
73 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',
|
|
'TIMEOUT',
|
|
]
|
|
|
|
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='oss-fuzz')
|
|
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()
|