Commit be1dc1ea authored by Martin Aspeli's avatar Martin Aspeli

Merge c112780 from 2.12 branch

parent e09400ae
...@@ -32,23 +32,40 @@ def _decode(text, charsets): ...@@ -32,23 +32,40 @@ def _decode(text, charsets):
pass pass
return text return text
def processInputValue(value, charsets):
"""Recursively look for values (e.g. elements of lists, tuples or dicts)
and attempt to decode.
"""
if isinstance(value, list):
return [processInputValue(v, charsets) for v in value]
elif isinstance(value, tuple):
return tuple([processInputValue(v, charsets) for v in value])
elif isinstance(value, dict):
for k, v in value.items():
value[k] = processInputValue(v, charsets)
return value
elif isinstance(value, str):
return _decode(value, charsets)
else:
return value
def processInputs(request, charsets=None): def processInputs(request, charsets=None):
"""Process the values in request.form to decode strings to unicode, using
the passed-in list of charsets. If none are passed in, look up the user's
preferred charsets. The default is to use utf-8.
"""
if charsets is None: if charsets is None:
envadapter = IUserPreferredCharsets(request) envadapter = IUserPreferredCharsets(request, None)
charsets = envadapter.getPreferredCharsets() or ['utf-8'] if envadapter is None:
charsets = ['utf-8']
else:
charsets = envadapter.getPreferredCharsets() or ['utf-8']
for name, value in request.form.items(): for name, value in request.form.items():
if not (isCGI_NAME(name) or name.startswith('HTTP_')): if not (isCGI_NAME(name) or name.startswith('HTTP_')):
if isinstance(value, str): request.form[name] = processInputValue(value, charsets)
request.form[name] = _decode(value, charsets)
elif isinstance(value, list):
request.form[name] = [ _decode(val, charsets)
for val in value
if isinstance(val, str) ]
elif isinstance(value, tuple):
request.form[name] = tuple([ _decode(val, charsets)
for val in value
if isinstance(val, str) ])
def setPageEncoding(request): def setPageEncoding(request):
"""Set the encoding of the form page via the Content-Type header. """Set the encoding of the form page via the Content-Type header.
......
...@@ -46,6 +46,42 @@ def test_processInputs(): ...@@ -46,6 +46,42 @@ def test_processInputs():
>>> processInputs(request, charsets) >>> processInputs(request, charsets)
>>> request.form['foo'] == (u'f\xf6\xf6',) >>> request.form['foo'] == (u'f\xf6\xf6',)
True True
Ints in lists are not lost::
>>> request.form['foo'] = [1, 2, 3]
>>> processInputs(request, charsets)
>>> request.form['foo'] == [1, 2, 3]
True
Ints in tuples are not lost::
>>> request.form['foo'] = (1, 2, 3,)
>>> processInputs(request, charsets)
>>> request.form['foo'] == (1, 2, 3)
True
Mixed lists work:
>>> request.form['foo'] = [u'f\xf6\xf6'.encode('iso-8859-1'), 2, 3]
>>> processInputs(request, charsets)
>>> request.form['foo'] == [u'f\xf6\xf6', 2, 3]
True
Mixed dicts work:
>>> request.form['foo'] = {'foo': u'f\xf6\xf6'.encode('iso-8859-1'), 'bar': 2}
>>> processInputs(request, charsets)
>>> request.form['foo'] == {'foo': u'f\xf6\xf6', 'bar': 2}
True
Deep recursion works:
>>> request.form['foo'] = [{'foo': u'f\xf6\xf6'.encode('iso-8859-1'), 'bar': 2}, {'foo': u"one", 'bar': 3}]
>>> processInputs(request, charsets)
>>> request.form['foo'] == [{'foo': u'f\xf6\xf6', 'bar': 2}, {'foo': u"one", 'bar': 3}]
True
""" """
def test_suite(): def test_suite():
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment