27 lines
888 B
Python
27 lines
888 B
Python
import sys
|
|
import subprocess
|
|
import importlib
|
|
|
|
def ensure_package(package, install_package_name=None):
|
|
# Try to import the package
|
|
try:
|
|
importlib.import_module(package)
|
|
except ImportError:
|
|
print(f"Package {package} is not installed. Installing now...")
|
|
|
|
if "python_embeded" in sys.executable or "python_embedded" in sys.executable:
|
|
pip_install = [sys.executable, "-s", "-m", "pip", "install"]
|
|
else:
|
|
pip_install = [sys.executable, "-m", "pip", "install"]
|
|
|
|
subprocess.check_call(pip_install + [install_package_name or package])
|
|
# else:
|
|
# print(f"Package {package} is already installed.")
|
|
|
|
ensure_package("dotenv", "python-dotenv")
|
|
ensure_package("argon2", "argon2-cffi")
|
|
|
|
|
|
from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|