save code

basic_api_example.py

@@ -1,42 +1,41 @@
-import json
+# -*- coding=utf-8
-from urllib import request
+from qcloud_cos import CosConfig
+from qcloud_cos import CosS3Client
+import sys
+import os
+import logging
 
-#This is the ComfyUI api prompt format.
+# 正常情况日志级别使用 INFO，需要定位时可以修改为 DEBUG，此时 SDK 会打印和服务端的通信信息
+logging.basicConfig(level=logging.INFO, stream=sys.stdout)
 
-#If you want it for a specific workflow you can "enable dev mode options"
+# 1. 设置用户属性, 包括 secret_id, secret_key, region等。Appid 已在 CosConfig 中移除，请在参数 Bucket 中带上 Appid。Bucket 由 BucketName-Appid 组成
-#in the settings of the UI (gear beside the "Queue Size: ") this will enable
+secret_id = 'AKIDDIBrNhoiOtBF5rcEdoBRUKwU2Rj2OgxA'     # 用户的 SecretId，建议使用子账号密钥，授权遵循最小权限指引，降低使用风险。子账号密钥获取可参见 https://cloud.tencent.com/document/product/598/37140
-#a button on the UI to save workflows in api format.
+secret_key = '5wkKhttgYQD2iEZgaInEVYVPCo2BJK9l'   # 用户的 SecretKey，建议使用子账号密钥，授权遵循最小权限指引，降低使用风险。子账号密钥获取可参见 https://cloud.tencent.com/document/product/598/37140
-
+region = 'ap-beijing' #'ap-guangzhou'      # 替换为用户的 region，已创建桶归属的 region 可以在控制台查看，https://console.cloud.tencent.com/cos5/bucket
-#keep in mind ComfyUI is pre alpha software so this format will change a bit.
+                           # COS 支持的所有 region 列表参见 https://cloud.tencent.com/document/product/436/6224
-
+token = None               # 如果使用永久密钥不需要填入 token，如果使用临时密钥需要填入，临时密钥生成和使用指引参见 https://cloud.tencent.com/document/product/436/14048
-#this is the one for the default workflow
+scheme = 'https'           # 指定使用 http/https 协议来访问 COS，默认为 https，可不填
-
-def queue_prompt(prompt):
-    p = {"prompt": prompt}
-
-    # If the workflow contains API nodes, you can add a Comfy API key to the `extra_data`` field of the payload.
-    # p["extra_data"] = {
-    #     "api_key_comfy_org": "comfyui-87d01e28d*******************************************************"  # replace with real key
-    # }
-    # See: https://docs.comfy.org/tutorials/api-nodes/overview
-    # Generate a key here: https://platform.comfy.org/login
-
-    data = json.dumps(p).encode('utf-8')
-    req =  request.Request("http://127.0.0.1:8188/prompt", data=data)
-    request.urlopen(req)
 
 
-with open('/home/xsl/code/ComfyUI/change_cloth/basic_api.json', 'r', encoding='utf-8') as file:
+config = CosConfig(Region=region, SecretId=secret_id, SecretKey=secret_key, Token=token, Scheme=scheme)
-    prompt_text = file.read()
+client = CosS3Client(config)
 
-prompt = json.loads(prompt_text)
+#### 文件流简单上传（不支持超过5G的文件，推荐使用下方高级上传接口）
-#set the text prompt for our positive CLIPTextEncode
+# 强烈建议您以二进制模式(binary mode)打开文件,否则可能会导致错误
-prompt["6"]["inputs"]["text"] = "masterpiece best quality man"
+with open('out_image.png', 'rb') as fp:
-
+    response = client.put_object(
-#set the seed for our KSampler node
+        Bucket='b-1304254135',
-prompt["3"]["inputs"]["seed"] = 5
+        Body=fp,
-
+        Key='out_image.png',
-
+        StorageClass='STANDARD',
-queue_prompt(prompt)
+        EnableMD5=False
+    )
+print(response['ETag'])
 
 
+####  获取文件到本地
+response = client.get_object(
+    Bucket='b-1304254135',
+    Key='out_image.png'
+)
+response['Body'].get_stream_to_file('cos_girl.png')