fabfile.py

from __future__ import with_statement
import datetime

from fabric.api import *
from config.fabconfig import *

env.hosts = REMOTE_HOSTS
env.user = USER
env.sudo_user = SUDO_USER
env.key_filename = KEY_FILENAME


def setup_clean():
    with prefix('source %s/bin/activate' % VENV_PATH):
        with cd(WORKING_PATH):
            try:
                run("rm -rf %s" % CURRENT_DIR)
            except:
                pass
            try:
                run("rm -rf %s" % RELEASES_DIR)
            except:
                pass
            try:
                run("rm -rf %s" % CHECKOUT_DIR)
            except:
                pass
            run("mkdir -p %s %s" % (RELEASES_DIR, SHARED_DIR))
            run("git clone %s %s" % (GIT_URL, CHECKOUT_DIR))
    create_release()
    symlink_current()


def use_branch():
    with cd("%s/%s" % (WORKING_PATH, CHECKOUT_DIR)):
        run("git checkout %s" % GIT_BRANCH)


def update_remote_code():
    with cd("%s/%s" % (WORKING_PATH, CHECKOUT_DIR)):
        run("git pull")


def create_release():
    release = datetime.datetime.now().isoformat()
    release_dir = "%s/%s" % (RELEASES_DIR, release)
    with cd(WORKING_PATH):
        run("mkdir %s" % release_dir)
        run("cp -Rv %s/* %s/" % (CHECKOUT_DIR, release_dir))
        run("rm -rf %s/.git %s/.gitignore" % (release_dir, release_dir))


def symlink_current():
    with cd(WORKING_PATH):
        dirs = run("ls -d %s/*/" % RELEASES_DIR).split()
        dirs.sort()
        dirs.reverse()
        dir = dirs[0].strip('/')
        try:
            run("unlink %s" % CURRENT_DIR)
        except:
            pass
        run("ln -nfs %s %s" % (dir, CURRENT_DIR))
        with cd(dir):
            for path in SHARED_FILES:
                run("ln -nfs %s/%s/%s %s" % (WORKING_PATH, SHARED_DIR,
                                             path.split('/')[-1],
                                             path))
    with cd("%s/bin" % HOME_PATH):
        try:
            run("unlink ./run")
            run("unlink ./stop")
        except:
            pass
        run("ln -nfs %s/%s/config/run" % (WORKING_PATH, CURRENT_DIR))
        run("ln -nfs %s/%s/config/stop" % (WORKING_PATH, CURRENT_DIR))


def sync_remote_assets():
    with prefix("source %s/bin/activate" % VENV_PATH):
        with cd("%s/%s" % (WORKING_PATH, CURRENT_DIR)):
            run("./manage.py collectstatic --noinput")
            try:
                run("./manage.py compress --force")
            except:
                pass


def install_dependencies():
    with prefix("source %s/bin/activate" % VENV_PATH):
        with cd("%s/%s" % (WORKING_PATH, CURRENT_DIR)):
            run("pip install -r requirements.txt")


def run_migrations():
    with prefix("source %s/bin/activate" % VENV_PATH):
        with cd("%s/%s" % (WORKING_PATH, CURRENT_DIR)):
            try:
                run("./manage.py createcachetable %s" % CACHE_TABLE_NAME)
            except:
                pass
            run("./manage.py syncdb --noinput")
            run("./manage.py migrate --noinput")


def stop():
    run("sudo /sbin/stop tcamp", pty=False)
    # with cd(HOME_PATH):
    #     run("./bin/stop")


def restart():
    run("sudo /sbin/restart tcamp", pty=False)
    # with cd(HOME_PATH):
    #     run("./bin/run")


def dump_cache():
    run("sudo /etc/init.d/memcached restart", pty=False)


def cleanup():
    with cd(WORKING_PATH):
        dirs = run("ls -d %s/*/" % RELEASES_DIR).split()
        dirs.sort()
        dirs.reverse()
        try:
            to_clean = dirs[PAST_RELEASES:]
            for dir in to_clean:
                run("rm -rf %s" % dir)
        except IndexError:
            pass


def stage():
    local("git push heroku master")


def deploy():
    update_remote_code()
    create_release()
    symlink_current()
    install_dependencies()
    sync_remote_assets()
    run_migrations()
    cleanup()
    restart()


def rollback():
    with cd(WORKING_PATH):
        dirs = run("ls -d %s/*/" % RELEASES_DIR).split()
        dirs.sort()
        dirs.reverse()
        dir = dirs[1].strip('/')
        try:
            run("unlink %s" % CURRENT_DIR)
        except:
            pass
        run("ln -s %s %s" % (dir, CURRENT_DIR))
        run("rm -rf %s" % dirs[0].strip('/'))
    install_dependencies()
    sync_remote_assets()
    restart()