添加MinIO客户端配置和文件上传功能

minio_api.py

@@ -0,0 +1,108 @@
+from minio import Minio
+from minio.error import S3Error
+import os
+MINIO_HOST=os.getenv("MINIO_HOST", "127.0.0.1")
+
+MINIO_CONFIG = {
+    "endpoint": f"{MINIO_HOST}:{os.getenv('MINIO_PORT', '9000')}",
+    "access_key": os.getenv("MINIO_USER", "rag_flow"),
+    "secret_key": os.getenv("MINIO_PASSWORD", "infini_rag_flow"),
+    "secure": False
+}
+
+def get_minio_client():
+    """创建MinIO客户端"""
+    return Minio(
+        endpoint=MINIO_CONFIG["endpoint"],
+        access_key=MINIO_CONFIG["access_key"],
+        secret_key=MINIO_CONFIG["secret_key"],
+        secure=MINIO_CONFIG["secure"]
+    )
+# def upload_file():
+#     if 'files' not in request.files:
+#         return jsonify({
+#             'code': 400,
+#             'message': '未选择文件',
+#             'data': None
+#         }), 400
+    
+#     files = request.files.getlist('files')
+#     upload_result = upload_files_to_server(files)
+    
+#     # 返回标准格式
+#     return jsonify({
+#         'code': 0,
+#         'message': '上传成功',
+#         'data': upload_result['data']
+#     })
+
+def upload_files_to_server(files, parent_id=None, user_id=None):
+    """
+    上传文件到MinIO服务器
+    :param files: 文件路径列表
+    :param parent_id: 父级ID（可选）
+    :param user_id: 用户ID（可选）
+    """
+    for file_path in files:
+        if not os.path.isfile(file_path):
+            print(f"文件不存在: {file_path}")
+            continue
+        
+        # 获取文件名
+        file_name = os.path.basename(file_path)
+        
+        # 上传文件到MinIO
+        try:
+            with open(file_path, 'rb') as file_data:
+                minio_client.put_object (
+                    bucket_name=parent_id,
+                    object_name=file_name,
+                    data=file_data,
+                    length=os.path.getsize(file_path)
+                )
+            print(f"文件已上传到MinIO: {parent_id}/{file_name}")
+        except S3Error as e:
+            print(f"上传文件 '{file_name}' 失败: {e}")
+        except Exception as e:
+            print(f"发生错误: {e}")
+
+
+
+
+
+
+
+# 配置MinIO客户端
+# minio_client = Minio(
+#     endpoint="192.168.0.250:9000",  # 替换为你的MinIO服务器地址，例如"localhost:9000"
+#     access_key="jAGDJmMJ63k3MyRMmi4N",  # 替换为你的Access Key
+#     secret_key="gH2ZOiVh9pJk8bLI8jBz6d6ABNiaTrCuFiDLWojK",  # 替换为你的Secret Key
+#     secure=False  # 使用HTTPS（如果是本地测试且未配置SSL，可设置为False）
+# )
+
+minio_client= get_minio_client()
+
+
+# 要上传的存储桶信息
+bucket_name = "my-bucket"  # 替换为你的存储桶名称
+object_name = "my-file.txt"  # 文件在MinIO中存储的名称
+file_path = "./ragflow.txt"  # 本地文件路径
+
+try:
+    # 检查存储桶是否存在，如果不存在则创建（可选）
+    if not minio_client.bucket_exists(bucket_name):
+        minio_client.make_bucket(bucket_name)
+        print(f"Bucket '{bucket_name}' created")
+    
+    # 上传文件
+    minio_client.fput_object(
+        bucket_name=bucket_name,
+        object_name=object_name,
+        file_path=file_path
+    )
+    print(f"文件 '{file_path}' 成功上传到存储桶 '{bucket_name}' 为 '{object_name}'")
+    
+except S3Error as exc:
+    print("MinIO错误:", exc)
+except Exception as e:
+    print("发生错误:", e)