#!/usr/bin/env python
from __future__ import print_function
import argparse
import subprocess
import sys
verbose = False
def call(*args, **kwargs):
if verbose:
print(' '.join(args[0]))
ret = subprocess.call(*args, **kwargs)
if ret != 0:
sys.exit(ret)
def main(argv):
parser = argparse.ArgumentParser(prog='run')
parser.add_argument('-v', '--verbose', action='store_true')
subps = parser.add_subparsers()
subp = subps.add_parser('build', help='build the package')
subp.set_defaults(func=run_build)
subp = subps.add_parser('clean', help='Remove any local files.')
subp.set_defaults(func=run_clean)
subp = subps.add_parser('coverage',
help='Run tests and report code coverage.')
subp.set_defaults(func=run_coverage)
subp = subps.add_parser('format',
help='Reformat the source code.')
subp.set_defaults(func=run_format)
subp = subps.add_parser('help',
help='Get help on a subcommand.')
subp.add_argument(nargs='?', action='store', dest='subcommand',
help='The command to get help for.')
subp.set_defaults(func=run_help)
subp = subps.add_parser('install',
help='build the package and install locally.')
subp.set_defaults(func=run_install)
subp.add_argument('--system', action='store_true',
help=('Install to the system site-package dir '
'rather than the user\'s (requires root).'))
subp = subps.add_parser('lint',
help='run lint over the source')
subp.set_defaults(func=run_lint)
subp = subps.add_parser('tests',
help='run the tests')
subp.set_defaults(func=run_tests)
args = parser.parse_args(argv)
global verbose
if args.verbose:
verbose = True
args.func(args)
def run_build(args):
del args
call([sys.executable, 'setup.py', 'build', '--quiet'])
def run_clean(args):
del args
call(['git', 'clean', '-fxd'])
def run_coverage(args):
del args
call(['python', '-m', 'coverage', 'run', '-m', 'unittest',
'discover', '-p', '*_test.py'])
call(['python3', '-m', 'coverage', 'run', '--append', '-m', 'unittest',
'discover', '-p', '*_test.py'])
call([sys.executable, '-m', 'coverage', 'report', '--show-missing'])
def run_format(args):
del args
call('autopep8 --in-place *.py */*.py', shell=True)
def run_help(args):
if args.subcommand:
main([args.subcommand, '--help'])
main(['--help'])
def run_install(args):
if args.system:
argv = []
else:
argv = ['--user']
call([sys.executable, 'setup.py', 'install'] + argv)
def run_lint(args):
del args
call('pylint --rcfile=pylintrc */*.py', shell=True)
def run_tests(args):
del args
call([sys.executable, '-m', 'unittest', 'discover',
'-p', '*_test.py'])
if __name__ == '__main__':
sys.exit(main(sys.argv[1:]))