80 lines
3.0 KiB
Python
80 lines
3.0 KiB
Python
import glob
|
|
import os
|
|
import shlex
|
|
import shutil
|
|
|
|
from ...utils.logging import debug_log
|
|
from ...utils.process import get_python_executable
|
|
|
|
|
|
class LaunchCommandBuilder:
|
|
"""Build command-lines for launching worker ComfyUI processes."""
|
|
|
|
def _find_windows_terminal(self):
|
|
"""Find Windows Terminal executable."""
|
|
possible_paths = [
|
|
os.path.expandvars(r"%LOCALAPPDATA%\Microsoft\WindowsApps\wt.exe"),
|
|
os.path.expandvars(r"%PROGRAMFILES%\WindowsApps\Microsoft.WindowsTerminal_*\wt.exe"),
|
|
"wt.exe",
|
|
]
|
|
|
|
for path in possible_paths:
|
|
if os.path.exists(path):
|
|
return path
|
|
if "*" in path:
|
|
matches = glob.glob(path)
|
|
if matches:
|
|
return matches[0]
|
|
|
|
wt_path = shutil.which("wt")
|
|
if wt_path:
|
|
return wt_path
|
|
return None
|
|
|
|
def build_launch_command(self, worker_config, comfy_root):
|
|
"""Build the command to launch a worker."""
|
|
main_py = os.path.join(comfy_root, "main.py")
|
|
|
|
if os.path.exists(main_py):
|
|
cmd = [
|
|
get_python_executable(),
|
|
main_py,
|
|
"--port",
|
|
str(worker_config["port"]),
|
|
"--enable-cors-header",
|
|
]
|
|
debug_log(f"Using main.py: {main_py}")
|
|
else:
|
|
error_msg = f"Could not find main.py in {comfy_root}\n"
|
|
error_msg += f"Searched for: {main_py}\n"
|
|
error_msg += f"Directory contents of {comfy_root}:\n"
|
|
try:
|
|
if os.path.exists(comfy_root):
|
|
files = os.listdir(comfy_root)[:20]
|
|
error_msg += " " + "\n ".join(files)
|
|
if len(os.listdir(comfy_root)) > 20:
|
|
error_msg += f"\n ... and {len(os.listdir(comfy_root)) - 20} more files"
|
|
else:
|
|
error_msg += f" Directory {comfy_root} does not exist!"
|
|
except Exception as exc:
|
|
error_msg += f" Error listing directory: {exc}"
|
|
|
|
error_msg += "\n\nPossible solutions:\n"
|
|
error_msg += "1. Check if ComfyUI is installed in a different location\n"
|
|
error_msg += "2. For Docker: ComfyUI might be in /ComfyUI or /app\n"
|
|
error_msg += "3. Ensure the custom node is installed in the correct location\n"
|
|
raise RuntimeError(error_msg)
|
|
|
|
if worker_config.get("extra_args"):
|
|
raw_args = worker_config["extra_args"].strip()
|
|
if raw_args:
|
|
extra_args_list = shlex.split(raw_args)
|
|
forbidden_chars = set(";|>&<`$()[]{}*!?")
|
|
for arg in extra_args_list:
|
|
if any(char in forbidden_chars for char in arg):
|
|
forbidden = "".join(forbidden_chars)
|
|
raise ValueError(f"Invalid characters in extra_args: {arg}. Forbidden: {forbidden}")
|
|
cmd.extend(extra_args_list)
|
|
|
|
return cmd
|