cloudbuild_helper.py

# Copyright 2017, Google Inc. All rights reserved.
#
# 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.

"""Helpers for accessing Google Cloud Container Builder in Python code.

`build`: Builds a Docker image with the content and Dockerfile in
`source_dir`, and pushes the image to Google Cloud Container Registry.

For more information:
https://cloud.google.com/container-builder/
https://cloud.google.com/container-registry/
"""

from googleapiclient import discovery
from oauth2client.client import GoogleCredentials

from gcs_helper import archive_and_upload

credentials = GoogleCredentials.get_application_default()


def _make_body(source_bucket_name, source_object_name, image_name):
    body = {
        'source': {
            'storageSource': {
                'bucket': source_bucket_name,
                'object': source_object_name
            }
        },
        'steps': [
            {
                'name': 'gcr.io/cloud-builders/docker',
                'args': ['build', '-t', 'gcr.io/$PROJECT_ID/{}'.format(image_name), '.']
            }
        ],
        'images': [
            'gcr.io/$PROJECT_ID/{}'.format(image_name)
        ]
    }
    return body


def build(project_id, source_dir, bucket_name, image_name):
    """This uses the provided bucket (must be writable by the project)
    to store the intermediate source archive, and then builds an image using
    the Dockerfile in source_dir.

    If the build is successful the image will be pushed to container registry.
    """
    archive_and_upload(bucket_name, source_dir)

    body = _make_body(bucket_name, '{}.zip'.format(source_dir), image_name)

    service = discovery.build('cloudbuild', 'v1', credentials=credentials)
    build = service.projects().builds().create(projectId=project_id, body=body).execute()

    return build