imghdr模塊

功能描述：imghdr模塊用于識別圖片的格式。它通過檢測文件的前幾個字節(jié)，從而判斷圖片的格式。

唯一一個API

imghdr.what(file, h=None)

第一個參數(shù)file可以是用rb模式打開的file對象或者表示路徑的字符串和PathLike對象。h參數(shù)是一段字節(jié)串。函數(shù)返回表示圖片格式的字符串。

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

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

模塊內(nèi)部缺陷

當h參數(shù)不為空時，模塊會忽略掉file參數(shù)，直接檢測h參數(shù)，但此時file參數(shù)又是必須提供的，算是一個設計缺陷吧。博主個人感覺這個h參數(shù)根本沒有存在的意義，沒必要放在參數(shù)列表里面。

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

自定義檢測流程

imghdr內(nèi)部使用了test_jpeg、test_png、test_gif等函數(shù)檢測文件的格式。模塊內(nèi)部維護了一個函數(shù)列表imghdr.tests，每次調(diào)用what函數(shù)的時候，會按列表里的順序調(diào)用檢測函數(shù)，當檢測函數(shù)返回結(jié)果時退出循環(huán)。用戶可以通過修改這個列表達到修改檢測流程的目的。同時，也可以自行添加檢測函數(shù)到列表里面。

下面這個例子，博主添加了一個函數(shù)放在檢測流程的最后提示文件不是圖片：

>>> 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!

自行添加檢測函數(shù)需要接收兩個參數(shù)h和f，h是用來檢測的字節(jié)串，f是file對象。但是在模塊內(nèi)部自帶的檢測函數(shù)都是沒用到這個f參數(shù)的…

命令行啟動imghdr模塊

博主讀源碼的時候發(fā)現(xiàn)了官方文檔里面沒提到的兩個函數(shù)，提供了直接用命令行啟動imghdr模塊的方法。

只需要調(diào)用python -m imghdr [-r] file1 file2...就可以直接檢測文件的類型。file可以是文件也可以是文件夾。該命令默認只檢測文件夾下一層的文件類型，如果需要遞歸檢測，則加上-r參數(shù)。

每個文件以”文件名: 文件類型/None”的形式打印一行輸出結(jié)果。

總結(jié)：可能是因為使用的人少的原因，imghdr這個模塊內(nèi)部有挺多小問題，不過只要老老實實按照官方提供的文檔使用是不會出事的。博主后面試著改一下這個模塊然后pr上去。

以上就是本文關于Python用imghdr模塊識別圖片格式實例解析的全部內(nèi)容，希望對大家有所幫助。感興趣的朋友可以繼續(xù)參閱本站其他相關專題，如有不足之處，歡迎留言指出。感謝朋友們對本站的支持！

最新評論