如何在Python中使用imghdr模块识别图片的格式

本篇文章为大家展示了如何在Python中使用imghdr模块识别图片的格式，内容简明扼要并且容易理解，绝对能使你眼前一亮，通过这篇文章的详细介绍希望你能有所收获。

imghdr模块

功能描述：imghdr模块用于识别图片的格式。它通过检测文件的前几个字节，从而判断图片的格式。

唯一一个API

imghdr.what(file, h=None)

第一个参数file可以是用rb模式打开的file对象或者表示路径的字符串和PathLike对象。h参数是一段字节串。函数返回表示图片格式的字符串。

>>> import imghdr
>>> imghdr.what('test.jpg')
'jpeg'

具体的返回值和描述如下：

模块内部缺陷

当h参数不为空时，模块会忽略掉file参数，直接检测h参数，但此时file参数又是必须提供的，算是一个设计缺陷吧。博主个人感觉这个h参数根本没有存在的意义，没必要放在参数列表里面。

>>> import imghdr
>>> imghdr.what('test.jpg', b'\x89PNG\r\n\x1a\n')
'png'
>>>

自定义检测流程

imghdr内部使用了test_jpeg、test_png、test_gif等函数检测文件的格式。模块内部维护了一个函数列表imghdr.tests，每次调用what函数的时候，会按列表里的顺序调用检测函数，当检测函数返回结果时退出循环。用户可以通过修改这个列表达到修改检测流程的目的。同时，也可以自行添加检测函数到列表里面。

下面这个例子，博主添加了一个函数放在检测流程的最后提示文件不是图片：

>>> import imghdr
>>> def final(h, f):
... print("This file isn\'t a image!")
...
>>> imghdr.tests.append(final)
>>> imghdr.what("imghdr.md")
This file isn't a image!

自行添加检测函数需要接收两个参数h和f，h是用来检测的字节串，f是file对象。但是在模块内部自带的检测函数都是没用到这个f参数的…

命令行启动imghdr模块

博主读源码的时候发现了官方文档里面没提到的两个函数，提供了直接用命令行启动imghdr模块的方法。

只需要调用python -m imghdr [-r] file1 file2...就可以直接检测文件的类型。file可以是文件也可以是文件夹。该命令默认只检测文件夹下一层的文件类型，如果需要递归检测，则加上-r参数。

每个文件以”文件名: 文件类型/None”的形式打印一行输出结果。

上述内容就是如何在Python中使用imghdr模块识别图片的格式，你们学到知识或技能了吗？如果还想学到更多技能或者丰富自己的知识储备，欢迎关注亿速云行业资讯频道。