1.优化隔离工作目录2.新增图像生成和编辑功能3.生成模型替换为本地flux2 klein

src/server/utils/new_oss_client.py

@@ -36,7 +36,7 @@ http_client = urllib3.PoolManager(
 
 
 #  获取图片
-def oss_get_image(oss_client, bucket, object_name, data_type):
+def oss_get_image(oss_client, bucket, object_name):
     # cv2 默认全通道读取
     image_object = None
     try:
@@ -57,9 +57,44 @@ def oss_upload_image(oss_client, bucket, object_name, image_bytes):
     return req
 
 
+def oss_upload_image_file(oss_client, bucket, object_name, file_path):
+    req = None
+    try:
+        req = oss_client.fput_object(
+            bucket_name=bucket,
+            object_name=object_name,
+            file_path=file_path
+        )
+    except Exception as e:
+        logger.warning(f" | 上传图片出现异常 ######: {e}")
+    return req
+
+
+def get_presigned_url(oss_client, bucket, object_name):
+    try:
+        presigned_url = oss_client.presigned_get_object(
+            bucket_name=bucket,
+            object_name=object_name,
+            expires=3600
+        )
+        return presigned_url
+    except Exception as e:
+        print(f"get_presigned_url exception :{e}")
+        return "object not found"
+
+
+def is_minio_file_exist(minio_client: Minio, bucket_name: str, object_name: str) -> bool:
+    try:
+        # 核心判断：检查MinIO中指定bucket+object是否存在
+        minio_client.stat_object(bucket_name, object_name)
+        return True
+    except Exception as e:
+        return False
+
+
 if __name__ == '__main__':
-    url = "fida-test/furniture/sketches/4449a66d-6267-43f7-86a2-1e42bd19ec61.png"
+    url = 'fida-test/furniture/sketches/9356e3d8-d56e-4478-adde-61b29119979b.png'
     read_type = "2"
-    img = oss_get_image(oss_client=minio_client, bucket=url.split('/')[0], object_name=url[url.find('/') + 1:], data_type=read_type)
+    img = oss_get_image(oss_client=minio_client, bucket=url.split('/')[0], object_name=url[url.find('/') + 1:])
     img.show()
     img.save("result.png")