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

imghdr是Python中的内置模块，用于识别图片文件的格式。它可以根据文件的字节流判断图片文件的格式，并返回对应的文件扩展名。

imghdr模块提供了一个函数imghdr.what(file, h=None)，该函数接受一个文件名或文件对象作为参数，返回文件的格式。如果无法识别或不是图片文件，则返回None。可以通过可选的参数h来指定文件的字节流。

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

import imghdr

def get_image_format(file_path):
    image_format = imghdr.what(file_path)
    if image_format:
        print(f"The image format of {file_path} is {image_format}")
    else:
        print(f"Cannot determine the image format of {file_path}")

# 识别一个JPEG格式的图片
get_image_format("image.jpg")

# 识别一个PNG格式的图片
get_image_format("image.png")

# 识别一个GIF格式的图片
get_image_format("image.gif")

# 识别一个BMP格式的图片
get_image_format("image.bmp")

# 识别一个不存在的文件
get_image_format("nonexistent.jpg")

# 使用字节流识别图片格式
file = open("image.jpg", "rb")
image_format = imghdr.what(None, file.read())
if image_format:
    print(f"The image format is {image_format}")
else:
    print("Cannot determine the image format")
file.close()

上述代码中，我们定义了一个函数get_image_format，该函数接受一个文件路径作为参数，使用imghdr.what来识别文件的图片格式。如果能够识别，则打印出文件的格式，否则打印无法确定格式。

此外，我们还演示了如何使用字节流来识别图片格式。首先，我们以只读模式打开文件，并使用rb模式指定二进制模式。然后，我们通过file.read()读取文件的字节流，并传递给imghdr.what进行格式识别。

总结一下，imghdr模块提供了简单的图片格式识别功能，可以通过文件路径或字节流来判断图片文件的格式。这对于处理图片文件的应用程序非常有用。