Scheduled Lambda that uses the Codeship API in order to trigger a build.

codeship-ami-builder.py

import boto3
import http.client
import json
import logging
import os
from base64 import b64decode


conn = http.client.HTTPSConnection("api.codeship.com")
payload = "{}"

logger = logging.getLogger()
logger.setLevel(logging.INFO)

auth_encrypted = os.environ['AUTH']
auth = boto3.client('kms').decrypt(CiphertextBlob=b64decode(auth_encrypted))['Plaintext']
auth = auth.decode("utf-8") 
print(auth)

# Set UUIDs of Codeship build to be restarted.
org_uuid = os.environ['ORG_UUID']
project_uuid = os.environ['PROJECT_UUID']
build_uuid = os.environ['BUILD_UUID']


def authenticate():
    payload = "{}"

    headers = {
        'content-type': "application/json",
        'authorization': "Basic " + auth
    }

    conn.request("POST", "/v2/auth", payload, headers)

    res = conn.getresponse()
    data = res.read()
    decoded_data = data.decode("utf-8")
    parsed_json = json.loads(decoded_data)

    return parsed_json['access_token']


def restart_build(org_uuid, project_uuid, build_uuid):
    auth_token = authenticate()
    headers = {
        'content-type': "application/json",
        'authorization': auth_token
    }

    conn.request("POST", "/v2/organizations/" + org_uuid + "/projects/" + project_uuid + "/builds/" + build_uuid + "/restart", payload, headers)

    res = conn.getresponse()
    data = res.read()

    logger.info(data.decode("utf-8"))


def lambda_handler(event, context):
    # Restart a build on Codeship
    restart_build(org_uuid, project_uuid, build_uuid)