aoxuan
4d4d6dbedf
- 使用 Path 对象替换字符串表示文件路径，提高代码可读性和功能 - 优化类型注解，包括函数参数和返回值- 重构 ImageUtils.compress_image_if_needed 方法，简化逻辑并提高性能 - 更新 notification.py 中使用该方法的代码，直接返回压缩后的 Path 对象
68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
import base64
|
||
import hashlib
|
||
from pathlib import Path
|
||
|
||
from PIL import Image
|
||
|
||
class ImageUtils:
|
||
@staticmethod
|
||
def get_base64_from_file(image_path):
|
||
"""从本地文件读取并返回base64编码字符串"""
|
||
with open(image_path, "rb") as f:
|
||
return base64.b64encode(f.read()).decode("utf-8")
|
||
|
||
@staticmethod
|
||
def calculate_md5_from_file(image_path):
|
||
"""从本地文件读取并返回md5值（hex字符串）"""
|
||
with open(image_path, "rb") as f:
|
||
return hashlib.md5(f.read()).hexdigest()
|
||
|
||
@staticmethod
|
||
def calculate_md5_from_base64(base64_content):
|
||
"""从base64字符串计算md5"""
|
||
image_data = base64.b64decode(base64_content)
|
||
return hashlib.md5(image_data).hexdigest()
|
||
|
||
@staticmethod
|
||
def compress_image_if_needed(image_path: Path, max_size_mb=2) -> Path:
|
||
"""
|
||
如果图片大于max_size_mb，则压缩并覆盖原文件，返回原始路径（Path对象）
|
||
"""
|
||
if hasattr(Image, "Resampling"): # Pillow 9.1.0及以后
|
||
RESAMPLE = Image.Resampling.LANCZOS
|
||
else:
|
||
RESAMPLE = Image.ANTIALIAS
|
||
|
||
max_size = max_size_mb * 1024 * 1024
|
||
if image_path.stat().st_size <= max_size:
|
||
return image_path
|
||
|
||
img = Image.open(image_path)
|
||
suffix = image_path.suffix.lower()
|
||
quality = 90 if suffix in [".jpg", ".jpeg"] else None
|
||
step = 5
|
||
|
||
if suffix in [".jpg", ".jpeg"]:
|
||
while True:
|
||
img.save(image_path, quality=quality, optimize=True)
|
||
if image_path.stat().st_size <= max_size or quality <= 10:
|
||
break
|
||
quality -= step
|
||
elif suffix == ".png":
|
||
width, height = img.size
|
||
while True:
|
||
img.save(image_path, optimize=True)
|
||
if (
|
||
image_path.stat().st_size <= max_size
|
||
or width <= 200
|
||
or height <= 200
|
||
):
|
||
break
|
||
width = int(width * 0.95)
|
||
height = int(height * 0.95)
|
||
img = img.resize((width, height), RESAMPLE)
|
||
else:
|
||
raise ValueError("仅支持JPG/JPEG和PNG格式图片的压缩。")
|
||
|
||
return image_path
|