dssg/triage

import argparse
from pathlib import Path

from argcmdr import local, LocalRoot, Local
from plumbum import local as plumlocal


ROOT_PATH = Path(__file__).parent.resolve()


class Development(LocalRoot):
    """Commands to aid in Triage library development"""
    pass


@Development.register
@local('remainder', metavar='alembic arguments', nargs=argparse.REMAINDER)
def alembic(context, args):
    """Configuration wrapper to use the Alembic schema migrations library for Triage development.
    Try `alembic -h` or `manage alembic -- -h` to see a description of all
    the available subcommands"""
    return context.local['env'][
        'PYTHONPATH=' + str(ROOT_PATH / 'src'),
        'alembic',
        '-c', ROOT_PATH / 'src' / 'triage' / 'component' / 'results_schema' / 'alembic.ini',
        '-x', 'db_config_file=database.yaml',
        args.remainder,
    ]


@Development.register
class Docs(Local):
    """View Triage documentation through local server"""
    def prepare(self, args):
        yield plumlocal['python']['docs/update_docs.py']
        with plumlocal.cwd(ROOT_PATH / 'docs'):
            yield plumlocal['mkdocs']['serve']