#!/usr/bin/env python
import subprocess
import sys

def get_staged():
    proc = subprocess.Popen(['git', 'diff', '--name-only', '--cached'], stdout=subprocess.PIPE)
    staged_files = proc.stdout.readlines()
    staged_files = [f.decode('utf-8') for f in staged_files]
    staged_files = [f.strip() for f in staged_files]
    staged_files = [f for f in staged_files if f.endswith('.py')]
    return staged_files

def main():
    abort = False
    staged = get_staged()
    print("RUNNING PEP8")
    try:
        out = subprocess.check_output((sys.executable, '-m', 'pytest', '--pep8',
                                       '-m',  'pep8', '-n0', *staged),
                                stderr=subprocess.STDOUT)
    except subprocess.CalledProcessError as e:
        out = e.stdout
    out = out.decode('utf8')
    print(out)
    if 'FAILURES' in out:
        abort = True

    print("Running Tests")
    try:
        out = subprocess.check_output((sys.executable, '-m', 'pytest', 'tests'),
                                stderr=subprocess.STDOUT)
    except subprocess.CalledProcessError as e:
        out = e.stdout
    out = out.decode('utf8')
    print(out)
    if 'FAILURE' in out:
        abort = True

    if abort:
        sys.exit(1)
    else:
        sys.exit(0)

if __name__ == '__main__':
    main()