229 lines
8.9 KiB
Python
229 lines
8.9 KiB
Python
#!/usr/bin/env python
|
|
#
|
|
# Copyright 2016 - 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.
|
|
|
|
"""A client that talks to Android Build APIs."""
|
|
|
|
import collections
|
|
import io
|
|
import logging
|
|
|
|
import apiclient
|
|
|
|
from acloud import errors
|
|
from acloud.internal.lib import base_cloud_client
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# The BuildInfo namedtuple data structure.
|
|
# It will be the data structure returned by GetBuildInfo method.
|
|
BuildInfo = collections.namedtuple("BuildInfo", [
|
|
"branch", # The branch name string
|
|
"build_id", # The build id string
|
|
"build_target", # The build target string
|
|
"release_build_id"]) # The release build id string
|
|
|
|
|
|
class AndroidBuildClient(base_cloud_client.BaseCloudApiClient):
|
|
"""Client that manages Android Build."""
|
|
|
|
# API settings, used by BaseCloudApiClient.
|
|
API_NAME = "androidbuildinternal"
|
|
API_VERSION = "v2beta1"
|
|
SCOPE = "https://www.googleapis.com/auth/androidbuild.internal"
|
|
|
|
# other variables.
|
|
DEFAULT_RESOURCE_ID = "0"
|
|
# TODO(b/27269552): We should use "latest".
|
|
DEFAULT_ATTEMPT_ID = "0"
|
|
DEFAULT_CHUNK_SIZE = 20 * 1024 * 1024
|
|
NO_ACCESS_ERROR_PATTERN = "does not have storage.objects.create access"
|
|
# LKGB variables.
|
|
BUILD_STATUS_COMPLETE = "complete"
|
|
BUILD_TYPE_SUBMITTED = "submitted"
|
|
ONE_RESULT = 1
|
|
BUILD_SUCCESSFUL = True
|
|
LATEST = "latest"
|
|
|
|
# Message constant
|
|
COPY_TO_MSG = ("build artifact (target: %s, build_id: %s, "
|
|
"artifact: %s, attempt_id: %s) to "
|
|
"google storage (bucket: %s, path: %s)")
|
|
# pylint: disable=invalid-name
|
|
def DownloadArtifact(self,
|
|
build_target,
|
|
build_id,
|
|
resource_id,
|
|
local_dest,
|
|
attempt_id=None):
|
|
"""Get Android build attempt information.
|
|
|
|
Args:
|
|
build_target: Target name, e.g. "aosp_cf_x86_phone-userdebug"
|
|
build_id: Build id, a string, e.g. "2263051", "P2804227"
|
|
resource_id: Id of the resource, e.g "avd-system.tar.gz".
|
|
local_dest: A local path where the artifact should be stored.
|
|
e.g. "/tmp/avd-system.tar.gz"
|
|
attempt_id: String, attempt id, will default to DEFAULT_ATTEMPT_ID.
|
|
"""
|
|
attempt_id = attempt_id or self.DEFAULT_ATTEMPT_ID
|
|
api = self.service.buildartifact().get_media(
|
|
buildId=build_id,
|
|
target=build_target,
|
|
attemptId=attempt_id,
|
|
resourceId=resource_id)
|
|
logger.info("Downloading artifact: target: %s, build_id: %s, "
|
|
"resource_id: %s, dest: %s", build_target, build_id,
|
|
resource_id, local_dest)
|
|
try:
|
|
with io.FileIO(local_dest, mode="wb") as fh:
|
|
downloader = apiclient.http.MediaIoBaseDownload(
|
|
fh, api, chunksize=self.DEFAULT_CHUNK_SIZE)
|
|
done = False
|
|
while not done:
|
|
_, done = downloader.next_chunk()
|
|
logger.info("Downloaded artifact: %s", local_dest)
|
|
except (OSError, apiclient.errors.HttpError) as e:
|
|
logger.error("Downloading artifact failed: %s", str(e))
|
|
raise errors.DriverError(str(e))
|
|
|
|
def CopyTo(self,
|
|
build_target,
|
|
build_id,
|
|
artifact_name,
|
|
destination_bucket,
|
|
destination_path,
|
|
attempt_id=None):
|
|
"""Copy an Android Build artifact to a storage bucket.
|
|
|
|
Args:
|
|
build_target: Target name, e.g. "aosp_cf_x86_phone-userdebug"
|
|
build_id: Build id, a string, e.g. "2263051", "P2804227"
|
|
artifact_name: Name of the artifact, e.g "avd-system.tar.gz".
|
|
destination_bucket: String, a google storage bucket name.
|
|
destination_path: String, "path/inside/bucket"
|
|
attempt_id: String, attempt id, will default to DEFAULT_ATTEMPT_ID.
|
|
"""
|
|
attempt_id = attempt_id or self.DEFAULT_ATTEMPT_ID
|
|
copy_msg = "Copying %s" % self.COPY_TO_MSG
|
|
logger.info(copy_msg, build_target, build_id, artifact_name,
|
|
attempt_id, destination_bucket, destination_path)
|
|
api = self.service.buildartifact().copyTo(
|
|
buildId=build_id,
|
|
target=build_target,
|
|
attemptId=attempt_id,
|
|
artifactName=artifact_name,
|
|
destinationBucket=destination_bucket,
|
|
destinationPath=destination_path)
|
|
try:
|
|
self.Execute(api)
|
|
finish_msg = "Finished copying %s" % self.COPY_TO_MSG
|
|
logger.info(finish_msg, build_target, build_id, artifact_name,
|
|
attempt_id, destination_bucket, destination_path)
|
|
except errors.HttpError as e:
|
|
if e.code == 503:
|
|
if self.NO_ACCESS_ERROR_PATTERN in str(e):
|
|
error_msg = "Please grant android build team's service account "
|
|
error_msg += "write access to bucket %s. Original error: %s"
|
|
error_msg %= (destination_bucket, str(e))
|
|
raise errors.HttpError(e.code, message=error_msg)
|
|
raise
|
|
|
|
def GetBranch(self, build_target, build_id):
|
|
"""Derives branch name.
|
|
|
|
Args:
|
|
build_target: Target name, e.g. "aosp_cf_x86_phone-userdebug"
|
|
build_id: Build ID, a string, e.g. "2263051", "P2804227"
|
|
|
|
Returns:
|
|
A string, the name of the branch
|
|
"""
|
|
api = self.service.build().get(buildId=build_id, target=build_target)
|
|
build = self.Execute(api)
|
|
return build.get("branch", "")
|
|
|
|
def GetLKGB(self, build_target, build_branch):
|
|
"""Get latest successful build id.
|
|
|
|
From branch and target, we can use api to query latest successful build id.
|
|
e.g. {u'nextPageToken':..., u'builds': [{u'completionTimestamp':u'1534157869286',
|
|
... u'buildId': u'4949805', u'machineName'...}]}
|
|
|
|
Args:
|
|
build_target: String, target name, e.g. "aosp_cf_x86_phone-userdebug"
|
|
build_branch: String, git branch name, e.g. "aosp-master"
|
|
|
|
Returns:
|
|
A string, string of build id number.
|
|
|
|
Raises:
|
|
errors.CreateError: Can't get build id.
|
|
"""
|
|
api = self.service.build().list(
|
|
branch=build_branch,
|
|
target=build_target,
|
|
buildAttemptStatus=self.BUILD_STATUS_COMPLETE,
|
|
buildType=self.BUILD_TYPE_SUBMITTED,
|
|
maxResults=self.ONE_RESULT,
|
|
successful=self.BUILD_SUCCESSFUL)
|
|
build = self.Execute(api)
|
|
logger.info("GetLKGB build API response: %s", build)
|
|
if build:
|
|
return str(build.get("builds")[0].get("buildId"))
|
|
raise errors.GetBuildIDError(
|
|
"No available good builds for branch: %s target: %s"
|
|
% (build_branch, build_target)
|
|
)
|
|
|
|
def GetBuildInfo(self, build_target, build_id, branch):
|
|
"""Get build info namedtuple.
|
|
|
|
Args:
|
|
build_target: Target name.
|
|
build_id: Build id, a string or None, e.g. "2263051", "P2804227"
|
|
If None or latest, the last green build id will be
|
|
returned.
|
|
branch: Branch name, a string or None, e.g. git_master. If None, the
|
|
returned branch will be searched by given build_id.
|
|
|
|
Returns:
|
|
A build info namedtuple with keys build_target, build_id, branch and
|
|
gcs_bucket_build_id
|
|
"""
|
|
if build_id and build_id != self.LATEST:
|
|
# Get build from build_id and build_target
|
|
api = self.service.build().get(buildId=build_id,
|
|
target=build_target)
|
|
build = self.Execute(api) or {}
|
|
elif branch:
|
|
# Get last green build in the branch
|
|
api = self.service.build().list(
|
|
branch=branch,
|
|
target=build_target,
|
|
successful=True,
|
|
maxResults=1,
|
|
buildType="submitted")
|
|
builds = self.Execute(api).get("builds", [])
|
|
build = builds[0] if builds else {}
|
|
else:
|
|
build = {}
|
|
|
|
build_id = build.get("buildId")
|
|
build_target = build_target if build_id else None
|
|
return BuildInfo(build.get("branch"), build_id, build_target,
|
|
build.get("releaseCandidateName"))
|