Skip to content

Instantly share code, notes, and snippets.

@tychoish
Last active August 29, 2015 14:07
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save tychoish/bd822864d70412fe58f1 to your computer and use it in GitHub Desktop.
Save tychoish/bd822864d70412fe58f1 to your computer and use it in GitHub Desktop.

Delegated Build

Introduction

This is the initial version of a script that clones the current git repository into build/ directory and runs make targets using this source as opposed the current working tree. More posts to come.

This is the initial implementation, and probably needs to some modification to work with your deployment. For more information and the latest version see:

The Code

 #!/usr/bin/python

 import os
 import subprocess
 import argparse
 import datetime

 try:
     from subprocess import DEVNULL # py3k
 except ImportError:
     DEVNULL = open(os.devnull, 'wb')

 branches = []
 builders = []

 def get_branch(path=None):
     if path is None:
         path = os.getcwd()

     return get_command_values(['git', 'symbolic-ref', 'HEAD'], path)[0].split('/')[2]

 def run_command(cmd, path, output=None):
     if output is None:
         output = DEVNULL

     p = subprocess.Popen(cmd, cwd=path, stdout=output, stderr=output)
     p.wait()

 def get_command_values(cmd, path):
     p = subprocess.Popen(cmd, cwd=path, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     r = p.communicate()
     return r[0].decode().split('\n')[:-1]

 def log_command_output(cmd, path, logfile, wait=False):
     with open(logfile, 'a') as f:
         p = subprocess.Popen(cmd, cwd=path, stdout=f, stderr=f)
         if wait is True:
             p.wait()

 def change_branch(branch, path):
     run_command(['git', 'checkout', branch], path)
     print('[sub-repo]: checked out branch: ' + branch + ' in staging.')

 def get_avalible_branches():
     current_branch = get_branch(os.getcwd())

     if current_branch not in branches:
         branches.append(current_branch)

     return branches

 def branch_cleanup(path):
     extant_branches = get_command_values(['ls', '.git/refs/heads'], path)

     for branch in branches:
         if branch not in extant_branches:
             cmd = ['git', 'branch', branch, 'origin/' + branch ]
             run_command(cmd, path)
             print('[sub-repo]: created ' + branch + ' branch in staging.')

     for branch in extant_branches:
         if branch not in branches:
             run_command(['git', 'branch', '-D', branch], path)
             print('[sub-repo]: cleaned up stale branch: ' + branch)

 def update_repo(path, branch='master'):
     if not os.path.isdir(path):
         cmd = ['/usr/bin/git', 'clone', './', 'build/.docs-staging']
         run_command(cmd, os.getcwd())
         print('[sub-repo] created staging repo.')
     else:
         cmd = ['git', 'pull']
         run_command(cmd, path)
         print('[sub-repo] updated staging.')

     if not os.path.join(path, 'build'):
         os.symlink(source='../../build', link_name='build')
         print('[sub-repo] created "build/" symlink in staging.')

     branch_cleanup(path)
     change_branch(branch, path)

 def build_branch(branch, path, logfile, target='publish', wait=False):
     if get_branch(path) != branch:
         change_branch(branch, path)

     cmd = ['make', target]

     print('[sub-repo]: building ' + branch + ' check log for details.')
     if wait is True:
         print('[sub-repo]: building now, waiting for the build to finish.')

     log_command_output(cmd, path, logfile, wait)
     print('[sub-repo]: build complete.')

 def ui():
     targets = builders
     targets.append('publish')
     targets.append('push')
     targets.append('stage')

     build_dir = os.path.join(os.getcwd(), 'build/.docs-staging/')
     default_log = 'build/docs-staging-' + str(datetime.date.today()) + '.log'

     parser = argparse.ArgumentParser('checks out and builds current repo in "build/" directory.')
     parser.add_argument('--branch', '-b', choices=branches, default=get_branch(), help='the name of the branch to build.')
     parser.add_argument('--target', '-t', choices=targets, default='publish', help='the make target to build.')
     parser.add_argument('--logfile', '-l', default=default_log, help='a path to a logfile for make output. defaults to "' + default_log + '" .')
     parser.add_argument('--wait', '-w', default=False, action='store_true', help='wait to return until build completes if specified.')

     return parser.parse_args()

 def main():
     build_dir = os.path.join(os.getcwd(), 'build/.docs-staging/')
     user = ui()

     update_repo(build_dir, branch=user.branch)
     build_branch(user.branch, build_dir, logfile=user.logfile, target=user.target, wait=user.wait)

 if __name__ == '__main__':
     main()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment