pre_push.py

#!/usr/bin/env python3
"""Run static analysis on the project."""

import argparse
import sys
from os import path
from shutil import rmtree
from subprocess import CalledProcessError, check_call
from tempfile import mkdtemp

current_directory = path.abspath(path.join(__file__, ".."))


def do_process(args, shell=False):
    """Run program provided by args.

    Return True on success.

    Output failed message on non-zero exit and return False.

    Exit if command is not found.
    """
    print("Running: {}".format(" ".join(args)))
    try:
        check_call(args, shell=shell)
    except CalledProcessError:
        print("\nFailed: {}".format(" ".join(args)))
        return False
    except Exception as exc:
        sys.stderr.write(str(exc) + "\n")
        sys.exit(1)
    return True


def run_static():
    """Runs the static tests.

    Returns a statuscode of 0 if everything ran correctly.
    Otherwise, it will return statuscode 1
    """
    success = True
    success &= do_process(["black", "."])
    success &= do_process(["flake8", "--exclude=.eggs,build,docs"])
    success &= do_process(["pydocstyle", "prawdditions"])
    tmp_dir = mkdtemp()
    try:
        success &= do_process(
            ["sphinx-build", "-W", "--keep-going", "docs", tmp_dir]
        )
    finally:
        rmtree(tmp_dir)
    return success


def run_unit():
    """Runs the unit-tests.

    Follows the behavior of the static tests,
    where any failed tests cause pre_push.py to fail.
    """
    return do_process([sys.executable, "pytest"])


def main():
    """Runs the main function.

    usage: pre_push.py [-h] [-n] [-u] [-a]

    Run static and/or unit-tests
    """
    parser = argparse.ArgumentParser(
        description="Run static and/or unit-tests"
    )
    parser.add_argument(
        "-n",
        "--unstatic",
        action="store_true",
        help="Do not run static tests (black/flake8/pydocstyle/sphinx-build)",
        default=False,
    )
    parser.add_argument(
        "-u",
        "--unit-tests",
        "--unit",
        action="store_true",
        default=False,
        help="Run the unit tests",
    )
    parser.add_argument(
        "-a",
        "--all",
        action="store_true",
        default=False,
        help="Run all of the tests (static and unit). "
        "Overrides the unstatic argument.",
    )
    args = parser.parse_args()
    success = True
    try:
        if not args.unstatic or args.all:
            success &= run_static()
        if args.all or args.unit_tests:
            success &= run_unit()
    except KeyboardInterrupt:
        return int(not False)
    return int(not success)


if __name__ == "__main__":
    exit_code = main()
    print(
        "\npre_push.py: Success!" if not exit_code else "\npre_push.py: Fail"
    )
    sys.exit(exit_code)