如何在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模塊識別圖片的格式，你們學到知識或技能了嗎？如果還想學到更多技能或者豐富自己的知識儲備，歡迎關注億速云行業資訊頻道。