存储Python PEP8模块输出

时间:2016-03-30 10:58:43

标签: python pep8

我在我的代码中使用了python的PEP8模块。

import pep8

pep8_checker = pep8.StyleGuide(format='pylint')

pep8_checker.check_files(paths=['./test.py'])
r = pep8_checker.check_files(paths=['./test.py'])

这是输出:

./test.py:6: [E265] block comment should start with '# '
./test.py:23: [E265] block comment should start with '# '
./test.py:24: [E302] expected 2 blank lines, found 1
./test.py:30: [W293] blank line contains whitespace
./test.py:35: [E501] line too long (116 > 79 characters)
./test.py:41: [E302] expected 2 blank lines, found 1
./test.py:53: [E501] line too long (111 > 79 characters)
./test.py:54: [E501] line too long (129 > 79 characters)

但是这个结果会打印在终端上,而最终的值会被分配给' r'是8(即错误的总数)。

我想将这些错误存储在变量中。我怎么能这样做?

编辑: 这是test.py文件:http://paste.fedoraproject.org/347406/59337502/raw/

1 个答案:

答案 0 :(得分:1)

至少有两种方法可以做到这一点。最简单的方法是将sys.stdout重定向到文本文件,然后在闲暇时阅读该文件:

import pep8
import sys

saved_stdout = sys.stdout
sys.stdout = open('pep8.out', 'w')

pep8_checker = pep8.StyleGuide(format='pylint')
pep8_checker.check_files(paths=['./test.py'])
r = pep8_checker.check_files(paths=['./test.py'])

sys.stdout.close()
sys.stdout = saved_stdout

# Now you can read "pep.out" into a variable

或者,您可以使用StringIO

写入变量
import pep8
import sys

# The module name changed between python 2 and 3
if sys.version_info.major == 2:
    from StringIO import StringIO
else:
    from io import StringIO

saved_stdout = sys.stdout
sys.stdout  = StringIO()

pep8_checker = pep8.StyleGuide(format='pylint')

pep8_checker.check_files(paths=['./test.py'])
r = pep8_checker.check_files(paths=['./test.py'])

testout = sys.stdout.getvalue()
sys.stdout.close()
sys.stdout = saved_stdout

# testout contains the output.  You might wish to testout.spilt("\n")