Files
M1kep-ComfyLiterals/startup_utils.py
T
2023-07-22 00:10:46 -07:00

30 lines
1.1 KiB
Python

import os
from pathlib import Path
import folder_paths
def symlink_web_dir(local_path, extension_name):
comfy_web_ext_root = Path(os.path.join(folder_paths.base_path, "web", "extensions"))
target_dir = Path(os.path.join(comfy_web_ext_root, extension_name))
extension_path = Path(__file__).parent.resolve()
if target_dir.exists():
print(f"Web extensions folder found at {target_dir}")
elif comfy_web_ext_root.exists():
try:
os.symlink((os.path.join(extension_path, local_path)), target_dir)
except OSError as e: # OSError
print(
f"Error:\n{e}\n"
f"Failed to create symlink to {target_dir}. Please copy the folder manually.\n"
f"Source: {os.path.join(extension_path, local_path)}\n"
f"Target: {target_dir}"
)
except Exception as e:
print(f"Unexpected error:\n{e}")
else:
print(
f"Failed to find comfy root automatically, please copy the folder {os.path.join(extension_path, 'web')} manually in the web/extensions folder of ComfyUI"
)