fabfile.py

from fabric.api import *
from fabric.colors import green, red
import time

env.project_name = 'civomega'
env.hosts = ['dev.civomega.com']
env.path = '/home/ubuntu/civomega_deploy'
env.user = 'ubuntu'
env.virtualhost_path = "/"


def setup():
    """
    Setup a fresh virtualenv as well as a few useful directories, then run
    a full deployment
    """
    require('hosts')
    require('path')
    sudo('aptitude install -y nginx python-setuptools postgresql-client libpq-dev python-dev uwsgi uwsgi-plugin-python libffi-dev')
    sudo('easy_install -q pip')
    sudo('pip install -q virtualenv')

    # TODO: nginx config
    #sudo('cd /etc/apache2/sites-available/; a2dissite default;')

    run('mkdir -p %(path)s' % env)
    run('cd %(path)s; mkdir releases shared packages;' % env, warn_only=True)
    deploy()


def deploy():
    """
    Deploy the latest version of the site to the servers, install any
    required third party modules, install the virtual host and 
    then restart the webserver
    """
    require('hosts')
    require('path')
    env.release = time.strftime('%Y%m%d%H%M%S')

    upload_tar_from_git()
    bootstrap_venv()
    #install_site()
    symlink_current_release()
    migrate()
    restart_webserver()

def deploy_version(version):
    "Specify a specific version to be made live"
    require('hosts', provided_by=[local])
    require('path')
    env.version = version
    run('cd %(path)s; rm releases/previous; mv releases/current releases/previous;')
    run('cd %(path)s; ln -s $(version) releases/current')
    restart_webserver()

def rollback():
    """
    Limited rollback capability. Simple loads the previously current
    version of the code. Rolling back again will swap between the two.
    """
    require('hosts', provided_by=[local])
    require('path')
    run('cd %(path)s; mv releases/current releases/_previous;')
    run('cd %(path)s; mv releases/previous releases/current;')
    run('cd %(path)s; mv releases/_previous releases/previous;')
    restart_webserver()    


# Helpers. These are called by other functions rather than directly
def upload_tar_from_git():
    "Create an archive from the current Git develop branch and upload it"

    require('release', provided_by=[deploy, setup])
    local('git archive --format=tar develop | gzip > %(release)s.tar.gz' % env)
    run('mkdir %(path)s/releases/%(release)s' % env)
    put('%(release)s.tar.gz' % env, '%(path)s/packages/' % env)
    run('cd %(path)s/releases/%(release)s && tar zxf ../../packages/%(release)s.tar.gz' % env)
    put('civomega/settings_live.py', '%(path)s/releases/%(release)s/civomega/' % env)
    local('rm %(release)s.tar.gz' % env)

def bootstrap_venv():
    "Install the required packages from the requirements file using pip"
    require('release', provided_by=[deploy, setup])
    run('cd %(path)s/releases/%(release)s; virtualenv .; ./bin/pip install -UIq pip setuptools; ./bin/pip install -UI psycopg2 pyopenssl ndg-httpsclient pyasn1; ./bin/pip install -UI -r requirements.txt' % env)
def symlink_current_release():
    "Symlink our current release"
    require('release', provided_by=[deploy, setup])
    run('cd %(path)s; rm releases/previous; mv releases/current releases/previous;' % env, warn_only=True)
    run('cd %(path)s; ln -s %(release)s releases/current' % env)
def migrate():
    "Update the database"
    run('cd %(path)s/releases/current; ./bin/python manage.py syncdb -v 1 --noinput --settings=civomega.settings_live' % env)
    run('cd %(path)s/releases/current; ./bin/python manage.py migrate -v 1 codata --noinput --settings=civomega.settings_live' % env)
    run('cd %(path)s/releases/current; ./bin/python manage.py migrate -v 1 cologger --noinput --settings=civomega.settings_live' % env)
    run('cd %(path)s/releases/current; ./bin/python manage.py collectstatic -v1 --noinput --settings=civomega.settings_live' % env)
    run('cd %(path)s/releases/current; ./bin/python manage.py update_patterns --settings=civomega.settings_live' % env)

def restart_webserver():
    "Restart the web server"
    sudo('killall uwsgi', warn_only=True) # TODO
    sudo('cd %(path)s/releases/current; uwsgi --ini uwsgi.ini' % env)
    sudo('chmod 777 %(path)s/releases/current/civomega.sock' % env) # TODO
    sudo('service nginx restart' % env)