s''

app/sdk/minio.py

@@ -0,0 +1,92 @@
+"""
+MinIO SDK
+对象存储封装，支持配置注入
+"""
+import os
+import time
+import random
+import string
+from typing import Optional
+
+try:
+    from minio import Minio
+    MINIO_AVAILABLE = True
+except ImportError:
+    MINIO_AVAILABLE = False
+    Minio = None
+
+
+class MinioSDK:
+    """MinIO 对象存储 SDK"""
+
+    def __init__(
+        self,
+        end_point: str,
+        access_key: str,
+        secret_key: str,
+        bucket: str = "hackrobot",
+        use_ssl: bool = True,
+        port: Optional[int] = None,
+        region: str = "china",
+        public_url: Optional[str] = None,
+        upload_prefix: str = "joins",
+    ):
+        if not MINIO_AVAILABLE:
+            raise ImportError("minio 未安装，请执行: pip install minio")
+        self.end_point = end_point
+        self.port = port or (443 if use_ssl else 80)
+        self.access_key = access_key
+        self.secret_key = secret_key
+        self.bucket = bucket
+        self.use_ssl = use_ssl
+        self.region = region
+        self.public_url = public_url or f"https://{end_point}/{bucket}"
+        self.upload_prefix = upload_prefix.rstrip("/")
+        self._client: Optional[Minio] = None
+
+    def get_client(self) -> "Minio":
+        """获取 MinIO 客户端"""
+        if self._client is None:
+            endpoint = f"{self.end_point}:{self.port}" if self.port not in (80, 443) else self.end_point
+            self._client = Minio(
+                endpoint,
+                access_key=self.access_key,
+                secret_key=self.secret_key,
+                secure=self.use_ssl,
+                region=self.region,
+            )
+        return self._client
+
+    def generate_object_key(self, ext: str) -> str:
+        """生成上传用的对象键"""
+        ext = ext.lstrip(".")
+        suffix = "".join(random.choices(string.ascii_lowercase + string.digits, k=8))
+        return f"{self.upload_prefix}/{int(time.time() * 1000)}_{suffix}.{ext}"
+
+    def upload_buffer(
+        self,
+        data: bytes,
+        object_key: str,
+        content_type: str = "application/octet-stream",
+    ) -> dict:
+        """
+        上传文件到 MinIO
+        :return: {"url": str, "object_key": str}
+        """
+        client = self.get_client()
+        from io import BytesIO
+        stream = BytesIO(data)
+        client.put_object(
+            self.bucket,
+            object_key,
+            stream,
+            len(data),
+            content_type=content_type,
+        )
+        url = f"{self.public_url.rstrip('/')}/{object_key}"
+        return {"url": url, "object_key": object_key}
+
+    @property
+    def enabled(self) -> bool:
+        """是否启用（需配置 access_key/secret_key）"""
+        return bool(self.end_point and self.access_key and self.secret_key)