61 lines
1.8 KiB
Python
Executable file
61 lines
1.8 KiB
Python
Executable file
#!/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()
|