fabfile.py

from fabric.api import *
from fabric import operations
from fabtools import require
import fabtools
import requests
import os

PROJ_NAME = 'orl'
VENV_DIR = '~/{}_env/'.format(PROJ_NAME)
PROJ_DIR = '~/{}/'.format(PROJ_NAME)


@task
def deploy():
    with prefix('source ' + VENV_DIR + 'bin/activate'):
        with cd(PROJ_DIR):
            run('git pull')
            fabtools.python.install_requirements(PROJ_DIR + 'requirements.txt')
            run('python setup.py sdist')
            with cd('dist'):
                run('tar xfzv openrunlog-0.1.tar.gz')
                run('cp -r openrunlog-0.1/openrunlog $VIRTUAL_ENV/lib/python2.7/site-packages/')
    restart()


@task
def install_upstart_conf():
    with cd(PROJ_DIR):
        run('sudo cp openrunlog.conf /etc/init/openrunlog.conf')


@task
def start():
    run('sudo start openrunlog')


@task
def stop():
    run('sudo stop openrunlog')


@task
def restart():
    run('sudo restart openrunlog')


@task
def uninstall():
    run('rm -rf ' + PROJ_DIR)
    run('sudo rm /etc/init/openrunlog.conf')


@task
def setup():
    require.python.pip()
    fabtools.python.install('virtualenv', use_sudo=True)
    fabtools.require.deb.package('git-core')

    install_node()

    # Create venv
    run('virtualenv -p `which python2.7` ' + PROJ_NAME + '_env')

    # activate venv
    with prefix('source ' + VENV_DIR + 'bin/activate'):
        # clone repo
        run('git clone https://github.com/davidwilemski/openrunlog.git ' + PROJ_NAME)

        # install requirements
        fabtools.python.install_requirements(PROJ_DIR + 'requirements.txt')
        with cd(PROJ_DIR):
            run('python setup.py sdist')
            with cd('dist'):
                run('tar xfzv openrunlog-0.1.tar.gz')
                run('cp -r openrunlog-0.1/openrunlog $VIRTUAL_ENV/lib/python2.7/site-packages/')

    install_upstart_conf()
    install_mongo()
    start()


@task
def install_mongo():
    with prefix('source ' + VENV_DIR + 'bin/activate'):
        run('mongoctl install-mongodb')


@task
def install_node():
    # needed for PyExecJS
    fabtools.require.deb.package('nodejs')


@task
def migrate_mongoengine():
    with prefix('source ' + VENV_DIR + 'bin/activate'):
        with cd(PROJ_DIR):
            run('python openrunlog/scripts/migrations/mongoengine07to08.py')
    start()


@task
def shell():
    return operations.open_shell()


@task
def recalculate_streak(user=None):
    with prefix('source ' + VENV_DIR + 'bin/activate'):
        with cd(PROJ_DIR):
            run('python openrunlog/scripts/recalculate_streaks.py')


@task
def invalidate_cache(user=None):
    with prefix('source ' + VENV_DIR + 'bin/activate'):
        with cd(PROJ_DIR):
            run('python openrunlog/scripts/invalidate_cache.py')