python - 确定内存中的图像

Question

我想确定我(下载的)缓冲区是否是图像文件，而不将其保存到磁盘。我查了一下，发现:

1. imghdr 只能对文件执行此操作。
2. python-magic 可以为我提供文件类型，但我需要将其映射到图像类型。

有更好的方法吗？

Answer 1

根据 the documentation ，您可以通过字节流(您应该能够从缓冲区中获取)使用 imghdr 识别文件类型

import imghdr
imghdr.what('', byte_stream)

如果您提供第二个参数，文件名将被忽略。