bpo-33524: Fix the folding of email header when max_line_length is 0 or None (#13391)
and there are non-ascii characters in the header.
This commit is contained in:
parent
cbe72d8426
commit
feac6cd775
|
@ -68,6 +68,7 @@ XXX: provide complete list of token types.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import re
|
import re
|
||||||
|
import sys
|
||||||
import urllib # For urllib.parse.unquote
|
import urllib # For urllib.parse.unquote
|
||||||
from string import hexdigits
|
from string import hexdigits
|
||||||
from operator import itemgetter
|
from operator import itemgetter
|
||||||
|
@ -2590,7 +2591,7 @@ def _refold_parse_tree(parse_tree, *, policy):
|
||||||
|
|
||||||
"""
|
"""
|
||||||
# max_line_length 0/None means no limit, ie: infinitely long.
|
# max_line_length 0/None means no limit, ie: infinitely long.
|
||||||
maxlen = policy.max_line_length or float("+inf")
|
maxlen = policy.max_line_length or sys.maxsize
|
||||||
encoding = 'utf-8' if policy.utf8 else 'us-ascii'
|
encoding = 'utf-8' if policy.utf8 else 'us-ascii'
|
||||||
lines = ['']
|
lines = ['']
|
||||||
last_ew = None
|
last_ew = None
|
||||||
|
|
|
@ -3,6 +3,7 @@ code that adds all the email6 features.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import re
|
import re
|
||||||
|
import sys
|
||||||
from email._policybase import Policy, Compat32, compat32, _extend_docstrings
|
from email._policybase import Policy, Compat32, compat32, _extend_docstrings
|
||||||
from email.utils import _has_surrogates
|
from email.utils import _has_surrogates
|
||||||
from email.headerregistry import HeaderRegistry as HeaderRegistry
|
from email.headerregistry import HeaderRegistry as HeaderRegistry
|
||||||
|
@ -203,7 +204,7 @@ class EmailPolicy(Policy):
|
||||||
def _fold(self, name, value, refold_binary=False):
|
def _fold(self, name, value, refold_binary=False):
|
||||||
if hasattr(value, 'name'):
|
if hasattr(value, 'name'):
|
||||||
return value.fold(policy=self)
|
return value.fold(policy=self)
|
||||||
maxlen = self.max_line_length if self.max_line_length else float('inf')
|
maxlen = self.max_line_length if self.max_line_length else sys.maxsize
|
||||||
lines = value.splitlines()
|
lines = value.splitlines()
|
||||||
refold = (self.refold_source == 'all' or
|
refold = (self.refold_source == 'all' or
|
||||||
self.refold_source == 'long' and
|
self.refold_source == 'long' and
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
import io
|
import io
|
||||||
|
import sys
|
||||||
import types
|
import types
|
||||||
import textwrap
|
import textwrap
|
||||||
import unittest
|
import unittest
|
||||||
|
@ -134,6 +135,18 @@ class PolicyAPITests(unittest.TestCase):
|
||||||
for attr, value in expected.items():
|
for attr, value in expected.items():
|
||||||
self.assertEqual(getattr(added, attr), value)
|
self.assertEqual(getattr(added, attr), value)
|
||||||
|
|
||||||
|
def test_fold_zero_max_line_length(self):
|
||||||
|
expected = 'Subject: =?utf-8?q?=C3=A1?=\n'
|
||||||
|
|
||||||
|
msg = email.message.EmailMessage()
|
||||||
|
msg['Subject'] = 'á'
|
||||||
|
|
||||||
|
p1 = email.policy.default.clone(max_line_length=0)
|
||||||
|
p2 = email.policy.default.clone(max_line_length=None)
|
||||||
|
|
||||||
|
self.assertEqual(p1.fold('Subject', msg['Subject']), expected)
|
||||||
|
self.assertEqual(p2.fold('Subject', msg['Subject']), expected)
|
||||||
|
|
||||||
def test_register_defect(self):
|
def test_register_defect(self):
|
||||||
class Dummy:
|
class Dummy:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
|
|
|
@ -0,0 +1,3 @@
|
||||||
|
Fix the folding of email header when the max_line_length is 0 or None and the
|
||||||
|
header contains non-ascii characters. Contributed by Licht Takeuchi
|
||||||
|
(@Licht-T).
|
Loading…
Reference in New Issue