pre-commit/flake8
1
0
Fork 0
mirror of https://github.com/PyCQA/flake8.git synced 2026-04-04 20:26:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fixed the stdin, thanks to zeeg

Browse source
This commit is contained in:
Tarek Ziade 2012-07-27 11:52:37 +02:00
parent c41011901f
commit aaa14dc4e1
3 changed files with 22 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

17
flake8/run.py
View file

@ -6,6 +6,11 @@ import sys
import os
import os.path
from subprocess import PIPE, Popen
import select
try:
from StringIO import StringIO # NOQA
except ImportError:
from io import StringIO # NOQA
from flake8.util import skip_file
from flake8 import pep8
@ -22,7 +27,8 @@ def check_file(path, complexity=-1):
def check_code(code, complexity=-1):
warnings = pyflakes.check(code, '<stdin>')
warnings = pyflakes.check(code, 'stdin')
warnings += pep8.input_file(StringIO(code))
if complexity > -1:
warnings += mccabe.get_code_complexity(code, complexity)
return warnings
@ -53,10 +59,17 @@ def main():
if builtins:
orig_builtins = set(pyflakes._MAGIC_GLOBALS)
pyflakes._MAGIC_GLOBALS = orig_builtins | builtins
if args:
if args and options.filename is not None:
for path in _get_python_files(args):
warnings += check_file(path, complexity)
else:
# wait for 1 second on the stdin fd
reads, __, __ = select.select([sys.stdin], [], [], 1.)
if reads == []:
print('input not specified')
raise SystemExit(1)
stdin = sys.stdin.read()
warnings += check_code(stdin, complexity)