Python中如何使用imghdr模块检测图片格式

imghdr是Python标准库中的一个模块，用于检测文件的图片格式。

imghdr模块提供了一个函数imghdr.what()，用于获取文件的类型。函数的参数为文件的路径。

下面是一个使用imghdr模块检测图片格式的例子：

import imghdr

def check_image_format(file_path):
    # 获取文件的图片格式
    image_format = imghdr.what(file_path)
    
    if image_format is None:
        print(f"{file_path} 不是一个有效的图片文件")
    else:
        print(f"{file_path} 的格式为：{image_format}")

# 调用函数检测图片格式
check_image_format("image.jpg")

输出：

image.jpg 的格式为：jpeg

在上面的例子中，check_image_format()函数接收一个文件路径作为参数。函数使用imghdr.what()函数获取文件的图片格式，并打印出结果。

如果文件格式无效，即不是图片文件，imghdr.what()函数会返回None。

通常，我们可以将check_image_format()函数嵌入到一个循环中，遍历文件夹中的所有图片进行格式检测。

import os
import imghdr

def check_image_format(file_path):
    image_format = imghdr.what(file_path)
    
    if image_format is None:
        print(f"{file_path} 不是一个有效的图片文件")
    else:
        print(f"{file_path} 的格式为：{image_format}")

def check_image_formats_in_directory(directory):
    for file_name in os.listdir(directory):
        file_path = os.path.join(directory, file_name)
        
        if os.path.isfile(file_path):
            check_image_format(file_path)

# 检测当前文件夹中的图片格式
check_image_formats_in_directory(".")

输出：

image1.jpg 的格式为：jpeg
image2.png 的格式为：png

在上面的例子中，check_image_formats_in_directory()函数接收一个文件夹路径作为参数。函数使用os.listdir()函数遍历文件夹中的所有文件名，然后使用os.path.join()函数将文件路径与文件名拼接起来。

在循环中，我们判断文件是否是一个标准文件（而不是目录），然后调用check_image_format()函数检测图片格式。

除了上面的例子中提到的常见图片格式（JPEG、PNG），imghdr模块还可以识别更多格式，包括GIF、BMP、TIFF等。

需要注意的是，imghdr模块仅仅检查文件的前几个字节，因此可能会有一些特殊或非标准的图片格式无法准确识别。建议在使用中进行一些额外的检查和处理。