test-auth/fabfile.py

from __future__ import with_statement
import time
import os
from fabric.api import *
from fabric.contrib.files import *
from fabric.utils import warn
from hashlib import sha1

env.repo = 'git://dev.pleaseignore.com/dreddit-auth.git'

def production():
    "Use the production enviroment on Web1"
    env.hosts = ['dreddit@web1.pleaseignore.com']
    env.path = '/home/dreddit/apps'
    env.user = 'auth'
    env.vhost = '/auth'
    env.password = sha1('%s-%s' % (env.user, env.vhost)).hexdigest()

def test():
    "Use the test enviroment on Web2"
    env.hosts = ['dreddit@web2.pleaseignore.com']
    env.path = '/home/dreddit/apps'
    env.user = 'auth'
    env.vhost = '/auth'
    env.password = sha1('%s-%s' % (env.user, env.vhost)).hexdigest()

def deploy():
    """
    Do a fresh deployment to a new or clean server
     """
    setup()


def setup():
    """
    Deploy the files to the server and setup virtualenv and a simple
    SQlite DB setup
    """
    require('hosts')
    require('path')

    deploy_repo()

    setup_virtualenv()


def push():
    """
    Push the latest revision to the server and update the enviroment
    """
    require('hosts')
    require('path')

    stop()
    update_repo()
    setup_virtualenv()
    migrate()
    start()


def setup_virtualenv():
    """
    Generate the virtualenv enviroment
    """
    require('path')

    with cd('%(path)s/dreddit-auth/' % env):
        run('./bin/setup-env.sh' % env)


def setup_db():
    """
    Setup a simple SQlite DB using the default settings
    """
    require('path')

    with cd('%(path)s/dreddit-auth/' % env):
        run('env/bin/python app/manage.py syncdb --noinput --migrate')


def deploy_repo():
    """
    Do the initial repository checkout
    """
    require('hosts')
    require('path')

    run('mkdir -p %(path)s' % env)
    with cd(env.path):
        run('git clone %(repo)s' % env)

    with cd('%(path)s/dreddit-auth/' % env):
        run('mkdir logs')


def update_repo():
    """
    Pulls the latest commits to the repository
    """
    require('hosts')
    require('path')

    with cd('%(path)s/dreddit-auth/' % env):
        run('git pull')

    deploy_static()


def reset_repo():
    """
    Does a hard reset on the remote repo
    """
    require('hosts')
    require('path')

    with cd('%(path)s/dreddit-auth/' % env):
        run('git reset --hard')


def migrate():
    """
    Do any South database migrations
    """
    require('hosts')
    require('path')

    with cd('%(path)s/dreddit-auth/' % env):
        run('env/bin/python app/manage.py migrate')


def start_celeryd():
    """
    Start the celeryd server
    """
    require('hosts')
    require('path')

    with cd('%(path)s/dreddit-auth/' % env):
        run('. env/bin/activate; app/manage.py celeryd_detach -l INFO -B --pidfile logs/celeryd.pid -f logs/celeryd.log -n auth-processor' % env)


def stop_celeryd():
    """
    Stop any running FCGI instance
    """
    require('hosts')
    require('path')

    with cd('%(path)s/dreddit-auth/' % env):
        if exists('logs/celeryd.pid'):
            run('kill `cat logs/celeryd.pid`')
            time.sleep(2)
            run('rm -f logs/celeryd.pid')
        else:
            warn('celeryd isn\'t running')

def restart_celeryd():
    """
    Restart the celery daemon
    """
    stop_celeryd()
    time.sleep(2)
    start_celeryd()


def start_uwsgi():
    """
    Start uWSGI
    """
    pass


def restart_uwsgi():
    """
    Restart the uWSGI daemon
    """
    pass


def start():
    """
    Start uWSGI and Celery
    """
    start_uwsgi()
    start_celeryd()


def stop():
    """
    Stop Celery
    """
    stop_celeryd()
    info('Can\'t stop uWSGI')

def restart():
    """
    Restart uWSGI and Celery
    """
    stop_celeryd()
    time.sleep(5)
    restart_uwsgi()
    start_celeryd()

def clear_logs():
    with cd('%(path)s/dreddit-auth/' % env):
        run('rm ./logs/*.log')

def deploy_static():
    with cd('%(path)s/dreddit-auth/' % env):
        run('./app/manage.py collectstatic -v0 --noinput')