zwt13703
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
from minio import Minio
|
|
from config import settings
|
|
|
|
# MinIO 客户端
|
|
minio_client = Minio(
|
|
settings.MINIO_ENDPOINT,
|
|
access_key=settings.MINIO_ACCESS_KEY,
|
|
secret_key=settings.MINIO_SECRET_KEY,
|
|
secure=settings.MINIO_USE_SSL,
|
|
)
|
|
|
|
|
|
async def init_buckets():
|
|
"""初始化 MinIO 存储桶（应用启动时调用）"""
|
|
buckets = [
|
|
settings.MINIO_BUCKET_TEMPLATES, # 原始模板文件
|
|
settings.MINIO_BUCKET_UPLOADS, # 用户上传的参考文件
|
|
settings.MINIO_BUCKET_OUTPUTS, # 生成的文档
|
|
]
|
|
for bucket in buckets:
|
|
if not minio_client.bucket_exists(bucket):
|
|
minio_client.make_bucket(bucket)
|
|
print(f"[MinIO] 创建存储桶: {bucket}")
|
|
|
|
|
|
def get_file_url(bucket: str, object_name: str) -> str:
|
|
"""获取文件的公开访问 URL"""
|
|
if settings.MINIO_USE_SSL:
|
|
protocol = "https"
|
|
else:
|
|
protocol = "http"
|
|
return f"{protocol}://{settings.MINIO_ENDPOINT}/{bucket}/{object_name}"
|
|
|
|
|
|
def get_presigned_url(bucket: str, object_name: str, expires: int = 3600) -> str:
|
|
"""获取预签名下载 URL（带过期时间）"""
|
|
return minio_client.presigned_get_object(bucket, object_name, expires=expires)
|