Restart the last Codeship build for a project

codeship_restart_last_build.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# Usage:
#
# $ CODESHIP_API_KEY={your api key} CODESHIP_PROJECT_ID={project id} ./rebuild.py

import os
import json
import urllib2

API_KEY    = os.environ.get('CODESHIP_API_KEY')
PROJECT_ID = os.environ.get('CODESHIP_PROJECT_ID')
BRANCH     = os.environ.get('CODESHIP_BRANCH')

def fetch_builds(project_id, branch = 'master'):
    url      = 'https://codeship.com/api/v1/projects/{}.json?api_key={}&branch={}'.format(project_id, API_KEY, branch)
    request  = urllib2.Request(url)
    response = urllib2.urlopen(request)
    project  = json.load(response)

    return project['builds'] if 'builds' in project.keys() else []

def restart_build(build):
    url      = 'https://codeship.com/api/v1/builds/{}/restart.json?api_key={}'.format(build['id'], API_KEY)
    request  = urllib2.Request(url, {})
    response = urllib2.urlopen(request)
    build    = json.load(response)

    return build

if __name__ == '__main__':
    builds = fetch_builds(PROJECT_ID, BRANCH or 'master')

    if len(builds):
        result = restart_build(builds[0])
        print 'Rebuilding {}: "{}" on {}'.format(result['id'], result['message'].split('\n')[0], result['branch'])