manage.py

#!/usr/bin/env python
import logging
import subprocess

import pusto

log = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO)


def sh(cmd):
    log.info(cmd)
    code = subprocess.call(cmd, shell=True)
    if code:
        raise SystemExit(code)
    return 0


def ssh(cmd):
    return sh(
        'ssh yadro.org -p2200 "%s"'
        % cmd.replace('"', '\\"').replace('$', '\\$')
    )


def process_args():
    parser, cmd = pusto.get_parser()

    cmd('deploy', help='deploy to server')\
        .arg('-c', '--clear', action='store_true', help='clear virtualenv')\
        .arg('-t', '--target', default='origin/master', help='checkout it')\
        .exe(lambda a: ssh(
            'cd /home/pusto/src'
            '&& git fetch origin' +
            '&& git checkout {}'.format(a.target) +
            (
                '&& rm -rf $(cat .venv) && virtualenv $(cat .venv)'
                if a.clear else ''
            ) +
            '&& source $(cat .venv)/bin/activate' +
            '&& ./bootstrap'
            '&& ./pusto.py build -b build-tmp'
            '&& rm -rf build'
            '&& mv build-tmp build'
            '&& supervisorctl pid nginx | xargs kill -s HUP'
        ))

    cmd('wheels', help='prepare wheels')\
        .arg('--init', action='store_true')\
        .exe(lambda a: sh(
            'pip install -U wheel'
            if a.init else
            'pip wheel -r requirements.txt -w wheels'
        ))

    cmd('docker', help='run docker container with nginx')\
        .exe(lambda a: sh(
            'docker run'
            '   -d -v $(pwd):/var/www -p 80:80 --name=pusto'
            '   pusto /bin/nginx'
        ))

    cmd('napokaz', help='napokaz updater')\
        .arg('--push', action='store_true')\
        .arg('--init', action='store_true')\
        .exe(lambda a: sh(
            'git remote add napokaz git@github.com:naspeh/napokaz.git'
            if a.init else
            'git subtree %s -P data/s/napokaz/src napokaz master'
            % ('push' if a.push else 'pull --squash')
        ))

    pusto.process(parser=parser)


if __name__ == '__main__':
    process_args()