feat : 代码梳理 移除所有敏感密钥 通过环境变量方式配置

app/service/generate_image/utils/upload_sd_image.py

@@ -9,16 +9,13 @@
 """
 import io
 import logging
-
-# import boto3
 import cv2
-from PIL import Image
 from minio import Minio
 
-from app.core.config import *
-from app.service.utils.oss_client import oss_upload_image
+from app.core.config import settings
+from app.service.utils.new_oss_client import oss_upload_image
 
-minio_client = Minio(MINIO_URL, access_key=MINIO_ACCESS, secret_key=MINIO_SECRET, secure=MINIO_SECURE)
+minio_client = Minio(settings.MINIO_URL, access_key=settings.MINIO_ACCESS, secret_key=settings.MINIO_SECRET, secure=settings.MINIO_SECURE)
 
 
 # s3 = boto3.client('s3', aws_access_key_id=S3_ACCESS_KEY, aws_secret_access_key=S3_AWS_SECRET_ACCESS_KEY, region_name=S3_REGION_NAME)
@@ -52,7 +49,7 @@ def upload_SDXL_image(image, user_id, category, file_name):
         #     content_type='image/jpeg'
         # )
         object_name = f'{user_id}/{category}/{file_name}'
-        req = oss_upload_image(bucket=GI_MINIO_BUCKET, object_name=object_name, image_bytes=image_bytes)
+        oss_upload_image(oss_client=minio_client, bucket="aida-users", object_name=object_name, image_bytes=image_bytes)
         image_url = f"aida-users/{object_name}"
         return image_url
     except Exception as e:
@@ -63,7 +60,7 @@ def upload_png_sd(image, user_id, category, file_name):
     try:
         _, img_byte_array = cv2.imencode('.jpg', image)
         object_name = f'{user_id}/{category}/{file_name}'
-        req = oss_upload_image(bucket=GI_MINIO_BUCKET, object_name=object_name, image_bytes=img_byte_array)
+        oss_upload_image(oss_client=minio_client, bucket="aida-users", object_name=object_name, image_bytes=img_byte_array)
         image_url = f"aida-users/{object_name}"
         return image_url
     except Exception as e: