cli

#!/usr/bin/env python3
import click
from shiftuser.cli import user_cli
from boiler.cli import cli as boiler_cli
from boiler.cli import db

# add orm cli
user_cli.add_command(db.cli, name='db')

user_cli.add_command(boiler_cli.run, name='run')




# -----------------------------------------------------------------------------
# Commands
# -----------------------------------------------------------------------------

@user_cli.command(name='importable')
def importable():
    """ Make shiftuser module importable"""
    import site
    import os

    pth = 'shiftuser.pth'
    dst = os.path.join(site.getsitepackages()[0], pth)
    if os.path.exists(dst):
        print('Destination path file exists! {}\n'.format(dst))
        return

    import_from_here = os.path.dirname(os.path.realpath(__file__))
    with open(dst, 'a') as file:
        file.write('{}\n'.format(import_from_here))


@user_cli.command(name='test', context_settings=dict(ignore_unknown_options=True))
@click.argument('nose_argsuments', nargs=-1, type=click.UNPROCESSED)
def test(nose_argsuments):
    """ Run shiftuser tests """
    from nose import run
    params = ['__main__', '-c', 'nose.ini']
    params.extend(nose_argsuments)
    run(argv=params)

# -----------------------------------------------------------------------------
# And run
# -----------------------------------------------------------------------------


user_cli()