mirror of
https://github.com/PyCQA/flake8.git
synced 2026-04-03 19:56:54 +00:00
145 lines
3.7 KiB
Python
145 lines
3.7 KiB
Python
from __future__ import with_statement
|
|
import os
|
|
import sys
|
|
from subprocess import Popen, PIPE
|
|
try:
|
|
from configparser import ConfigParser
|
|
except ImportError: # Python 2
|
|
from ConfigParser import ConfigParser
|
|
|
|
from flake8.engine import get_parser, get_style_guide
|
|
from flake8.main import DEFAULT_CONFIG
|
|
from flake8.util import skip_file
|
|
|
|
|
|
def git_hook(complexity=-1, strict=False, ignore=None, lazy=False):
|
|
gitcmd = "git diff-index --cached --name-only HEAD"
|
|
if lazy:
|
|
gitcmd = gitcmd.replace('--cached ', '')
|
|
|
|
_, files_modified, _ = run(gitcmd)
|
|
|
|
flake8_style = get_style_guide(
|
|
config_file=DEFAULT_CONFIG,
|
|
ignore=ignore, max_complexity=complexity)
|
|
report = flake8_style.check_files(files_modified)
|
|
|
|
if strict:
|
|
return report.get_file_results()
|
|
|
|
return 0
|
|
|
|
|
|
def hg_hook(ui, repo, **kwargs):
|
|
complexity = ui.config('flake8', 'complexity', default=-1)
|
|
config = ui.config('flake8', 'config', default=True)
|
|
if config is True:
|
|
config = DEFAULT_CONFIG
|
|
|
|
flake8_style = get_style_guide(
|
|
config_file=config,
|
|
max_complexity=complexity)
|
|
|
|
paths = _get_files(repo, **kwargs)
|
|
report = flake8_style.check_files(paths)
|
|
|
|
strict = ui.configbool('flake8', 'strict', default=True)
|
|
|
|
if strict:
|
|
return report.get_file_results()
|
|
|
|
return 0
|
|
|
|
|
|
def run(command):
|
|
p = Popen(command.split(), stdout=PIPE, stderr=PIPE)
|
|
p.wait()
|
|
return (p.returncode, [line.strip() for line in p.stdout.readlines()],
|
|
[line.strip() for line in p.stderr.readlines()])
|
|
|
|
|
|
def _get_files(repo, **kwargs):
|
|
seen = set()
|
|
for rev in range(repo[kwargs['node']], len(repo)):
|
|
for file_ in repo[rev].files():
|
|
file_ = os.path.join(repo.root, file_)
|
|
if file_ in seen or not os.path.exists(file_):
|
|
continue
|
|
seen.add(file_)
|
|
if not file_.endswith('.py'):
|
|
continue
|
|
if skip_file(file_):
|
|
continue
|
|
yield file_
|
|
|
|
|
|
def find_vcs():
|
|
if os.path.isdir('.git'):
|
|
if not os.path.isdir('.git/hooks'):
|
|
os.mkdir('.git/hooks')
|
|
return '.git/hooks/pre-commit'
|
|
elif os.path.isdir('.hg'):
|
|
return '.hg/hgrc'
|
|
return ''
|
|
|
|
|
|
git_hook_file = """#!/usr/bin/env python
|
|
import sys
|
|
import os
|
|
from flake8.hooks import git_hook
|
|
|
|
COMPLEXITY = os.getenv('FLAKE8_COMPLEXITY', 10)
|
|
STRICT = os.getenv('FLAKE8_STRICT', False)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(git_hook(complexity=COMPLEXITY, strict=STRICT))
|
|
"""
|
|
|
|
|
|
def _install_hg_hook(path):
|
|
c = ConfigParser()
|
|
c.readfp(open(path, 'r'))
|
|
if not c.has_section('hooks'):
|
|
c.add_section('hooks')
|
|
|
|
if not c.has_option('hooks', 'commit'):
|
|
c.set('hooks', 'commit', 'python:flake8.hooks.hg_hook')
|
|
|
|
if not c.has_option('hooks', 'qrefresh'):
|
|
c.set('hooks', 'qrefresh', 'python:flake8.hooks.hg_hook')
|
|
|
|
if not c.has_section('flake8'):
|
|
c.add_section('flake8')
|
|
|
|
if not c.has_option('flake8', 'complexity'):
|
|
c.set('flake8', 'complexity', str(os.getenv('FLAKE8_COMPLEXITY', 10)))
|
|
|
|
if not c.has_option('flake8', 'strict'):
|
|
c.set('flake8', 'strict', os.getenv('FLAKE8_STRICT', False))
|
|
|
|
c.write(open(path, 'w+'))
|
|
|
|
|
|
def install_hook():
|
|
vcs = find_vcs()
|
|
|
|
if not vcs:
|
|
p = get_parser()
|
|
sys.stderr.write('Error: could not find either a git or mercurial '
|
|
'directory. Please re-run this in a proper '
|
|
'repository.')
|
|
p.print_help()
|
|
sys.exit(1)
|
|
|
|
status = 0
|
|
if 'git' in vcs:
|
|
with open(vcs, 'w+') as fd:
|
|
fd.write(git_hook_file)
|
|
os.chmod(vcs, 744)
|
|
elif 'hg' in vcs:
|
|
_install_hg_hook(vcs)
|
|
else:
|
|
status = 1
|
|
|
|
sys.exit(status)
|