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