From 5eb2c119bec0d5cb0ddd560590865b5a28e8a382 Mon Sep 17 00:00:00 2001 From: "dr.lt.data" Date: Wed, 26 Jul 2023 14:27:07 +0900 Subject: [PATCH] add colab-depedencies script --- scripts/colab-dependencies.py | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 scripts/colab-dependencies.py diff --git a/scripts/colab-dependencies.py b/scripts/colab-dependencies.py new file mode 100644 index 00000000..76d25153 --- /dev/null +++ b/scripts/colab-dependencies.py @@ -0,0 +1,34 @@ +import os +import subprocess + + +def get_enabled_subdirectories_with_files(base_directory): + subdirs_with_files = [] + for subdir in os.listdir(base_directory): + full_path = os.path.join(base_directory, subdir) + if os.path.isdir(full_path) and not subdir.endswith(".disabled") and not subdir.startswith('.') and subdir != '__pycache__': + print(f"## Install dependencies for '{subdir}'") + requirements_file = os.path.join(full_path, "requirements.txt") + install_script = os.path.join(full_path, "install.py") + + if os.path.isfile(requirements_file) and os.path.isfile(install_script): + subdirs_with_files.append((full_path, requirements_file, install_script)) + + return subdirs_with_files + + +def install_requirements(requirements_file_path): + subprocess.run(["pip", "install", "-r", requirements_file_path]) + + +def run_install_script(install_script_path): + subprocess.run(["python", install_script_path]) + + +custom_nodes_directory = "custom_nodes" +subdirs_with_files = get_enabled_subdirectories_with_files(custom_nodes_directory) + + +for subdir, requirements_file, install_script in subdirs_with_files: + install_requirements(requirements_file) + run_install_script(install_script)