bizydraft 0.2.63__py3-none-any.whl → 0.2.65__py3-none-any.whl

bizydraft/hijack_nodes.py

@@ -66,7 +66,15 @@ def get_data_load_classes_from_url(config_url):
         return []
 
 
-DATA_LOAD_CLASSES = []
+DATA_LOAD_CLASSES = [
+    "LoadImage",
+    "LoadVideo",
+    "LoadImageMask",
+    "LoadAudio",
+    "Load3D",
+    "VHS_LoadAudioUpload",
+    "VHS_LoadVideo",
+]
 
 if COMFYAGENT_NODE_CONFIG.startswith("http"):
     fetched_classes = get_data_load_classes_from_url(COMFYAGENT_NODE_CONFIG)

bizydraft/patch_handlers.py

@@ -50,7 +50,7 @@ async def view_image(request):
     try:
         if "http" in subfolder:
             subfolder = subfolder[subfolder.find("http") :]
-        subfolder = unquote(subfolder)
+        subfolder = unquote(subfolder).replace("https:/", "https://")
         filename = (
             f"{subfolder}/{filename}"
             if not filename.startswith(http_prefix_options)
@@ -58,13 +58,19 @@ async def view_image(request):
         )  # preview 3d request: https://host:port/api/view?filename=filename.glb&type=output&subfolder=https://bizyair-dev.oss-cn-shanghai.aliyuncs.com/outputs&rand=0.5763957215362988
 
         content_type, _ = mimetypes.guess_type(filename)
-        if content_type and any(x in content_type for x in ("image", "video")):
-            return web.HTTPFound(filename)
 
         timeout = ClientTimeout(total=BIZYDRAFT_REQUEST_TIMEOUT)
         async with ClientSession(timeout=timeout) as session:
             async with session.get(filename) as resp:
                 resp.raise_for_status()
+
+                # 优先使用服务器返回的Content-Type，如果无法获取则使用猜测的类型
+                final_content_type = (
+                    resp.headers.get("Content-Type")
+                    or content_type
+                    or "application/octet-stream"
+                )
+
                 content_length = int(resp.headers.get("Content-Length", 0))
                 if content_length > BIZYDRAFT_MAX_FILE_SIZE:
                     logger.warning(
@@ -77,7 +83,7 @@ async def view_image(request):
 
                 headers = {
                     "Content-Disposition": f'attachment; filename="{uuid.uuid4()}"',
-                    "Content-Type": "application/octet-stream",
+                    "Content-Type": final_content_type,
                 }
 
                 proxy_response = web.StreamResponse(headers=headers)

{bizydraft-0.2.63.dist-info → bizydraft-0.2.65.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bizydraft
-Version: 0.2.63
+Version: 0.2.65
 Summary: bizydraft
 Requires-Dist: loguru
 Requires-Dist: aiohttp

{bizydraft-0.2.63.dist-info → bizydraft-0.2.65.dist-info}/RECORD

@@ -1,10 +1,10 @@
 bizydraft/__init__.py,sha256=OM-sKCQrPh25nHVJIX-DgF1raMYyoWLSuyduIAHt0Gs,78
 bizydraft/block_nodes.py,sha256=Lqn3oSCaGDHR2OICc8a2iRoRCVVK9v1-9MM3r-qIZgA,1092
 bizydraft/env.py,sha256=VFmGopVL2TGWA6hwxyFhIglCEcQxy6iVvL_raMNd6u4,407
-bizydraft/hijack_nodes.py,sha256=i-HCZvEqCx25NhClM2wHHCBnBi4IYxuuAcRviVRFkcg,3331
+bizydraft/hijack_nodes.py,sha256=flE0FH3pDUBAej5T3X-DiyLwxSz478OFC4ewcNLUBvM,3466
 bizydraft/hijack_routes.py,sha256=lf6x3xDzbo9yQIRwfG_1oxcUNfrX_1ogbiff3WOV9gM,3268
 bizydraft/oss_utils.py,sha256=FxTe9TxA9Q13pxKaOpcN0Jui-BG1SBWySM71PqpIlsM,8601
-bizydraft/patch_handlers.py,sha256=WPf4xeV5sBpoJ2MkHHhx0wem7B0KQNuhONElN7c9v3g,5989
+bizydraft/patch_handlers.py,sha256=341y4YpzXOngQk9vLO8KLr7MXqAgM-_DrGrtIVGbPWo,6193
 bizydraft/postload.py,sha256=XFElKcmCajT_oO7SVJJBaN04XcWro54N5HB5cSCxfvI,1308
 bizydraft/prestartup_patch.py,sha256=4FGjmRcDHELjtlQOrfTfk2Un5OS89QIqfq-gEcB9WDs,998
 bizydraft/resp.py,sha256=8INvKOe5Dgai3peKfqKjrhUoYeuXWXn358w30-_cY-A,369
@@ -23,7 +23,7 @@ bizydraft/static/js/socket.js,sha256=VE3fTAgEfM0FZhL526Skt7OCRokOa3mzTCAjAomI_tE
 bizydraft/static/js/tool.js,sha256=VupamUuh7tYiDnBTrL5Z_yLmhJinskhzRXwE3zfsKZM,2901
 bizydraft/static/js/uploadFile.js,sha256=WvglKzHMeOzDhOH3P-fLcPHxCLbKOJpo4DntoRxeJtI,4908
 bizydraft/static/js/workflow_io.js,sha256=FWAjncvWhvy-3nN_legD2fpRwgnIncpRLHU5X016a-U,5236
-bizydraft-0.2.63.dist-info/METADATA,sha256=k1aofr7e1X1o9sL5PGa5kbNg8vVXVfLz7f0cDpTyCbM,162
-bizydraft-0.2.63.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bizydraft-0.2.63.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
-bizydraft-0.2.63.dist-info/RECORD,,
+bizydraft-0.2.65.dist-info/METADATA,sha256=C4ZX1SbyFzrB-6ABgnyvncqMI87fQqOAnJ-rtt8lSbM,162
+bizydraft-0.2.65.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bizydraft-0.2.65.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
+bizydraft-0.2.65.dist-info/RECORD,,