|
|
import os, uuid
|
|
|
from flask import render_template, request, send_file
|
|
|
|
|
|
OUTPUT_FOLDER = os.path.join(os.path.dirname(__file__), "..", "output")
|
|
|
os.makedirs(OUTPUT_FOLDER, exist_ok=True)
|
|
|
|
|
|
def compress_image(input_stream, quality=85):
|
|
|
"""使用 PIL 压缩图片"""
|
|
|
try:
|
|
|
from PIL import Image
|
|
|
img = Image.open(input_stream)
|
|
|
|
|
|
# 转为 RGB（支持 JPG）
|
|
|
if img.mode in ("RGBA", "P"):
|
|
|
img = img.convert("RGB")
|
|
|
|
|
|
out_name = f"compressed_{uuid.uuid4().hex}.jpg"
|
|
|
out_path = os.path.join(OUTPUT_FOLDER, out_name)
|
|
|
|
|
|
img.save(out_path, "JPEG", quality=quality, optimize=True)
|
|
|
return out_name
|
|
|
except Exception as e:
|
|
|
raise Exception(f"图片处理失败: {e}")
|
|
|
|
|
|
def page():
|
|
|
from flask import render_template, request
|
|
|
download_file = None
|
|
|
message = None
|
|
|
quality = 85
|
|
|
|
|
|
if request.method == "POST":
|
|
|
file = request.files.get("file")
|
|
|
quality = int(request.form.get("quality", 85))
|
|
|
|
|
|
if file:
|
|
|
try:
|
|
|
suffix = os.path.splitext(file.filename)[-1].lower()
|
|
|
if suffix not in [".jpg", ".jpeg", ".png", ".webp"]:
|
|
|
raise Exception("仅支持 JPG/PNG/WEBP 格式")
|
|
|
|
|
|
tmp_path = os.path.join(OUTPUT_FOLDER, f"tmp_{uuid.uuid4().hex}{suffix}")
|
|
|
file.save(tmp_path)
|
|
|
|
|
|
download_file = compress_image(tmp_path, quality)
|
|
|
os.remove(tmp_path)
|
|
|
message = "✅ 压缩成功"
|
|
|
except Exception as e:
|
|
|
message = f"❌ {e}"
|
|
|
|
|
|
return render_template("image_compress.html", download_file=download_file, message=message, quality=quality) |
