PIL的open()方法在BytesIO中无效

Question

出于某种原因，当我尝试从一个BytesIO流中创建图片时，它无法识别这个图片。以下是我的代码：

from PIL import Image, ImageGrab
from io import BytesIO

i = ImageGrab.grab()
i.resize((1280, 720))
output = BytesIO()
i.save(output, format = "JPEG")
output.flush()
print(isinstance(Image.open(output), Image.Image))

还有它抛出的错误的堆栈跟踪信息：

Traceback (most recent call last):
  File "C:/Users/Natecat/PycharmProjects/Python/test.py", line 9, in <module>
    print(isinstance(Image.open(output), Image.Image))
  File "C:\Python27\lib\site-packages\PIL\Image.py", line 2126, in open
    % (filename if filename else fp))
IOError: cannot identify image file <_io.BytesIO object at 0x02394DB0>

我正在使用PIL的Pillow实现。

Answer 1

把BytesIO想象成一个文件对象。当你写完图片后，文件的光标会停在文件的最后面。所以当你用Image.open()去调用output.read()的时候，它会立刻遇到文件结束的情况。

在把output传给Image.open()之前，你需要先加一个output.seek(0)，这样光标就会回到文件的开头。