summaryrefslogtreecommitdiff
path: root/cloudinit/handlers/__init__.py
diff options
context:
space:
mode:
authorBarry Warsaw <barry@python.org>2015-01-27 15:11:53 -0500
committerBarry Warsaw <barry@python.org>2015-01-27 15:11:53 -0500
commit96d130e7732f1242d71c65a32412ae56cb229abf (patch)
tree3fb39ac6dd43988f36507a1c0b82e8944dfe95ff /cloudinit/handlers/__init__.py
parent6e742d20e9ed56498925c7c850cd5da65d063b4b (diff)
downloadvyos-cloud-init-96d130e7732f1242d71c65a32412ae56cb229abf.tar.gz
vyos-cloud-init-96d130e7732f1242d71c65a32412ae56cb229abf.zip
Respond to review:
- Refactor "fully" decoding the payload of a text/* part. In Python 3, decode=True only means to decode according to Content-Transfer-Encoding, not according to any charset in the Content-Type header. So do that.
Diffstat (limited to 'cloudinit/handlers/__init__.py')
-rw-r--r--cloudinit/handlers/__init__.py11
1 files changed, 1 insertions, 10 deletions
diff --git a/cloudinit/handlers/__init__.py b/cloudinit/handlers/__init__.py
index cdccf122..6b7abbcd 100644
--- a/cloudinit/handlers/__init__.py
+++ b/cloudinit/handlers/__init__.py
@@ -233,16 +233,7 @@ def walk(msg, callback, data):
headers = dict(part)
LOG.debug(headers)
headers['Content-Type'] = ctype
- payload = part.get_payload(decode=True)
- # In Python 3, decoding the payload will ironically hand us a bytes
- # object. 'decode' means to decode according to
- # Content-Transfer-Encoding, not according to any charset in the
- # Content-Type. So, if we end up with bytes, first try to decode to
- # str via CT charset, and failing that, try utf-8 using surrogate
- # escapes.
- if six.PY3 and isinstance(payload, bytes):
- charset = part.get_charset() or 'utf-8'
- payload = payload.decode(charset, errors='surrogateescape')
+ payload = util.fully_decoded_payload(part)
callback(data, filename, payload, headers)
partnum = partnum + 1