imghdr模块的常用函数及其用法

imghdr是Python标准库中的一个模块，用于识别图像文件的类型。它提供了一些常用函数，可以用来确定给定文件的图像类型。下面是imghdr模块的常用函数及其用法，以及使用实例。

1. imghdr.what(filename)

该函数可以根据文件名来识别图像类型。它接受一个文件名作为参数，并返回该文件的图像类型，如果无法识别类型，则返回None。

使用示例：

import imghdr

filename = 'example.jpg'
image_type = imghdr.what(filename)
print(image_type)

2. imghdr.whatfile(fp)

该函数可以根据文件对象来识别图像类型。它接受一个文件对象作为参数，并返回该文件的图像类型，如果无法识别类型，则返回None。

使用示例：

import imghdr

with open('example.jpg', 'rb') as fp:
    image_type = imghdr.whatfile(fp)
    print(image_type)

3. imghdr.tests

该属性是一个包含了所有已知图像类型的列表。

使用示例：

import imghdr

print(imghdr.tests)

4. imghdr.test(filename, imgtype)

该函数可以基于给定的图像类型测试给定文件名的图像类型是否正确。它接受一个文件名和一个图像类型作为参数，并返回True或False。

使用示例：

import imghdr

filename = 'example.jpg'
image_type = 'jpeg'
result = imghdr.test(filename, image_type)
print(result)

5. imghdr.what_bytes(data)

该函数可以根据文件数据来识别图像类型。它接受一个包含图像数据的字节数组作为参数，并返回图像类型，如果无法识别类型，则返回None。

使用示例：

import imghdr

with open('example.jpg', 'rb') as fp:
    data = fp.read()
    image_type = imghdr.what_bytes(data)
    print(image_type)

综上所述，imghdr模块提供了一些实用的函数，可以用来确定给定文件的图像类型。你可以使用它来识别图像文件的类型，并根据需要采取相应的操作。