Метод PIL open() не работает с BytesIO

По какой-то причине, когда я пытаюсь сделать изображение с байтового потока 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>

Я использую Pillow для реализации PIL.

Ответы

Ответ 1

Подумайте о BytesIO как о файле, после того как вы закончите запись изображения, курсор находится в конце файла, поэтому, когда Image.open() пытается вызвать output.read(), он сразу получает EOF.

Вам нужно добавить output.seek(0) перед тем, как передать output в Image.open().