tasks.py

import os
import sys

from python_boilerplate.tasks import *
from python_boilerplate.tasks import django
from python_boilerplate.tasks import js

ns.add_collection(js)
ns.add_collection(django)
sys.path += ['src']


@task
def makemessages(ctx):
    """
    Runs the manage.py makemessages command with sane defaults.
    """

    paths = os.listdir(os.path.dirname(__file__))
    paths.remove('src')
    globs = [repr('%s/*' % f if os.path.isdir(f) else f) for f in paths]
    ignore_patterns = globs * 2
    ignore_patterns[::2] = ['-i'] * len(globs)
    cmd = ['python', 'manage.py', 'makemessages', '-i', 'codeschool/vendor/*']
    cmd.extend(ignore_patterns)
    ctx.run(' '.join(cmd), echo=True, pty=True)


@task
def compilemessages(ctx):
    """
    Runs the manage.py makemessages command with sane defaults.
    """

    paths = os.listdir(os.path.dirname(__file__))
    paths.remove('src')
    globs = [repr('%s*' % f if os.path.isdir(f) else f) for f in paths]
    ignore_patterns = globs * 2
    ignore_patterns[::2] = ['-i'] * len(globs)
    cmd = ['python', 'manage.py', 'makemessages', '-i', 'codeschool/vendor/*']
    cmd.extend(ignore_patterns)
    ctx.run(' '.join(cmd), echo=True, pty=True)


@task
def develop(ctx):
    """
    Prepares environment for development.
    """

    # Install [dev]
    print('Installing Python dev dependencies...')
    ctx.run('pip install .[dev] -r requirements.txt')

    # Js configurations
    js.install()
    print()
    js.build()

    # Run manage.py commands
    # django_manage('makemigrations')
    # django_manage('migrate')
    # createsuperuser(ctx)


@task
def docker_build(ctx, rebuild_static=False):
    if rebuild_static:
        ctx.run('tar czpf static.tar.gz collect/static/')
    ctx.run('docker build -t codeschool:deploy .', pty=True)


@task
def docker_run(ctx, deploy=False, shell=False):
    """
    Run dev:
        docker run -it -p 8080:80 -v /app/codeschool/src:/app/src/ -v /app/db:/app/db -v /app/collect/media:/var/www/media -e PYTHONPATH=/app/src/ codeschool:deploy shell
    Run production:
        docker run -p 80:80 -v /app/codeschool/src:/app/src/ -v /app/db:/app/db -v /app/collect/media:/var/www/media -e PYTHONPATH=/app/src/ codeschool:deploy
    """
    cmd = (
        'docker run -ti -p {port}:80 '
        '-v {src}:/app/src/ '
        '-v {db}:/app/db '
        '-v {collect}/media:/var/www/media '
        '-e PYTHONPATH=/app/src/ '
        'codeschool:deploy{tail}'
    )

    kwargs = {
        'src': os.path.abspath('src'),
        'db': os.path.abspath('db'),
        'collect': os.path.abspath('collect'),
        'tail': ' shell' if shell else '',
    }

    def run(cmd):
        print('sh: %s' % cmd)
        ctx.run(cmd, pty=True)

    if deploy:
        run(cmd.format(port=80, **kwargs))
    else:
        run(cmd.format(port=8080, **kwargs))


def is_redis_running(port=None):
    """
    Check if redis is running in the given port.
    """

    import redis
    conn = redis.Connection(port=port or 6379)
    try:
        conn.connect()
        return True
    except redis.ConnectionError:
        return False


@task
def redis(ctx, docker=False, port=0, verbose=False, sentinel=False):
    """
    Run redis server (possibly using version in docker hub).
    """

    if is_redis_running(port or None):
        print('Redis is already running at port %s' % (port or 6379))
        print('Bye!')
        return

    if docker:
        print('Running dockerized redis-server')
        cmd = 'docker run redis:alpine'
    else:
        result = ctx.run('redis-server -v', hide=True)
        if not result.ok:
            print('There is no redis-server installed in your system')
            print('Please install it using apt-get install redis or any other '
                  'means your distribution provides.')
            raise SystemExit

        cmd = 'redis-server'

        if port:
            cmd += ' --port %s' % port
        if verbose:
            cmd += ' --loglevel verbose'
        if sentinel:
            cmd += ' --sentinel'

        ctx.run(cmd)


@task
def celery(ctx):
    """
    Runs celery.
    """

    ctx.run('celery --app=codeschool.celery:app worker --loglevel=INFO')