mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-26 19:19:53 +00:00
24 lines
743 B
Python
24 lines
743 B
Python
from __future__ import annotations
|
|
|
|
from aiohttp import web
|
|
|
|
from typing import TYPE_CHECKING
|
|
if TYPE_CHECKING:
|
|
from comfy_api.latest._node_replace import NodeReplace
|
|
|
|
REGISTERED_NODE_REPLACEMENTS: dict[str, list[NodeReplace]] = {}
|
|
|
|
def register_node_replacement(node_replace: NodeReplace):
|
|
REGISTERED_NODE_REPLACEMENTS.setdefault(node_replace.old_node_id, []).append(node_replace)
|
|
|
|
def registered_as_dict():
|
|
return {
|
|
k: [v.as_dict() for v in v_list] for k, v_list in REGISTERED_NODE_REPLACEMENTS.items()
|
|
}
|
|
|
|
class NodeReplaceManager:
|
|
def add_routes(self, routes):
|
|
@routes.get("/node_replacements")
|
|
async def get_node_replacements(request):
|
|
return web.json_response(registered_as_dict())
|