fabfile.py

from __future__ import with_statement
from fabric.api import *
from fabric.contrib.files import exists
import sys
import time


env.project_name = 'bugle_project'
env.roledefs = {
    'localhost': ['127.0.0.1',],
    'live': ['root@bugle.fort',],
}
 
######################################
# Environment
######################################
 
def localhost():
    """
    Set up a development environment locally.
    """
    env.hosts = env.roledefs['localhost']
    env.path = '/'.join(sys.modules[__name__].__file__.split('/')[:-1])
    env.version_path = env.path

def live():
    env.hosts = env.roledefs['live']
    env.path = '/home/bugle'


######################################
# Tasks
######################################

def management_command(c):
    require('path', 'version_path')
    with cd(env.version_path):
        with cd(env.project_name):
            run('./manage.py %s --settings=configs.live.settings' % c)
    
def test():
    "Run the test suite and bail out if it fails"
    management_command('test --noinput')

def syncdb_migrate():
    management_command('syncdb --noinput')
    management_command('migrate --noinput')

@runs_once
def setup():
    require('hosts', 'path')
    put('dependencies/pip-0.8.1.tar.gz', env.path)
    put('dependencies/virtualenv-1.5.1.tar.gz', env.path)
    
    with cd(env.path):
        sudo('easy_install pip-0.8.1.tar.gz')
        sudo('pip install virtualenv-1.5.1.tar.gz')
        run('mkdir -p packages')
        run('mkdir -p releases')
        run('mkdir -p uploads')
        sudo('chown -R www-data:www-data uploads')

def version(version):
    env.version = version
    env.version_path = '%s/releases/%s' % (env.path, env.version)

def create_version():
    require('hosts', 'path')
    version(time.strftime('%Y%m%d%H%M%S'))
    
    with cd(env.path):
        run('mkdir -p releases/%s' % env.version)
    
    upload_tar_from_git()
    setup_virtualenv()
    install_requirements()

def deploy():
    "Specify a specific version to be made live"
    require('hosts', 'path', 'version')
    
    with cd(env.path):
        if exists('releases/previous'):
            run('rm releases/previous')
        if exists('releases/current'):
            run('mv releases/current releases/previous')
        run('ln -s %s releases/current' % env.version_path)

        # Install virtualenv
        sudo('cp %(version_path)s/%(project_name)s/configs/live/virtualhosts/bugle /etc/apache2/sites-available/')
        sudo('a2ensite bugle')
    
        run('rm -f %(version_path)s/%(project_name)s/static/admin' % env)
        run('ln -s %(version_path)s/%(project_name)s_ve/lib/python2.5/site-packages/django/contrib/admin/media %(version_path)s/%(project_name)s/static/admin' % env)
        run('rm -f %(version_path)s/%(project_name)s/uploads' % env)
        run('ln -s %(path)s/uploads %(version_path)s/%(project_name)s/uploads' % env)

    
    restart_apache()
    
def setup_virtualenv():
    require('version_path')
    
    with cd(env.version_path):
        run('mkdir -p %s_ve' % env.project_name)
        run('virtualenv %s_ve/' % env.project_name)

def upload_tar_from_git():
    "Create an archive from the current Git master branch and upload it"
    require('version', 'path')
    
    local('git archive --format=tar master | gzip > %s.tar.gz' % env.version)
    with cd(env.path):
        run('mkdir -p releases/%s' % env.version)
        put('%s.tar.gz' % env.version, '%s/packages/' % env.path)
        with cd('releases/%s' % env.version):
            run('tar zxf ../../packages/%s.tar.gz' % env.version)
    local('rm %s.tar.gz' % env.version)

def install_requirements():
    "Install the required packages from the requirements file using pip"
    require('version_path')
    with cd(env.version_path):
        run('pip install --upgrade -E %s_ve/ -r requirements.txt' % env.project_name)

def restart_apache():
    sudo('/etc/init.d/apache2 force-reload')

def setup_dev():
    require('version_path')
    with cd(env.path):
        sudo('easy_install dependencies/pip-0.8.1.tar.gz')
        sudo('pip install dependencies/virtualenv-1.5.1.tar.gz')
    setup_virtualenv()
    install_requirements()

def clean():
    local("find ./%s/ -name '*.pyc' -exec rm -rf {} \;" % env.project_name)


######################################
# Utils
######################################

def use_virtualenv():
    # This can use prefix when it's available!
    return 'source %s/%s_ve/bin/activate && ' % (env.version_path, env.project_name)