46 lines
1.2 KiB
Python
Executable File
46 lines
1.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import glob
|
|
import os
|
|
import sys
|
|
|
|
os.environ['PYFLAKES_NODOCTEST'] = '1'
|
|
|
|
# pep8.py uses sys.argv to find setup.cfg
|
|
sys.argv = [os.path.join(os.path.dirname(__file__), os.pardir, os.pardir)]
|
|
|
|
# git usurbs your bin path for hooks and will always run system python
|
|
if 'VIRTUAL_ENV' in os.environ:
|
|
site_packages = glob.glob(
|
|
'%s/lib/*/site-packages' % os.environ['VIRTUAL_ENV'])[0]
|
|
sys.path.insert(0, site_packages)
|
|
|
|
|
|
def py_lint(files_modified):
|
|
from flake8.main import DEFAULT_CONFIG
|
|
from flake8.engine import get_style_guide
|
|
|
|
# remove non-py files and files which no longer exist
|
|
files_modified = filter(lambda x: x.endswith('.py'), files_modified)
|
|
|
|
flake8_style = get_style_guide(parse_argv=True, config_file=DEFAULT_CONFIG)
|
|
report = flake8_style.check_files(files_modified)
|
|
|
|
return report.total_errors != 0
|
|
|
|
|
|
def main():
|
|
from flake8.hooks import run
|
|
|
|
gitcmd = "git diff-index --cached --name-only HEAD"
|
|
|
|
_, files_modified, _ = run(gitcmd)
|
|
|
|
files_modified = filter(lambda x: os.path.exists(x), files_modified)
|
|
|
|
if py_lint(files_modified):
|
|
return 1
|
|
return 0
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main()) |