bizydraft 0.2.78.dev20251117024007__py3-none-any.whl → 0.2.82.dev20251209023307__py3-none-any.whl

bizydraft/hijack_routes.py

@@ -1,3 +1,5 @@
+import json
+
 from aiohttp import web
 from loguru import logger
 
@@ -51,6 +53,31 @@ def hijack_routes_pre_add_routes():
 
         return middleware_handler
 
+    # 覆盖 /settings 响应，修改设置 NodeIdBadgeMode 为 ShowAll显示id，将 LinkRenderMode 为（Spline=2），显示连线
+    async def settings_response_middleware(app, handler):
+        async def middleware_handler(request):
+            if request.method == "GET":
+                p = request.path
+                if p == "/settings" or p == "/api/settings":
+                    resp = await handler(request)
+                    if getattr(resp, "content_type", None) == "application/json":
+                        body_bytes = getattr(resp, "body", b"")
+                        charset = getattr(resp, "charset", None) or "utf-8"
+                        payload = json.loads(body_bytes.decode(charset))
+                        if payload.get("Comfy.NodeBadge.NodeIdBadgeMode") != "ShowAll":
+                            payload["Comfy.NodeBadge.NodeIdBadgeMode"] = "ShowAll"
+                        if payload.get("Comfy.LinkRenderMode") != 2:
+                            payload["Comfy.LinkRenderMode"] = 2
+                        if payload.get("Comfy.VueNodes.Enabled") != False:
+                            payload["Comfy.VueNodes.Enabled"] = False
+                        return web.json_response(
+                            payload, status=getattr(resp, "status", 200)
+                        )
+                    return resp
+            return await handler(request)
+
+        return middleware_handler
+
     async def access_control_middleware(app, handler):
         base_white_list = [
             "/prompt",
@@ -98,7 +125,7 @@ def hijack_routes_pre_add_routes():
 
         return middleware_handler
 
-    app.middlewares.extend([custom_business_middleware])
+    app.middlewares.extend([custom_business_middleware, settings_response_middleware])
 
     logger.info("Optimized middleware setup complete.")
 

bizydraft/oss_utils.py

@@ -312,9 +312,17 @@ async def upload_mask(request):
         if "http" in original_subfolder:
             # subfolder 中包含 URL 基础路径
             original_subfolder = original_subfolder[original_subfolder.find("http") :]
-            original_subfolder = unquote(original_subfolder).replace(
-                "https:/", "https://"
-            )
+            original_subfolder = unquote(original_subfolder)
+            if "https:/" in original_subfolder and not original_subfolder.startswith(
+                "https://"
+            ):
+                original_subfolder = original_subfolder.replace(
+                    "https:/", "https://", 1
+                )
+            if "http:/" in original_subfolder and not original_subfolder.startswith(
+                "http://"
+            ):
+                original_subfolder = original_subfolder.replace("http:/", "http://", 1)
             original_url = f"{original_subfolder}/{original_filename}"
         elif original_filename.startswith(http_prefix_options):
             # filename 本身就是完整 URL