comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-07 22:30:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: pass filtering fields to frontend instead of server-side filtering

Browse Source 
Remove server-side includeOnDistributions filtering and fileByDistribution
resolution. Instead, forward requiresCustomNodes and includeOnDistributions
in the info object so the frontend handles filtering client-side.

Amp-Thread-ID: https://ampcode.com/threads/T-019c6f43-6212-7308-bea6-bfc35a486cbf
This commit is contained in:
bymyself
2026-02-20 02:21:55 -08:00
parent 6797e20bcf
commit 44dbb093c3
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
app/subgraph_manager.py
View File

@@ -42,7 +42,6 @@ class SubgraphManager:
def __init__(self):
self.cached_custom_node_subgraphs: dict[SubgraphEntry] | None = None
self.cached_blueprint_subgraphs: dict[SubgraphEntry] | None = None
self.distribution = os.environ.get("DISTRIBUTION", "localhost")
def _create_entry(self, file: str, source: str, node_pack: str) -> tuple[str, SubgraphEntry]:
"""Create a subgraph entry from a file path. Expects normalized path (forward slashes)."""
@@ -121,12 +120,7 @@ class SubgraphManager:
logging.warning("Blueprint entry missing 'name' in category '%s', skipping", module_name)
continue
include_on = blueprint.get("includeOnDistributions")
if include_on is not None and self.distribution not in include_on:
continue
file_by_dist = blueprint.get("fileByDistribution", {})
filename = file_by_dist.get(self.distribution, f"{name}.json")
filename = f"{name}.json"
filepath = os.path.realpath(os.path.join(blueprints_dir, filename))
if not filepath.startswith(os.path.realpath(blueprints_dir) + os.sep):
logging.warning("Blueprint path escapes blueprints directory: %s", filepath)
@@ -137,6 +131,13 @@ class SubgraphManager:
continue
entry_id, entry = self._create_entry(filepath, Source.templates, module_name)
info = entry["info"]
include_on = blueprint.get("includeOnDistributions")
if include_on is not None:
info["includeOnDistributions"] = include_on
requires = blueprint.get("requiresCustomNodes")
if requires is not None:
info["requiresCustomNodes"] = requires
subgraphs_dict[entry_id] = entry
elif os.path.exists(blueprints_dir):
logging.warning("No blueprint index.json found at %s, falling back to glob", index_path)