Python中的imghdr模块：识别图片格式的利器

imghdr模块是Python标准库中的一个模块，用于识别图片文件的格式。它可以根据文件的内容来判断图片的格式，从而避免了依赖文件后缀名来确定图片格式的问题。

imghdr模块提供了一个函数imghdr.what(filename, h=None)来判断指定文件的图片格式，其中filename是要判断的文件名，h是一个可选参数，用于传递一个文件的头部信息。这个函数会返回一个表示图片格式的字符串，或者返回None表示无法识别。

下面是一个使用imghdr模块的例子：

import imghdr

def get_image_format(filename):
    format = imghdr.what(filename)
    if format is None:
        print("无法识别的图片格式")
    else:
        print("图片格式为：", format)

# 测试
get_image_format("image.jpg")

在这个例子中，我们定义了一个函数get_image_format，该函数接受一个文件名作为参数，并使用imghdr.what来判断图片的格式。如果返回的格式不是None，则打印出格式名称，否则打印出错误消息表示无法识别的图片格式。

另外，imghdr模块还提供了一个函数imghdr.tests函数，可以用于判断图片的格式。这个函数接受一个文件名作为参数，返回一个包含可能的格式的列表。例如：

import imghdr

def test_image_formats(filename):
    formats = imghdr.tests(filename)
    print("可能的图片格式：", formats)

# 测试
test_image_formats("image.jpg")

这个例子中，我们使用imghdr.tests函数来尝试识别指定文件可能的图片格式，并打印出这些格式。

总结：imghdr模块是用于识别图片格式的一个很实用的工具，在处理图片文件时可以用来判断文件的格式从而进行相应的处理。通过这个模块，我们可以更加灵活和准确地判断图片的格式，而不是依赖于文件后缀名或其他方式来判断。