.git.hooks.pre-commit

#!/usr/bin/env python
#-*- mode: python -*-

all_okay = True

from subprocess import Popen, PIPE
import sys
from termcolor import colored

syntax_checker = "pylint"

def run(command):
    p = Popen(command.split(), stdout=PIPE, stderr=PIPE)
    p.wait()
    return p.returncode, p.stdout.read().strip(), p.stderr.read()

_, files_modified, _= run("git diff-index --name-only HEAD")

for fname in files_modified.split():
    fname = fname.decode('utf-8')
    if fname.endswith(".py") and not fname.endswith('urls.py') and not fname.endswith('__init__.py'):
        exit_code, _, errors = run("%s %s"%(syntax_checker, fname))
        if exit_code != 0:
            all_okay = False
            for err in _.decode('utf-8').split(r'\n'):
                print(colored(err, 'red'))
            print("\rChecking syntax on %s: FAILED! \n"%(fname))
        else:
            print(colored("\rChecking syntax on %s: OK!"%(fname),'green'))

if not all_okay:
    sys.exit(1)