1#!/usr/bin/env python 2 3import glob 4import os 5import sys 6 7os.environ['PYFLAKES_NODOCTEST'] = '1' 8 9# pep8.py uses sys.argv to find setup.cfg 10sys.argv = [os.path.join(os.path.dirname(__file__), os.pardir, os.pardir)] 11 12# git usurbs your bin path for hooks and will always run system python 13if 'VIRTUAL_ENV' in os.environ: 14 site_packages = glob.glob( 15 '%s/lib/*/site-packages' % os.environ['VIRTUAL_ENV'])[0] 16 sys.path.insert(0, site_packages) 17 18 19def py_lint(files_modified): 20 from flake8.main import DEFAULT_CONFIG 21 from flake8.engine import get_style_guide 22 23 # remove non-py files and files which no longer exist 24 files_modified = filter(lambda x: x.endswith('.py'), files_modified) 25 26 if not files_modified: 27 return False 28 29 flake8_style = get_style_guide(config_file=DEFAULT_CONFIG) 30 report = flake8_style.check_files(files_modified) 31 32 return report.total_errors != 0 33 34 35def main(): 36 from flake8.hooks import run 37 38 gitcmd = "git diff-index --cached --name-only HEAD" 39 40 _, files_modified, _ = run(gitcmd) 41 42 files_modified = filter(lambda x: os.path.exists(x), files_modified) 43 44 if py_lint(files_modified): 45 return 1 46 return 0 47 48if __name__ == '__main__': 49 sys.exit(main()) 50