19 lines
657 B
Python
19 lines
657 B
Python
import glob
|
|
import importlib
|
|
import os
|
|
|
|
from .captures import CAPTURE_FIELD_LIST
|
|
from .samplers import SAMPLERS
|
|
|
|
# load CAPTURE_FIELD_LIST and SAMPLERS in ext folder
|
|
dir_name = os.path.dirname(os.path.abspath(__file__))
|
|
for module_path in glob.glob(dir_name + "/ext/*.py"):
|
|
module_name = os.path.basename(module_path)
|
|
module_name = os.path.splitext(module_name)[0]
|
|
package_name = (
|
|
f"custom_nodes.ComfyUI-SaveImageWithMetaData.py.defs.ext.{module_name}"
|
|
)
|
|
module = importlib.import_module(package_name)
|
|
CAPTURE_FIELD_LIST.update(getattr(module, "CAPTURE_FIELD_LIST", {}))
|
|
SAMPLERS.update(getattr(module, "SAMPLERS", {}))
|