first commit

controllers/compress_image.py

@@ -0,0 +1,52 @@
+import os
+from PIL import Image
+from models.logger import setup_logger
+
+logger = setup_logger()
+
+def compress_image(input_path, output_folder: str = 'compressed_images', target_kb=200, quality_step=5) -> str:
+    """
+    将图片压缩到指定大小（以KB为单位）
+
+    Args:
+        input_path: 输入图片路径
+        output_folder: 输出文件夹
+        target_kb: 目标文件大小（默认200KB）
+        quality_step: 每次迭代中质量减少的步长（默认5）
+
+    Returns:
+         压缩后的图片路径
+    """
+    # 确保输出文件夹存在
+    os.makedirs(output_folder, exist_ok=True)
+
+    # 获取文件基本信息
+    file_size_kb = os.path.getsize(input_path) / 1024
+    base_name = os.path.basename(input_path)
+    output_path = os.path.join(output_folder, base_name.rsplit('.', 1)[0] + '.jpg')  # 默认转为jpg
+
+    if file_size_kb <= target_kb:
+        # 如果文件小于等于目标大小，直接复制
+        img = Image.open(input_path)
+        if img.mode in ("RGBA", "P"):
+            img = img.convert("RGB")
+        img.save(output_path, "JPEG", optimize=True, quality=95)
+        return str(output_path)
+
+    # 初始质量设置
+    quality = 85
+
+    while quality > 10:  # 设置最低质量限制避免过低质量的图片
+        with Image.open(input_path) as img:
+            if img.mode in ("RGBA", "P"):
+                img = img.convert("RGB")
+            img.save(output_path, "JPEG", optimize=True, quality=quality)
+
+        # 检查文件大小
+        if os.path.getsize(output_path) / 1024 < target_kb:
+            return str(output_path)
+
+        quality -= quality_step  # 减少质量设置
+
+    logger.error(f"图片 {input_path} 压缩失败!")
+    return ''