mirror of
https://github.com/python/cpython.git
synced 2025-08-03 00:23:06 +00:00
Merge #21476: Unwrap fp in BytesParser so the file isn't unexpectedly closed.
This commit is contained in:
commit
0c43ed8716
3 changed files with 31 additions and 1 deletions
|
@ -106,8 +106,10 @@ class BytesParser:
|
|||
meaning it parses the entire contents of the file.
|
||||
"""
|
||||
fp = TextIOWrapper(fp, encoding='ascii', errors='surrogateescape')
|
||||
with fp:
|
||||
try:
|
||||
return self.parser.parse(fp, headersonly)
|
||||
finally:
|
||||
fp.detach()
|
||||
|
||||
|
||||
def parsebytes(self, text, headersonly=False):
|
||||
|
|
|
@ -3390,6 +3390,31 @@ class TestParsers(TestEmailBase):
|
|||
self.assertIsInstance(msg.get_payload(), str)
|
||||
self.assertIsInstance(msg.get_payload(decode=True), bytes)
|
||||
|
||||
def test_bytes_parser_does_not_close_file(self):
|
||||
with openfile('msg_02.txt', 'rb') as fp:
|
||||
email.parser.BytesParser().parse(fp)
|
||||
self.assertFalse(fp.closed)
|
||||
|
||||
def test_bytes_parser_on_exception_does_not_close_file(self):
|
||||
with openfile('msg_15.txt', 'rb') as fp:
|
||||
bytesParser = email.parser.BytesParser
|
||||
self.assertRaises(email.errors.StartBoundaryNotFoundDefect,
|
||||
bytesParser(policy=email.policy.strict).parse,
|
||||
fp)
|
||||
self.assertFalse(fp.closed)
|
||||
|
||||
def test_parser_does_not_close_file(self):
|
||||
with openfile('msg_02.txt', 'r') as fp:
|
||||
email.parser.Parser().parse(fp)
|
||||
self.assertFalse(fp.closed)
|
||||
|
||||
def test_parser_on_exception_does_not_close_file(self):
|
||||
with openfile('msg_15.txt', 'r') as fp:
|
||||
parser = email.parser.Parser
|
||||
self.assertRaises(email.errors.StartBoundaryNotFoundDefect,
|
||||
parser(policy=email.policy.strict).parse, fp)
|
||||
self.assertFalse(fp.closed)
|
||||
|
||||
def test_whitespace_continuation(self):
|
||||
eq = self.assertEqual
|
||||
# This message contains a line after the Subject: header that has only
|
||||
|
|
|
@ -103,6 +103,9 @@ Core and Builtins
|
|||
Library
|
||||
-------
|
||||
|
||||
- Issue #21476: Make sure the email.parser.BytesParser TextIOWrapper is
|
||||
discarded after parsing, so the input file isn't unexpectedly closed.
|
||||
|
||||
- Issue #20295: imghdr now recognizes OpenEXR format images.
|
||||
|
||||
- Issue #21729: Used the "with" statement in the dbm.dumb module to ensure
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue