python - 将 PIL 图像转换为 MIMEImage

Question

我想使用 PIL 创建图像并能够通过电子邮件发送它而无需将其保存到磁盘。

这是可行的，但涉及保存到磁盘:

from email.mime.image import MIMEImage
from email.mime.multipart import MIMEMultipart

msg = MIMEMultipart()

im = Image.new("RGB", (200, 200))

with open("tempimg.jpg", "w") as f:
    im.save(f, "JPEG")

with open("tempimg.jpg", 'rb') as f:
    img = MIMEImage(f.read())

msg.attach(img)

现在我希望能够做类似的事情:

import StringIO

tempimg = StringIO.StringIO()
tempimg.write(im.tostring())
img = MIMEImage(tempimage.getvalue(), "JPG")
msg.attach(img)

，这是行不通的。我找到了一些 discussion in Spanish看起来它解决了同样的问题，除了指向 StringIO 的指针之外没有解决方案。

Answer 1

im.tostring 返回原始图像数据，但您需要将整个图像文件数据传递给 MIMEImage，因此请使用 StringIO将图像保存到内存并使用该数据的模块:

from email.mime.image import MIMEImage
from email.mime.multipart import MIMEMultipart
from PIL import Image
import cStringIO

msg = MIMEMultipart()

im = Image.new("RGB", (200, 200))
memf = cStringIO.StringIO()
im.save(memf, "JPEG")
img = MIMEImage(memf.getvalue())

msg.attach(img)