Python项目中如何验证输入为图片路径并拦截非图片输入？

以下是几种可靠的验证方法，可根据项目需求组合使用：

1. 扩展名初步校验 #

通过文件后缀快速筛选，适合做第一道过滤：

• 先定义支持的图片扩展名集合，再提取文件后缀做匹配
• 代码示例（Python）：

import os

def is_image_by_ext(file_path):
    valid_exts = {'.jpg', '.jpeg', '.png', '.gif', '.bmp', '.webp'}
    file_ext = os.path.splitext(file_path)[1].lower()
    return file_ext in valid_exts

# 调用示例
input_path = "./docs/readme.txt"
if not is_image_by_ext(input_path):
    raise ValueError("输入路径指向的不是图片文件，请提供后缀为jpg/png/gif等格式的图片路径")

• 局限性：无法识别被篡改扩展名的假图片，仅作为初步校验

2. 魔术数字精准校验 #

读取文件开头的固定字节（魔术数字），匹配图片格式的专属标识，准确性更高：

• 每种图片格式都有唯一的头部字节特征，比如PNG开头是0x89504E47，JPEG开头是0xFFD8FF
• 代码示例（Python）：

def is_image_by_magic(file_path):
    # 映射图片格式与对应的头部字节
    magic_signatures = {
        b'\xFF\xD8\xFF': ('jpg', 'jpeg'),
        b'\x89\x50\x4E\x47\x0D\x0A\x1A\x0A': ('png',),
        b'\x47\x49\x46\x38\x37\x61': ('gif',),
        b'\x47\x49\x46\x38\x39\x61': ('gif',),
        b'\x42\x4D': ('bmp',),
        b'\x52\x49\x46\x46': ('webp',)
    }

    try:
        with open(file_path, 'rb') as f:
            header_bytes = f.read(8)  # 读取前8字节覆盖绝大多数格式头部
    except PermissionError:
        raise PermissionError("无权限读取该文件，请检查路径权限")

    for sig, _ in magic_signatures.items():
        if header_bytes.startswith(sig):
            return True
    return False

# 调用示例
input_path = "./fake_image.jpg"  # 文本文件改后缀
if not is_image_by_magic(input_path):
    raise ValueError("输入文件不是有效图片，请勿篡改扩展名或提供损坏文件")

• 优势：不受扩展名影响，能精准识别真实图片格式

3. 图像库加载验证 #

借助专业图像库尝试加载文件，同时能验证图片是否损坏：

• 使用Pillow（Python）或OpenCV等库，若加载失败则判定为非图片
• 代码示例（Pillow）：

from PIL import Image

def is_valid_image(file_path):
    try:
        with Image.open(file_path) as img:
            img.verify()  # 验证图片完整性
            return True
    except (IOError, SyntaxError, ValueError) as e:
        return False

# 调用示例
input_path = "./corrupted_image.png"
if not is_valid_image(input_path):
    raise ValueError("无法识别为有效图片，请检查文件是否损坏或路径是否正确")

• 优势：不仅能判断是否为图片，还能检测图片是否损坏
• 注意：需提前安装依赖库（pip install pillow）

推荐组合方案 #

先用扩展名校验快速排除明显非图片文件，再用魔术数字校验做精准识别，若需要确保图片可正常使用，再加上图像库加载验证，兼顾性能与准确性。

内容的提问来源于stack exchange，提问作者Beginner coder