python PIL图像如何将图像保存到缓冲区以便以后使用?

时间:2015-01-23 03:59:02

标签: python io gridfs stringio

我有一个png文件应转换为jpg并保存到gridfs,我使用python的PIL lib加载文件并执行转换作业,问题是我想将转换后的图像存储到MongoDB Gridfs,在保存过程中,我不能只使用im.save()方法。所以我使用StringIO来保存临时文件,但它不起作用。

这是代码段:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from PIL import Image
from pymongo import MongoClient
import gridfs
from StringIO import StringIO


im = Image.open("test.png").convert("RGB")

#this is what I tried, define a 
#fake_file with StringIO that stored the image temporarily.
fake_file = StringIO()
im.save(fake_file,"jpeg")

fs = gridfs.GridFS(MongoClient("localhost").stroage)

with fs.new_file(filename="test.png") as fp:
    # this will not work
    fp.write(fake_file.read())


# vim:ai:et:sts=4:sw=4:

我在python的IO机制中非常青翠,如何使这个工作?

1 个答案:

答案 0 :(得分:4)

使用getvalue method代替read

with fs.new_file(filename="test.png") as fp:
    fp.write(fake_file.getvalue())

或者,如果您首先read从StringIO的开头读取,则可以使用seek(0)

with fs.new_file(filename="test.png") as fp:
    fake_file.seek(0)
    fp.write(fake_file.read())