BAPSicle/dev/pre-commit

#!/usr/bin/env python3

# from https://github.com/chibiegg/git-autopep8
from __future__ import with_statement, print_function
import os
import re
import subprocess
import sys

# See for codes https://pypi.org/project/autopep8/ #features
# don't fill in both of these
select_codes = []
# "E121", "E122", "E123", "E124", "E125", "E126", "E127", "E128", "E129", "E131", "E501"]
ignore_codes = ["E402", "E226", "E24", "W50", "W690"]
# Add things like "--max-line-length=120" below
overrides = ["--max-line-length=127"]


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


def autopep8(filepath):
    args = ['autopep8', '--in-place']
    if select_codes and ignore_codes:
        print(u'Error: select and ignore codes are mutually exclusive')
        sys.exit(1)
    elif select_codes:
        args.extend(('--select', ','.join(select_codes)))
    elif ignore_codes:
        args.extend(('--ignore', ','.join(ignore_codes)))
    args.extend(overrides)
    args.append(filepath)
    system(*args)


def main():
    print("Running pre-commit hook.")
    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)
        print("Processing file:", filepath)
        autopep8(filepath)
        system("git", "add", filepath)


if __name__ == '__main__':
    main()