pavement.py

import os
from paver.easy import call_task, info, sh, task


@task
def reset_db():
    """
    Reset the Django db, keeping the admin user
    """
    # TODO read stuff from settings instead than hardcoding
    sh('export PGPASSWORD=hypermap')
    sh('psql -U hypermap -h localhost -c "drop database hypermap;" postgres')
    sh('psql -U hypermap -h localhost -c "create database hypermap with owner hypermap;" postgres')
    sh("python manage.py makemigrations aggregator")
    sh("python manage.py migrate")
    sh("echo \"from django.contrib.auth.models import User; User.objects.create_superuser('admin', 'admin@example.com', 'admin')\" | python manage.py shell")


@task
def run_tests():
    """
    Executes the entire test suite.
    """
    os.environ["REGISTRY_SKIP_CELERY"] = "True"
    sh('python manage.py test hypermap.aggregator --failfast')
    sh('python manage.py test hypermap.dynasty --failfast')
    sh('flake8 hypermap')


@task
def run_integration_tests():
    """
    Executes the entire test suite.
    """
    call_task('start')
    sh('python manage.py test tests.integration --settings=settings.test --failfast')
    call_task('stop')


@task
def start():
    sh('python manage.py runserver 0.0.0.0:8000 &')


@task
def stop():
    kill_process('python', 'hypermap/manage.py')


def kill_process(procname, scriptname):
    """kill WSGI processes that may be running in development"""

    # from http://stackoverflow.com/a/2940878
    import signal
    import subprocess

    p = subprocess.Popen(['ps', 'aux'], stdout=subprocess.PIPE)
    out, err = p.communicate()

    for line in out.decode().splitlines():
        if procname in line and scriptname in line:
            pid = int(line.split()[1])
            info('Stopping %s %s %d' % (procname, scriptname, pid))
            os.kill(pid, signal.SIGKILL)