fabfile.py

#!/usr/bin/env python

from fabric.api import env,hide,hosts
from fabric.colors import red,green #,yellow,cyan
from fabric.context_managers import cd,settings #,lcd
from fabric.decorators import task
from fabric.operations import run,sudo #local,put
from fabric.utils import puts,abort #,warn
# from fabric.contrib.files import sed

# import os
import time
# from shutil import copyfile

# Configure user, private key, etc. for SFTP deployment
env.user = 'ubuntu'
#env.hosts = [] if not env.hosts else env.hosts # No development servers available!
env.colors = True
env.release = time.strftime('%Y%m%d%H%M%S')
env.basepath = '/opt/reportgrid/visualization' if not 'basepath' in env else env.basepath
env.repopath = '%(basepath)s/visualizations' % env if not 'repopath' in env else env.repopath
env.repository = 'git@github.com:precog/visualizations.git' if not 'repository' in env else env.repository
env.sitepath = '%(repopath)s/services' % env if not 'sitepath' in env else env.sitepath
env.linkpath = '%(basepath)s/services' % env if not 'linkpath' in env else env.linkpath
env.branch = 'origin/master' if not 'branch' in env else env.branch

@task
@hosts('localhost')
def production():
    """
        Run the tasks that follow it on production
    """
    env.hosts = ["appserver07.reportgrid.com", "appserver08.reportgrid.com", "appserver11.reportgrid.com", "appserver12.reportgrid.com", ]
    env.branch = 'origin/deploy'

@task
def deploy():
    """
        Deploy visualizations from github to server
    """
    with settings(hide('everything'), warn_only=True):
        if run("test -d %(repopath)s" % env).failed:
            abort(red("Could not find repository path %(repopath)s: run setup task first" % env))

    puts(green("Deploying..."))
    with cd(env.repopath):
        run("git fetch")
        run("git reset --hard %(branch)s" % env)
        sudo("chmod -R g+w %(sitepath)s" % env)

@task
def rollback():
    """
        Rollback to previously deployed version
    """
    puts(green("Rolling back..."))
    with cd(env.repopath):
        with settings(hide('everything')):
            version=run("git log -1 '--pretty=format:%h%d %ar %an %s' HEAD@{1}")
        puts(green("Previous version:\n%s" % version))
        run("git reset --hard HEAD@{1}")
        run("git reflog delete HEAD@{0}")
        run("git reflog delete HEAD@{0}")

@task
def setup():
    """
        Setup environment for deployment -- does not configure web server
    """
    puts(green("Setting up deploy environment"))

    # Handles different apache group on qclus-demo01
    if env.host_string == 'qclus-demo01.reportgrid.com':
        env.group = 'apache'
    else:
        env.group = 'www-data'

    sudo("mkdir -p %(basepath)s" % env)
    sudo("chown reportgrid:reportgrid %(basepath)s" % env)
    sudo("rm -fr %(repopath)s %(linkpath)s" % env)
    sudo("mkdir -p %(repopath)s" % env)
    sudo("chown ubuntu:%(group)s %(repopath)s" % env)
    sudo("chmod 6775 %(repopath)s" % env)
    with settings(warn_only=True):
        if run("git clone %(repository)s %(repopath)s" % env).failed:
            abort(red("Could not clone repository: does the user have read permission on it?"))
    with cd(env.repopath):
        run("git config core.sharedRepository true")
    sudo("chmod -R g+w %(sitepath)s" % env)
    sudo("ln -s %(sitepath)s %(linkpath)s" % env)