27 lines
1.1 KiB
Python
27 lines
1.1 KiB
Python
import subprocess
|
|
import sys
|
|
import os
|
|
|
|
requset_path = os.path.join(os.path.dirname(__file__), "./requirements.txt")
|
|
def install_requirements():
|
|
try:
|
|
# 检查是否已经安装了所有依赖项
|
|
with open(requset_path, 'r') as f:
|
|
requirements = f.read().splitlines()
|
|
|
|
for requirement in requirements:
|
|
try:
|
|
# 提取纯包名，去除版本号和其他特殊字符
|
|
package_name = requirement.split('==')[0].split('>')[0].split('<')[0].split('~')[0].strip()
|
|
# print(f"{package_name} 正在检查...")
|
|
if package_name == 'uuid7':
|
|
from uuid_extensions import uuid7
|
|
else:
|
|
__import__(package_name)
|
|
except ImportError:
|
|
print(f"{requirement} 未安装，正在安装...")
|
|
subprocess.check_call([sys.executable, '-m', 'pip', 'install', requirement])
|
|
|
|
print("WeiLin-Comfyui-Tools ==> 所有依赖项已安装。")
|
|
except Exception as e:
|
|
print(f"WeiLin-Comfyui-Tools ==> 安装依赖项时出错: {e}") |