manage.py

#!/usr/bin/env python
# vim: sw=4:ts=4:expandtab

""" 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_WHERE = ["meza", "tests", "examples", "setup.py", "manage.py"]


def _upload():
    """Upload distribution files"""
    _uploaddir = p.join(BASEDIR, "dist", "*")
    url = "https://upload.pypi.org/legacy/"
    check_call(["twine", "upload", "--repository-url", url, _uploaddir])


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.command
def lint(where=None, strict=False):
    """Check style with linters"""
    extra = where.split(" ") if where else DEF_WHERE
    args = ["pylint", "--rcfile=tests/pylintrc", "-rn", "-fparseable"]

    try:
        if strict:
            check_call(args + extra)
        else:
            check_call(["flake8"] + extra)
    except CalledProcessError as e:
        exit(e.returncode)


@manager.arg("where", "w", help="Modules to check")
@manager.command
def prettify(where=None):
    """Prettify code with black"""
    extra = where.split(" ") if where else DEF_WHERE

    try:
        check_call(["black"] + extra)
    except CalledProcessError as e:
        exit(e.returncode)


@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("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(where=None, stop=None, **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 %s" % where if where 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()