#MinIO 实战教程：从0到1搭建轻量S3兼容存储

from flask import Flask, request, jsonify
from minio import Minio
from minio.error import S3Error
import uuid
from werkzeug.utils import secure_filename

# 1. 初始化MinIO客户端
minio_client = Minio(
    "localhost:9000",
    access_key="minio_admin",
    secret_key="minio_strong_pwd123",
    secure=False  # 生产环境务必用HTTPS+true
)

# 2. 确保存储桶存在
if not minio_client.bucket_exists("user_uploads"):
    minio_client.make_bucket("user_uploads")

app = Flask(__name__)
ALLOWED_EXT = {"png", "jpg", "jpeg", "pdf", "mp4"}

def allowed_file(filename):
    return "." in filename and filename.rsplit(".", 1)[1].lower() in ALLOWED_EXT

# 3. 上传接口
@app.route("/api/upload", methods=["POST"])
def upload():
    if "file" not in request.files:
        return jsonify({"msg": "无文件"}), 400
    file = request.files["file"]
    if file.filename == "" or not allowed_file(file.filename):
        return jsonify({"msg": "文件名空或格式不支持"}), 400

    try:
        # 生成唯一文件名避免冲突
        unique_name = f"{uuid.uuid4().hex}_{secure_filename(file.filename)}"
        # 直接上传内存中的文件（无需存本地临时文件）
        file_size = len(file.read())
        file.seek(0)
        minio_client.put_object(
            "user_uploads", unique_name, file, file_size,
            content_type=file.content_type
        )
        # 生成公共访问URL（前提是bucket设为public）
        public_url = f"http://localhost:9000/user_uploads/{unique_name}"
        return jsonify({"msg": "上传成功", "url": public_url})
    except S3Error as e:
        return jsonify({"msg": f"存储错误: {e}"}), 500

if __name__ == "__main__":
    app.run(debug=True, port=5000)

from PIL import Image
import io
from minio import Minio

minio_client = Minio("localhost:9000", "minio_admin", "minio_strong_pwd123", secure=False)
if not minio_client.bucket_exists("image_assets"):
    minio_client.make_bucket("image_assets")

def upload_with_thumb(original_img_path, thumb_size=(200, 200)):
    # 1. 读取原图
    with open(original_img_path, "rb") as f:
        original_data = f.read()
    original_name = original_img_path.split("/")[-1]
    thumb_name = f"thumb_{original_name}"

    # 2. 生成缩略图
    img = Image.open(io.BytesIO(original_data))
    img.thumbnail(thumb_size)
    thumb_buffer = io.BytesIO()
    img.save(thumb_buffer, format=img.format)
    thumb_buffer.seek(0)

    # 3. 批量上传
    minio_client.put_object("image_assets", original_name, io.BytesIO(original_data), len(original_data))
    minio_client.put_object("image_assets", thumb_name, thumb_buffer, thumb_buffer.getbuffer().nbytes)

    return f"http://localhost:9000/image_assets/{original_name}", f"http://localhost:9000/image_assets/{thumb_name}"

# 测试
original_url, thumb_url = upload_with_thumb("./cat.jpg")
print(f"原图: {original_url}")
print(f"缩略图: {thumb_url}")