mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-18 14:34:23 +01:00
15c6b0c1c3
* Implement directory-based plugin system * Fix Selenium test --------- Co-authored-by: Nicholas Tindle <nick@ntindle.com> Co-authored-by: Merwane Hamadi <merwanehamadi@gmail.com>
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
import os
|
|
import subprocess
|
|
import sys
|
|
import zipfile
|
|
from glob import glob
|
|
from pathlib import Path
|
|
|
|
|
|
def install_plugin_dependencies():
|
|
"""
|
|
Installs dependencies for all plugins in the plugins dir.
|
|
|
|
Args:
|
|
None
|
|
|
|
Returns:
|
|
None
|
|
"""
|
|
plugins_dir = Path(os.getenv("PLUGINS_DIR", "plugins"))
|
|
|
|
# Install zip-based plugins
|
|
for plugin in plugins_dir.glob("*.zip"):
|
|
with zipfile.ZipFile(str(plugin), "r") as zfile:
|
|
try:
|
|
basedir = zfile.namelist()[0]
|
|
basereqs = os.path.join(basedir, "requirements.txt")
|
|
extracted = zfile.extract(basereqs, path=plugins_dir)
|
|
subprocess.check_call(
|
|
[sys.executable, "-m", "pip", "install", "-r", extracted]
|
|
)
|
|
os.remove(extracted)
|
|
os.rmdir(os.path.join(plugins_dir, basedir))
|
|
except KeyError:
|
|
continue
|
|
|
|
# Install directory-based plugins
|
|
for requirements_file in glob(f"{plugins_dir}/*/requirements.txt"):
|
|
subprocess.check_call(
|
|
[sys.executable, "-m", "pip", "install", "-r", requirements_file],
|
|
stdout=subprocess.DEVNULL,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
install_plugin_dependencies()
|