Merge branch 'match_newlines_py3' into 'master'

Fix inconsistent newlines read from a file in python3

Closes #457

See merge request pycqa/flake8!253
This commit is contained in:
Anthony Sottile 2018-10-23 21:44:22 +00:00
commit 63b91c95ea
2 changed files with 29 additions and 6 deletions

View file

@ -1,6 +1,5 @@
"""Module containing our file processor that tokenizes a file for checks."""
import contextlib
import io
import logging
import sys
import tokenize
@ -308,11 +307,9 @@ class FileProcessor(object):
def _readlines_py3(self):
# type: () -> List[str]
try:
with open(self.filename, "rb") as fd:
(coding, lines) = tokenize.detect_encoding(fd.readline)
textfd = io.TextIOWrapper(fd, coding, line_buffering=True)
return [l.decode(coding) for l in lines] + textfd.readlines()
except (LookupError, SyntaxError, UnicodeError):
with tokenize.open(self.filename) as fd:
return fd.readlines()
except (SyntaxError, UnicodeError):
# If we can't detect the codec with tokenize.detect_encoding, or
# the detected encoding is incorrect, just fallback to latin-1.
with open(self.filename, encoding="latin-1") as fd:

View file

@ -27,6 +27,32 @@ def test_read_lines_splits_lines():
for line in lines)
def lines_from_file(tmpdir, lines):
f = tmpdir.join('f.py')
f.write(''.join(lines))
return processor.FileProcessor(f.strpath, options_from()).lines
def test_read_lines_universal_newlines(tmpdir):
r"""Verify that line endings are translated to \n."""
lines = lines_from_file(tmpdir, ['# coding: utf-8\r\n', 'x = 1\r\n'])
assert lines == ['# coding: utf-8\n', 'x = 1\n']
def test_read_lines_incorrect_utf_16(tmpdir):
"""Verify that a file which incorrectly claims it is utf16 is still read
as latin-1.
"""
lines = lines_from_file(tmpdir, ['# coding: utf16\n', 'x = 1\n'])
assert lines == ['# coding: utf16\n', 'x = 1\n']
def test_read_lines_unknown_encoding(tmpdir):
"""Verify that an unknown encoding is still read as latin-1."""
lines = lines_from_file(tmpdir, ['# coding: fake-encoding\n', 'x = 1\n'])
assert lines == ['# coding: fake-encoding\n', 'x = 1\n']
@pytest.mark.parametrize('first_line', [
'\xEF\xBB\xBF"""Module docstring."""\n',
u'\uFEFF"""Module docstring."""\n',