Python：检查上传的文件是否为jpg

Question

如何检查用户上传的文件在Python（Google App Engine）中是否为真实的jpg文件？

这是我现在所走的距离：

脚本通过HTML Form Post接收图像，并由以下代码处理

...
incomming_image = self.request.get("img")
image = db.Blob(incomming_image)
...

我找到了mimetypes.guess_type，但是它对我不起作用。

Answer 1

如果您需要的不仅仅是扩展名，一种方法是读取JPEG标头，并检查其是否与有效数据匹配。格式为：

Start Marker  | JFIF Marker | Header Length | Identifier
0xff, 0xd8    | 0xff, 0xe0  |    2-bytes    | "JFIF\0"

因此快速识别器将是：

def is_jpg(filename):
    data = open(filename,'rb').read(11)
    if data[:4] != '\xff\xd8\xff\xe0': return False
    if data[6:] != 'JFIF\0': return False
    return True

但是，这不会在体内捕获任何不良数据。如果您想要更强大的检查，可以尝试使用PIL加载它。例如：

from PIL import Image
def is_jpg(filename):
    try:
        i=Image.open(filename)
        return i.format =='JPEG'
    except IOError:
        return False