298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-06 23:37:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

account for multiple custom_nodes paths

Browse Source
This commit is contained in:
bezo97 2024-12-28 01:15:26 +01:00
parent 1ea319ee07
commit 1f583d2ef1
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/custom_node_manager.py
View File

@ -15,7 +15,11 @@ class CustomNodeManager:
@routes.get("/workflow_templates") @routes.get("/workflow_templates")
async def get_workflow_templates(request): async def get_workflow_templates(request):
"""Returns a web response that contains the map of custom_nodes names and their associated workflow templates. The ones without templates are omitted.""" """Returns a web response that contains the map of custom_nodes names and their associated workflow templates. The ones without templates are omitted."""
files = glob.glob(os.path.join(folder_paths.get_folder_paths("custom_nodes")[0], '*/example_workflows/*.json')) files = [
file
for folder in folder_paths.get_folder_paths("custom_nodes")
for file in glob.glob(os.path.join(folder, '*/example_workflows/*.json'))
]
workflow_templates_dict = {} # custom_nodes folder name -> example workflow names workflow_templates_dict = {} # custom_nodes folder name -> example workflow names
for file in files: for file in files:
custom_nodes_name = os.path.basename(os.path.dirname(os.path.dirname(file))) custom_nodes_name = os.path.basename(os.path.dirname(os.path.dirname(file)))