python - mimetools.Message() 到 python 3 email.message.Message

Question

我尝试将 python 2.x 代码移植到 python 3。我正在努力解决的问题是

from mimetools import Message
...
headers = Message(StringIO(data.split('\r\n', 1)[1]))

我发现 python 3 中不再存在 mimetools，替代品是 email 类。我尝试像这样替换它:

headers = email.message_from_file(io.StringIO(data.split('\r\n', 1)[1]))

但是我得到了这个错误:

    headers = email.message_from_file(io.StringIO(data.split('\r\n', 1)[1]))
TypeError: Type str doesn't support the buffer API

我正在寻找一个提示来正确地从 mimetools 移植到电子邮件。原始代码不是我写的。在这里能找到它 : https://gist.github.com/jkp/3136208

Answer 1

亚历克斯自己的解决方案来自他的评论:

import email
stream = io.StringIO() 
rxString = data.decode("utf-8").split('\r\n', 1)[1]
stream.write(rxString) 
headers = email.message_from_string(rxString)