dribdat/manage.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
import click

from flask.cli import FlaskGroup

from dribdat.app import init_app
from dribdat.settings import DevConfig, ProdConfig

HERE = os.path.abspath(os.path.dirname(__file__))
TEST_PATH = os.path.join(HERE, 'tests')


def shell_context():
    """Return context dict for a shell session"""
    from dribdat.user.models import User, Event, Project, Category, Activity
    return {
        'User': User, 'Event': Event, 'Project': Project,
        'Category': Category, 'Activity': Activity
    }


def create_app(script_info=None):
    """Initialise the app object"""
    if os.environ.get("DRIBDAT_ENV") == 'prod':
        app = init_app(ProdConfig)
    else:
        app = init_app(DevConfig)
    app.shell_context_processor(shell_context)
    return app


@click.command()
@click.option('--name', default="features", help='Which test file to run.')
def featuretest(name):
    """Run feature tests."""
    import pytest
    feat_test = os.path.join(TEST_PATH, "test_%s.py" % name)
    return pytest.main([feat_test, '--disable-warnings'])


@click.command()
def test():
    """Run all tests."""
    import pytest
    test_opts = [TEST_PATH]
    exit_code = pytest.main(test_opts)
    return exit_code


@click.group(cls=FlaskGroup, create_app=create_app)
def cli():
    """This is a management script for this application."""


cli.add_command(test)
cli.add_command(featuretest)

if __name__ == '__main__':
    cli()