Upload all_nodes.py

all_nodes.py

@@ -0,0 +1,118 @@
+import os
+import subprocess
+
+def colored(text, color):
+    """
+    Функция для вывода цветного текста в консоль.
+    """
+    colors = {
+        'red': '\033[91m',
+        'green': '\033[92m',
+        'yellow': '\033[93m',
+        'blue': '\033[94m',
+        'magenta': '\033[95m',
+        'cyan': '\033[96m',
+        'white': '\033[97m',
+    }
+    return colors.get(color, '') + text + '\033[0m'
+
+# Путь к папке ComfyUI (относительный путь)
+comfyui_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "ComfyUI"))
+
+# Путь к папке python_embeded (относительный путь)
+python_embeded_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "python_embeded"))
+
+# Путь к папке с git (относительный путь)
+# git_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "git", "bin"))
+git_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "git"))
+git_bin_path = os.path.join(git_path, "bin")
+git_libexec_path = os.path.join(git_path, "libexec")
+
+# Обновление переменной PATH
+os.environ['PATH'] = f"{git_bin_path};{git_libexec_path};{os.environ['PATH']}"
+
+# Словарь с вариантами установки и их репозиториями
+install_options = {
+    "Базовые узлы": [
+        "https://github.com/ltdrdata/ComfyUI-Manager",
+        "https://github.com/crystian/ComfyUI-Crystools"
+    ],
+    "Расширенные узлы": [
+        "https://github.com/11cafe/comfyui-workspace-manager",
+        "https://github.com/hayden-fr/ComfyUI-Model-Manager",
+        "https://github.com/talesofai/comfyui-browser"
+    ],
+    "Все узлы": [
+        "https://github.com/comfyui/ComfyUI-Nodes",
+        "https://github.com/AUTOMATIC1111/stable-diffusion-webui",
+        "https://github.com/your_username/your_custom_node_repo",
+        "https://github.com/another_username/another_custom_node_repo",
+        # Добавьте сюда ссылки на другие репозитории
+    ],
+}
+
+# Проверка наличия папки с custom_nodes
+custom_nodes_path = os.path.join(comfyui_path, "custom_nodes")
+if not os.path.exists(custom_nodes_path):
+    os.makedirs(custom_nodes_path)
+
+# Меню выбора варианта установки
+while True:
+    print("Выберите вариант установки:")
+    for i, option in enumerate(install_options.keys()):
+        print(f"{i+1}. {option}")
+    choice = input("Введите номер варианта: ")
+
+    # Проверка ввода
+    try:
+        choice = int(choice)
+        if 1 <= choice <= len(install_options):
+            break
+        else:
+            print("Неверный номер варианта. Пожалуйста, введите число от 1 до", len(install_options))
+    except ValueError:
+        print("Неверный ввод. Пожалуйста, введите число.")
+
+# Получение списка репозиториев для выбранного варианта
+selected_repos = []
+if choice == 2:  # Выбран вариант 2
+    selected_repos.extend(install_options["Базовые узлы"])
+    selected_repos.extend(install_options["Расширенные узлы"])
+elif choice == 3:  # Выбран вариант 3
+    selected_repos.extend(install_options["Базовые узлы"])
+    selected_repos.extend(install_options["Расширенные узлы"])
+    selected_repos.extend(install_options["Все узлы"])
+else:
+    selected_repos.extend(install_options[list(install_options.keys())[choice - 1]])
+
+# Скачивание custom_nodes
+for repo_url in selected_repos:
+    repo_name = repo_url.split("/")[-1].split(".")[0]
+    repo_path = os.path.join(custom_nodes_path, repo_name)
+
+    # Проверка, была ли уже скачана репозитория
+    if os.path.exists(repo_path):
+        print(f"Репозитория {repo_name} уже скачана.")
+        continue
+
+    # Скачивание репозитории
+    print(colored(f"Скачивание {repo_name}...", 'cyan'))
+    # print(f"Скачивание {repo_name}...")
+    try:
+        subprocess.run([os.path.join(git_bin_path, "git"), "clone", repo_url, repo_path], check=True)
+    except subprocess.CalledProcessError as e:
+        print(f"Ошибка при скачивании {repo_name}: {e}")
+        continue
+
+    # Проверка наличия requirements.txt
+    requirements_path = os.path.join(repo_path, "requirements.txt")
+    if os.path.exists(requirements_path):
+        print(colored(f"Установка зависимостей для {repo_name}...", 'blue'))
+        # print(f"Установка зависимостей для {repo_name}...")
+        try:
+            subprocess.run([os.path.join(python_embeded_path, "python"), "-m", "pip", "install", "-r", requirements_path], check=True)
+        except subprocess.CalledProcessError as e:
+            print(f"Ошибка при установке зависимостей: {e}")
+
+print(colored("Скачивание custom_nodes завершено...", 'green'))
+# print("Скачивание custom_nodes завершено.")