fabfile.py

import platform

from fabric.api import local, settings
from fabric.context_managers import shell_env, lcd


APP_SUBMODULE_DIRECTORIES = (
    'dashboard',
    'pmp',
    'travel',
    'ap',
    'apd',
    'tpm',
)


def ssh(service):
    # https://stackoverflow.com/a/25293275/8207
    class NoBashException(Exception):
        pass

    with settings(abort_exception=NoBashException):
        try:
            local('docker exec -it etoolsinfra_%s /bin/bash' % service)
        except NoBashException:
            # fallback in case bash isn't installed
            local('docker exec -it etoolsinfra_%s /bin/sh' % service)


def reallyclean():
    local('docker-compose down --rmi all --volumes')
    local('docker volume rm $(docker volume ls -qf dangling=true)')


def devup(quick=False, DB_PORT="51322"):
    with shell_env(DB_PORT=DB_PORT):
        local('docker-compose up --force-recreate %s' % ('--build' if not quick else ''))


def devup_built(quick=False, DB_PORT="51322"):
    nginx_config = " -c '/nginx-built.conf'"
    front_end_command = "node express.js"
    with shell_env(NX_CONFIG=nginx_config, FE_COMMAND=front_end_command, DB_PORT=DB_PORT):
        local('docker-compose up --force-recreate %s' % ('--build' if not quick else ''))


def devup_built_windows(quick=False):
    local('docker-compose -f docker-compose.dev-built-win.yml up --force-recreate %s' % ('--build' if not quick else ''))


def backend_migrations():
    local('docker exec etoolsinfra_backend python /code/manage.py migrate_schemas --noinput')


def debug(quick=False, DEBUG_PORT='51312', DB_PORT="51322"):
    try:
        import netifaces as ni
    except ImportError:
        print("You must have the 'netifaces' package installed")
        return
    ni.ifaddresses('en0')
    ip = ni.ifaddresses('en0')[2][0]['addr']
    with shell_env(BACKEND_DEBUG="_debug", DEBUG_IP=ip, DEBUG_PORT=DEBUG_PORT, DB_PORT=DB_PORT):
        local('docker-compose up --force-recreate %s' % ('--build' if not quick else ''))


def remove_docker_containers():
    local('docker stop $(docker ps -q)')
    local('docker rm $(docker ps -q)')


def stop_docker_containers():
    local('docker stop $(docker ps -q)')


def up(quick=False):
    local('docker-compose -f docker-compose.yml up --force-recreate')


def _frontend_deps_init():
    # TODO retry after timeouts
    for frontend_app_dir in APP_SUBMODULE_DIRECTORIES:
        with lcd(frontend_app_dir):
            local('npm install')
            local('bower install')


def _frontend_deps_update():
    # TODO retry after timeouts
    for frontend_app_dir in APP_SUBMODULE_DIRECTORIES:
        with lcd(frontend_app_dir):
            local('npm update')
            local('bower update')


def _update_submodules():
    # TODO retry after timeouts
    # see https://stackoverflow.com/a/18799234/8207 for more information about submodule branch tracking
    local('git submodule update --init --recursive --remote')
    local(
        'git submodule foreach -q --recursive \'branch="$(git config -f $toplevel/.gitmodules submodule.$name.branch)";'
        'git checkout $branch; git merge origin/$branch\''
    )


def update(quick=False):
    local('git fetch --all')
    # TODO should we use `develop` on unicef/etools-infra rather than `master`?
    local('git checkout master')
    local('git merge origin/master')
    _update_submodules()
    # delete .pyc files - something creates some of them as root, sigh...
    local("sudo find . -name '*.pyc' -delete")
    if not quick:
        _frontend_deps_update()


def init():
    local('git fetch --all')

    _update_submodules()
    _frontend_deps_init()

    if platform.system() == 'Darwin':
        # TODO complain if postgres and/or GDAL are not installed
        pass
    if platform.system() == 'Linux':
        # TODO complain if postgres and/or GDAL are not installed
        pass
    if platform.system() == 'Windows':
        # TODO complain if postgres and/or GDAL are not installed
        pass


def test(app=None):
    # TODO add flag so caller can indicate --keepdb vs --noinput
    if app is not None:
        local('docker exec etoolsinfra_backend python'
              '  /code/core/manage.py test %s.tests --keepdb' % app)
    else:
        local('docker exec etoolsinfra_backend python'
              ' /code/core/manage.py test --keepdb')