Skip to content

Add support for custom load order of external nodes #4782

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 1 commit into from
Closed

Add support for custom load order of external nodes #4782

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 20 additions & 0 deletions nodes.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2027,11 +2027,31 @@ def init_external_custom_nodes():
base_node_names = set(NODE_CLASS_MAPPINGS.keys())
node_paths = folder_paths.get_folder_paths("custom_nodes")
node_import_times = []

# Path to the file specifying the load order for custom modules.
# Each line in the file should represent a module name.
module_order_path = 'custom_nodes/custom_load_order.txt'

# If the file exists, read module names from it; otherwise, use an empty list.
if os.path.exists(module_order_path):
with open(module_order_path, 'r') as file:
ordered_modules = [line.strip() for line in file]
else:
ordered_modules = []

# Create a dictionary that maps module names to their index in 'ordered_modules'.
# Modules not in this list will be assigned 'inf' to load them last.
order_map = {name: i for i, name in enumerate(ordered_modules)}

for custom_node_path in node_paths:
possible_modules = os.listdir(os.path.realpath(custom_node_path))
if "__pycache__" in possible_modules:
possible_modules.remove("__pycache__")

# Sort modules based on their order in 'order_map'.
# Modules without an entry are sorted to load after those listed.
possible_modules.sort(key=lambda x: order_map.get(x, float('inf')))

for possible_module in possible_modules:
module_path = os.path.join(custom_node_path, possible_module)
if os.path.isfile(module_path) and os.path.splitext(module_path)[1] != ".py": continue
Expand Down