89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
# AUTO-MAS: A Multi-Script, Multi-Config Management and Automation Software
|
||
# Copyright © 2025 ClozyA
|
||
# Copyright © 2025 AUTO-MAS Team
|
||
|
||
# This file is part of AUTO-MAS.
|
||
|
||
# AUTO-MAS is free software: you can redistribute it and/or modify
|
||
# it under the terms of the GNU General Public License as published
|
||
# by the Free Software Foundation, either version 3 of the License,
|
||
# or (at your option) any later version.
|
||
|
||
# AUTO-MAS is distributed in the hope that it will be useful,
|
||
# but WITHOUT ANY WARRANTY; without even the implied warranty
|
||
# of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
|
||
# the GNU General Public License for more details.
|
||
|
||
# You should have received a copy of the GNU General Public License
|
||
# along with AUTO-MAS. If not, see <https://www.gnu.org/licenses/>.
|
||
|
||
# Contact: DLmaster_361@163.com
|
||
|
||
|
||
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对象）
|
||
"""
|
||
|
||
RESAMPLE = Image.Resampling.LANCZOS # Pillow 9.1.0及以后
|
||
|
||
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 quality is not None:
|
||
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
|