Apply autopep8 on all staged files and verify them with pycodestyle

pre-commit

#!/usr/bin/env python
from __future__ import with_statement, print_function
import os
import re
import shutil
import subprocess
import sys
import tempfile


def system(*args, **kwargs):
    kwargs.setdefault('stdout', subprocess.PIPE)
    proc = subprocess.Popen(args, **kwargs)
    out, err = proc.communicate()
    return out


def apply_autopep8(filepath):
    args = ['autopep8', '--in-place']
    args.append(filepath)
    output = system(*args)


def verify_pycodestyle(filepath):
    args = ['pycodestyle']
    args.append(filepath)
    output = system(*args)
    if len(output) != 0:
        print(output)
        exit(1)


def main():
    try:
        import autopep8 as ap8
    except ImportError:
        print("'autopep8' is required. Please install with `pip install autopep8`.", file=sys.stderr)
        exit(1)

    modified = re.compile('^[AM]+\s+(?P<name>.*\.py)', re.MULTILINE)
    basedir = system('git', 'rev-parse', '--show-toplevel').decode("utf-8").strip()
    files = system('git', 'status', '--porcelain').decode("utf-8")
    files = modified.findall(files)
    for name in files:
        filepath = os.path.join(basedir, name)
        apply_autopep8(filepath)
        verify_pycodestyle(filepath)
        system("git", "add", filepath)


if __name__ == '__main__':
    main()