python 判断文件是图片

原创

mob64ca12e7f20c 2024-07-25 11:42:10 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12e7f20c的原创作品，请联系作者获取转载授权，否则将追究法律责任

如何用Python判断文件是否为图片

作为一名刚入行的开发者，你可能会遇到需要判断文件类型的问题，尤其是在处理图片文件时。本文将教会你如何使用Python来判断一个文件是否是图片。

流程图

首先，我们通过一个流程图来展示整个判断过程：

flowchart TD
    A[开始] --> B{判断文件扩展名}
    B -- 是图片扩展名 --> C[是图片]
    B -- 不是图片扩展名 --> D[尝试读取文件内容]
    D --> E{判断文件内容}
    E -- 是图片格式 --> C
    E -- 不是图片格式 --> F[不是图片]

步骤详解

步骤1：判断文件扩展名

最简单的方法是检查文件的扩展名。常见的图片格式有.jpg, .jpeg, .png, .gif, .bmp等。

import os

def is_image_by_extension(file_path):
    # 获取文件扩展名
    _, file_extension = os.path.splitext(file_path)
    # 定义图片文件的扩展名列表
    image_extensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp']
    # 判断扩展名是否在图片扩展名列表中
    return file_extension.lower() in image_extensions

步骤2：尝试读取文件内容

如果文件扩展名不是图片格式，或者你想要更准确地判断，可以尝试读取文件内容，检查是否符合图片格式的特征。

def is_image_by_content(file_path):
    try:
        with open(file_path, 'rb') as file:
            # 读取文件的前几个字节
            file_content = file.read(24)
            # 检查是否是常见的图片格式
            if file_content.startswith(b'\xff\xd8'):
                return True  # JPEG
            elif file_content.startswith(b'\x89PNG\r\n\x1a\n'):
                return True  # PNG
            elif file_content.startswith(b'GIF87a') or file_content.startswith(b'GIF89a'):
                return True  # GIF
            elif file_content.startswith(b'BM'):
                return True  # BMP
            else:
                return False
    except Exception as e:
        print(f"Error reading file: {e}")
        return False

步骤3：综合判断

结合以上两个步骤，我们可以编写一个函数来综合判断文件是否为图片。

def is_image(file_path):
    # 首先尝试通过扩展名判断
    if is_image_by_extension(file_path):
        return True
    # 如果扩展名不是图片格式，尝试通过文件内容判断
    return is_image_by_content(file_path)

示例

现在，我们来测试一下这个函数。

# 测试文件路径
test_file_path = 'example.jpg'

# 判断文件是否为图片
if is_image(test_file_path):
    print(f"{test_file_path} 是图片")
else:
    print(f"{test_file_path} 不是图片")