往云端comfyui上传图片的代码

这个upload_image函数主要做了5个动作：

构建上传URL

url = f"http://{server_address}/api/upload/image"

记录日志

logging.info(f"上传文件中，文件路径: {file_path},重命名:{os.path.basename(file_path)}")

读取文件并准备上传数据

with open(file_path, 'rb') as file:
    files = {'image': (os.path.basename(file_path), file)}

发送POST请求

response = requests.post(url, files=files)

错误处理

# 处理三种情况：
# a. 请求成功(200)
if response.status_code == 200:
    pass

# b. 文件不存在错误
except FileNotFoundError:
    logging.error(f"上传图片时文件 {file_path} 未找到，请检查路径。")

# c. 其他错误
except Exception as e:
    logging.error(f"上传图片时发生未知错误: {e}")

每个动作都有明确的目的：

动作1：准备上传地址
动作2：记录操作日志
动作3：准备上传数据
动作4：执行上传操作
动作5：处理各种可能的错误

图生图

def upload_image(file_path,server_address,task_id):

url = f"http://{server_address}/api/upload/image"
# 打开文件并发送请求
logging.info(f"上传文件中，文件路径: {file_path},重命名:{os.path.basename(file_path)}")
# print(f"要上传到文件路径:{file_path}")
# print(f"要上传的地址:{url}")
# print(f"文件命名:{os.path.basename(file_path)}")
try:
    with open(file_path, 'rb') as file:
        # 构建文件部分的参数
        files = {'image': (os.path.basename(file_path), file)}  # 根据实际需求调整 MIME 类型

        # 发送 POST 请求
        response = requests.post(url, files=files)

    # 打印响应内容
    if response.status_code == 200:
        # print("上传成功:", response.json())
        pass
    else:
        raise FrpErrorException(f"任务ID：{task_id}，图片上传至宿主机失败导致生成失败，状态码: {response.status_code}, 返回内容: {response.text}")


except FileNotFoundError:
    logging.error(f"上传图片时文件 {file_path} 未找到，请检查路径。")
except Exception as e:
    logging.error(f"上传图片时发生未知错误: {e}")