356 lines
11 KiB
Python
356 lines
11 KiB
Python
|
"""
|
|||
|
|
图片处理模块
|
|||
|
|
|
|||
|
|
负责图片文件的处理，包括图片读取、尺寸调整、格式转换和对齐设置等功能。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
from typing import Tuple, Optional
|
|||
|
|
from PIL import Image
|
|||
|
|
from docx.shared import Inches
|
|||
|
|
from docx.enum.text import WD_ALIGN_PARAGRAPH
|
|||
|
|
from config import config
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ImageProcessor:
|
|||
|
|
"""图片处理器类"""
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def process_image(image_path: str) -> Tuple[Image.Image, float]:
|
|||
|
|
"""
|
|||
|
|
处理图片，包括方向矫正和尺寸调整
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
image_path: 图片文件路径
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Tuple[Image.Image, float]: 处理后的图片对象和宽度（英寸）
|
|||
|
|
|
|||
|
|
Raises:
|
|||
|
|
Exception: 处理图片失败时
|
|||
|
|
"""
|
|||
|
|
if not os.path.exists(image_path):
|
|||
|
|
raise Exception(f"图片文件不存在: {image_path}")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
with Image.open(image_path) as img:
|
|||
|
|
# 处理图片方向（EXIF旋转信息）
|
|||
|
|
img = ImageProcessor._fix_image_orientation(img)
|
|||
|
|
|
|||
|
|
# 调整图片尺寸
|
|||
|
|
img, width_inches = ImageProcessor._resize_image(img)
|
|||
|
|
|
|||
|
|
return img, width_inches
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
raise Exception(f"处理图片失败 {image_path}: {str(e)}")
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def _fix_image_orientation(img: Image.Image) -> Image.Image:
|
|||
|
|
"""
|
|||
|
|
根据EXIF信息修正图片方向
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
img: PIL图片对象
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Image.Image: 方向修正后的图片
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
# 检查是否有EXIF数据
|
|||
|
|
if hasattr(img, '_getexif'):
|
|||
|
|
exif = img._getexif()
|
|||
|
|
if exif is not None:
|
|||
|
|
# EXIF方向标签
|
|||
|
|
orientation_tag = 274
|
|||
|
|
if orientation_tag in exif:
|
|||
|
|
orientation = exif[orientation_tag]
|
|||
|
|
|
|||
|
|
# 根据方向值进行旋转
|
|||
|
|
if orientation == 3:
|
|||
|
|
img = img.rotate(180, expand=True)
|
|||
|
|
elif orientation == 6:
|
|||
|
|
img = img.rotate(270, expand=True)
|
|||
|
|
elif orientation == 8:
|
|||
|
|
img = img.rotate(90, expand=True)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"修正图片方向时出错: {e}")
|
|||
|
|
|
|||
|
|
return img
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def _resize_image(img: Image.Image) -> Tuple[Image.Image, float]:
|
|||
|
|
"""
|
|||
|
|
根据配置调整图片尺寸
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
img: PIL图片对象
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Tuple[Image.Image, float]: 调整后的图片和宽度（英寸）
|
|||
|
|
"""
|
|||
|
|
if config.image_resize == "width" and config.image_width > 0:
|
|||
|
|
# 按指定宽度调整
|
|||
|
|
target_width_px = config.image_width * 96 # 96 DPI
|
|||
|
|
width, height = img.size
|
|||
|
|
|
|||
|
|
if width > target_width_px:
|
|||
|
|
ratio = target_width_px / width
|
|||
|
|
new_height = int(height * ratio)
|
|||
|
|
img = img.resize((int(target_width_px), new_height), Image.LANCZOS)
|
|||
|
|
|
|||
|
|
return img, config.image_width
|
|||
|
|
else:
|
|||
|
|
# 不调整尺寸，计算当前宽度（英寸）
|
|||
|
|
width_inches = img.width / 96 # 假设96 DPI
|
|||
|
|
return img, width_inches
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def get_image_alignment():
|
|||
|
|
"""
|
|||
|
|
获取图片对齐方式的Word枚举值
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
WD_ALIGN_PARAGRAPH: Word对齐方式枚举
|
|||
|
|
"""
|
|||
|
|
alignment_map = {
|
|||
|
|
"left": WD_ALIGN_PARAGRAPH.LEFT,
|
|||
|
|
"center": WD_ALIGN_PARAGRAPH.CENTER,
|
|||
|
|
"right": WD_ALIGN_PARAGRAPH.RIGHT
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return alignment_map.get(config.image_alignment, WD_ALIGN_PARAGRAPH.CENTER)
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def validate_image(image_path: str) -> dict:
|
|||
|
|
"""
|
|||
|
|
验证图片文件的有效性
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
image_path: 图片文件路径
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
dict: 验证结果，包含有效性、错误信息和图片信息
|
|||
|
|
"""
|
|||
|
|
result = {
|
|||
|
|
"valid": False,
|
|||
|
|
"error": None,
|
|||
|
|
"info": {}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if not os.path.exists(image_path):
|
|||
|
|
result["error"] = "文件不存在"
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
with Image.open(image_path) as img:
|
|||
|
|
result["valid"] = True
|
|||
|
|
result["info"] = {
|
|||
|
|
"format": img.format,
|
|||
|
|
"mode": img.mode,
|
|||
|
|
"size": img.size,
|
|||
|
|
"width": img.width,
|
|||
|
|
"height": img.height
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 检查图片是否过大
|
|||
|
|
if img.width > 10000 or img.height > 10000:
|
|||
|
|
result["error"] = "图片尺寸过大"
|
|||
|
|
result["valid"] = False
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
result["error"] = f"无法打开图片: {str(e)}"
|
|||
|
|
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def get_supported_formats() -> list:
|
|||
|
|
"""
|
|||
|
|
获取支持的图片格式列表
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
list: 支持的图片格式扩展名列表
|
|||
|
|
"""
|
|||
|
|
return ['.jpg', '.jpeg', '.png', '.bmp', '.gif', '.webp', '.tiff']
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def convert_image_format(image_path: str, target_format: str, output_path: str) -> bool:
|
|||
|
|
"""
|
|||
|
|
转换图片格式
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
image_path: 源图片路径
|
|||
|
|
target_format: 目标格式（如'PNG', 'JPEG'）
|
|||
|
|
output_path: 输出文件路径
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
bool: 是否转换成功
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
with Image.open(image_path) as img:
|
|||
|
|
# 如果是JPEG格式且原图有透明通道，转为RGB
|
|||
|
|
if target_format.upper() == 'JPEG' and img.mode in ('RGBA', 'LA'):
|
|||
|
|
rgb_img = Image.new('RGB', img.size, (255, 255, 255))
|
|||
|
|
rgb_img.paste(img, mask=img.split()[-1] if img.mode == 'RGBA' else None)
|
|||
|
|
img = rgb_img
|
|||
|
|
|
|||
|
|
img.save(output_path, format=target_format)
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"转换图片格式失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def create_thumbnail(image_path: str, thumbnail_path: str, size: Tuple[int, int] = (200, 200)) -> bool:
|
|||
|
|
"""
|
|||
|
|
创建图片缩略图
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
image_path: 源图片路径
|
|||
|
|
thumbnail_path: 缩略图保存路径
|
|||
|
|
size: 缩略图尺寸（宽度, 高度）
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
bool: 是否创建成功
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
with Image.open(image_path) as img:
|
|||
|
|
img.thumbnail(size, Image.LANCZOS)
|
|||
|
|
img.save(thumbnail_path)
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"创建缩略图失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def get_image_info(image_path: str) -> Optional[dict]:
|
|||
|
|
"""
|
|||
|
|
获取图片详细信息
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
image_path: 图片文件路径
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
Optional[dict]: 图片信息字典，失败时返回None
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
with Image.open(image_path) as img:
|
|||
|
|
info = {
|
|||
|
|
"filename": os.path.basename(image_path),
|
|||
|
|
"format": img.format,
|
|||
|
|
"mode": img.mode,
|
|||
|
|
"size": img.size,
|
|||
|
|
"width": img.width,
|
|||
|
|
"height": img.height,
|
|||
|
|
"file_size": os.path.getsize(image_path)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 尝试获取EXIF信息
|
|||
|
|
if hasattr(img, '_getexif'):
|
|||
|
|
exif = img._getexif()
|
|||
|
|
if exif:
|
|||
|
|
info["has_exif"] = True
|
|||
|
|
# 获取一些常用的EXIF信息
|
|||
|
|
orientation = exif.get(274) # 方向
|
|||
|
|
if orientation:
|
|||
|
|
info["orientation"] = orientation
|
|||
|
|
else:
|
|||
|
|
info["has_exif"] = False
|
|||
|
|
else:
|
|||
|
|
info["has_exif"] = False
|
|||
|
|
|
|||
|
|
return info
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"获取图片信息失败: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def batch_validate_images(image_paths: list) -> dict:
|
|||
|
|
"""
|
|||
|
|
批量验证图片文件
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
image_paths: 图片文件路径列表
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
dict: 验证结果统计
|
|||
|
|
"""
|
|||
|
|
result = {
|
|||
|
|
"total": len(image_paths),
|
|||
|
|
"valid": 0,
|
|||
|
|
"invalid": 0,
|
|||
|
|
"errors": []
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for image_path in image_paths:
|
|||
|
|
validation = ImageProcessor.validate_image(image_path)
|
|||
|
|
if validation["valid"]:
|
|||
|
|
result["valid"] += 1
|
|||
|
|
else:
|
|||
|
|
result["invalid"] += 1
|
|||
|
|
result["errors"].append({
|
|||
|
|
"path": image_path,
|
|||
|
|
"error": validation["error"]
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
@staticmethod
|
|||
|
|
def optimize_image_for_docx(image_path: str, temp_dir: str) -> str:
|
|||
|
|
"""
|
|||
|
|
优化图片以适合插入DOCX文档
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
image_path: 原图片路径
|
|||
|
|
temp_dir: 临时文件目录
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
str: 优化后的图片路径
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
# 确保临时目录存在
|
|||
|
|
os.makedirs(temp_dir, exist_ok=True)
|
|||
|
|
|
|||
|
|
with Image.open(image_path) as img:
|
|||
|
|
# 修正方向
|
|||
|
|
img = ImageProcessor._fix_image_orientation(img)
|
|||
|
|
|
|||
|
|
# 根据配置调整尺寸
|
|||
|
|
img, _ = ImageProcessor._resize_image(img)
|
|||
|
|
|
|||
|
|
# 生成临时文件路径
|
|||
|
|
filename = os.path.basename(image_path)
|
|||
|
|
name, ext = os.path.splitext(filename)
|
|||
|
|
temp_path = os.path.join(temp_dir, f"{name}_optimized{ext}")
|
|||
|
|
|
|||
|
|
# 保存优化后的图片
|
|||
|
|
# 如果是PNG且没有透明通道，转为JPEG以减少文件大小
|
|||
|
|
if img.format == 'PNG' and img.mode == 'RGB':
|
|||
|
|
temp_path = os.path.join(temp_dir, f"{name}_optimized.jpg")
|
|||
|
|
img.save(temp_path, 'JPEG', quality=85, optimize=True)
|
|||
|
|
else:
|
|||
|
|
img.save(temp_path, optimize=True)
|
|||
|
|
|
|||
|
|
return temp_path
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"优化图片失败: {e}")
|
|||
|
|
return image_path # 返回原路径
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 创建全局图片处理器实例
|
|||
|
|
image_processor = ImageProcessor()
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 兼容旧接口的函数
|
|||
|
|
def process_image(image_path: str) -> Tuple[Image.Image, float]:
|
|||
|
|
"""处理图片（兼容旧接口）"""
|
|||
|
|
return ImageProcessor.process_image(image_path)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_image_alignment():
|
|||
|
|
"""获取图片对齐方式（兼容旧接口）"""
|
|||
|
|
return ImageProcessor.get_image_alignment()
|