python中的一个实用的库imghdr,用于探测图片格式

之前没有注意到，图片格式竟然有这么多种；

‘rgb’ SGI 图像库文件
‘gif’ GIF 87a 和 89a 文件
‘pbm’ 便携式位图文件
‘pgm’ 便携式灰度图文件
‘ppm’ 便携式像素表文件
‘tiff’ TIFF 文件
‘rast’ Sun 光栅文件
‘xbm’ X 位图文件
‘jpeg’ JFIF 或 Exif 格式的 JPEG 数据
‘bmp’ BMP 文件
‘png’ 便携式网络图像
webp’ WebP 文件
‘exr’ OpenEXR 文件

其中最常见的是png和jpeg了，通常都是通过文件名后缀来判断图片格式，但是都知道，这个后缀可以随意更改，不太准；

python内置了一个检测图片格式的库：

imghdr

imghdr是一个用来检测图片类型的模块,提供了一个api叫做

imghdr.what

传递给它的可以是一个文件对象,也可以是一个字节流; 然后防护一个图片格式；

参考imghdr的源码发现，其对图片做了一次遍历检测,看最终匹配到哪个结果；

tests = []

def test_jpeg(h, f):
    """JPEG data in JFIF or Exif format"""
    if h[6:10] in (b'JFIF', b'Exif'):
        return 'jpeg'

tests.append(test_jpeg)

def test_png(h, f):
1
2
3
4
5
6
7
8
9
10