manage.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# vim: sw=4:ts=4:expandtab
"""
    manage.py
    ~~~~~~~~~

    A script to manage development tasks
"""
from os import path as p
from subprocess import call, check_call, CalledProcessError
from manager import Manager

manager = Manager()
BASEDIR = p.dirname(__file__)


def upload_():
    """Upload distribution files"""
    upload_dir = p.join(BASEDIR, 'dist', '*')
    url = 'https://upload.pypi.org/legacy/'
    check_call(f'twine upload --repository-url {url} {upload_dir}', shell=True)


def sdist_():
    """Create a source distribution package"""
    check_call(p.join(BASEDIR, 'helpers', 'srcdist'))


def wheel_():
    """Create a wheel package"""
    check_call(p.join(BASEDIR, 'helpers', 'wheel'))


def clean_():
    """Remove Python file and build artifacts"""
    check_call(p.join(BASEDIR, 'helpers', 'clean'))


@manager.command
def check():
    """Check staged changes for lint errors"""
    exit(call(p.join(BASEDIR, 'helpers', 'check-stage')))


@manager.arg('where', 'w', help='Modules to check')
@manager.arg('strict', 's', help='Check with pylint')
@manager.arg(
    'parallel', 'p', help='Run linter in parallel in multiple processes',
    type=bool, default=False)
@manager.command
def lint(where=None, strict=False, parallel=False):
    """Check style with linters"""
    args = 'pylint --rcfile=tests/.pylintrc -rn mezmorize'
    args += ' -j 0' if parallel else ''

    try:
        check_call(['flake8', where] if where else 'flake8')
        check_call(args.split(' ')) if strict else None
    except CalledProcessError as e:
        exit(e.returncode)


@manager.command
def pipme():
    """Install requirements.txt"""
    exit(call('pip install -r requirements.txt'.split(' ')))


@manager.command
def require():
    """Create requirements.txt"""
    cmd = 'pip freeze -l | grep -vxFf dev-requirements.txt > requirements.txt'
    exit(call(cmd, shell=True))


@manager.arg('source', 's', help='the tests to run', default=None)
@manager.arg('where', 'w', help='test path', default=None)
@manager.arg(
    'stop', 'x', help='Stop after first error', type=bool, default=False)
@manager.arg(
    'failed', 'f', help='Run failed tests', type=bool, default=False)
@manager.arg(
    'cover', 'c', help='Add coverage report', type=bool, default=False)
@manager.arg('tox', 't', help='Run tox tests', type=bool, default=False)
@manager.arg('detox', 'd', help='Run detox tests', type=bool, default=False)
@manager.arg(
    'verbose', 'v', help='Use detailed errors', type=bool, default=False)
@manager.arg(
    'parallel', 'p', help='Run tests in parallel in multiple processes',
    type=bool, default=False)
@manager.arg(
    'debug', 'D', help='Use nose.loader debugger', type=bool, default=False)
@manager.command
def test(source=None, where=None, stop=False, **kwargs):
    """Run nose, tox, and script tests"""
    opts = '-xv' if stop else '-v'
    opts += ' --with-coverage' if kwargs.get('cover') else ''
    opts += ' --failed' if kwargs.get('failed') else ' --with-id'
    opts += ' --processes=-1' if kwargs.get('parallel') else ''
    opts += ' --detailed-errors' if kwargs.get('verbose') else ''
    opts += ' --debug=nose.loader' if kwargs.get('debug') else ''
    opts += ' -w {}'.format(where) if where else ''
    opts += ' {}'.format(source) if source else ''

    try:
        if kwargs.get('tox'):
            check_call('tox')
        elif kwargs.get('detox'):
            check_call('detox')
        else:
            check_call(('nosetests %s' % opts).split(' '))
    except CalledProcessError as e:
        exit(e.returncode)


@manager.command
def register():
    """Register package with PyPI"""
    exit(call(['python', p.join(BASEDIR, 'setup.py'), 'register']))


@manager.command
def release():
    """Package and upload a release"""
    try:
        clean_()
        sdist_()
        wheel_()
        upload_()
    except CalledProcessError as e:
        exit(e.returncode)


@manager.command
def build():
    """Create a source distribution and wheel package"""
    try:
        clean_()
        sdist_()
        wheel_()
    except CalledProcessError as e:
        exit(e.returncode)


@manager.command
def upload():
    """Upload distribution files"""
    try:
        upload_()
    except CalledProcessError as e:
        exit(e.returncode)


@manager.command
def sdist():
    """Create a source distribution package"""
    try:
        sdist_()
    except CalledProcessError as e:
        exit(e.returncode)


@manager.command
def wheel():
    """Create a wheel package"""
    try:
        wheel_()
    except CalledProcessError as e:
        exit(e.returncode)


@manager.command
def clean():
    """Remove Python file and build artifacts"""
    try:
        clean_()
    except CalledProcessError as e:
        exit(e.returncode)


if __name__ == '__main__':
    manager.main()