alphard/efficiency-nodes-comfyui
1
0
Fork 0
mirror of https://github.com/jags111/efficiency-nodes-comfyui.git synced 2026-03-21 21:22:13 -03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #82 from LucianoCirino/auto-requirements-installer-update

Browse Source 
Auto Package Installer Changes
This commit is contained in:
TSC
2023-08-07 22:27:36 -05:00
committed by GitHub
parent 66588a5959 1b882e0f11
commit 82ee7ed735
1 changed files with 22 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
tsc_utils.py
View File

@@ -460,24 +460,35 @@ def global_preview_method():
return args.preview_method return args.preview_method
#----------------------------------------------------------------------------------------------------------------------- #-----------------------------------------------------------------------------------------------------------------------
# Auto install Efficiency Nodes Python packages # Auto install Efficiency Nodes Python package dependencies
import subprocess import subprocess
def install_packages(required_packages): # Note: Auto installer targets ComfyUI's python_embedded folder
installed_packages = packages(versions=False)
def install_packages(my_dir):
# Compute path to the target site-packages
target_dir = os.path.abspath(os.path.join(my_dir, '..', '..', '..', 'python_embeded', 'Lib', 'site-packages'))
embedded_python_exe = os.path.abspath(os.path.join(my_dir, '..', '..', '..', 'python_embeded', 'python.exe'))
# Load packages from requirements.txt
with open(os.path.join(my_dir, 'requirements.txt'), 'r') as f:
required_packages = [line.strip() for line in f if line.strip()]
installed_packages = packages(embedded_python_exe, versions=False)
for pkg in required_packages: for pkg in required_packages:
if pkg not in installed_packages: if pkg not in installed_packages:
print(f"\033[32mEfficiency Nodes:\033[0m Installing {pkg}...") print(f"\033[32mEfficiency Nodes:\033[0m Installing required package '{pkg}'...", end='', flush=True)
subprocess.check_call([sys.executable, '-m', 'pip', 'install', pkg]) subprocess.check_call(['pip', 'install', pkg, '--target=' + target_dir, '--no-warn-script-location',
print(f"\033[32mEfficiency Nodes:\033[0m Installed {pkg}!") '--disable-pip-version-check'], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
print(f"\r\033[32mEfficiency Nodes:\033[0m Installing required package '{pkg}'... Installed!", flush=True)
def packages(versions=False): def packages(python_exe, versions=False):
# Get packages of the embedded Python environment
return [(r.decode().split('==')[0] if not versions else r.decode()) for r in return [(r.decode().split('==')[0] if not versions else r.decode()) for r in
subprocess.check_output([sys.executable, '-m', 'pip', 'freeze']).split()] subprocess.check_output([python_exe, '-m', 'pip', 'freeze']).split()]
# Packages to install # Install missing packages
required_packages = ['simpleeval', 'websockets'] install_packages(my_dir)
install_packages(required_packages)
#----------------------------------------------------------------------------------------------------------------------- #-----------------------------------------------------------------------------------------------------------------------
# Auto install efficiency nodes web extension '\js\efficiency_nodes.js' to 'ComfyUI\web\extensions' # Auto install efficiency nodes web extension '\js\efficiency_nodes.js' to 'ComfyUI\web\extensions'