| import os |
| import yaml |
| import folder_paths |
| import logging |
|
|
| def load_extra_path_config(yaml_path): |
| with open(yaml_path, 'r') as stream: |
| config = yaml.safe_load(stream) |
| for c in config: |
| conf = config[c] |
| if conf is None: |
| continue |
| base_path = None |
| if "base_path" in conf: |
| base_path = conf.pop("base_path") |
| base_path = os.path.expandvars(os.path.expanduser(base_path)) |
| is_default = False |
| if "is_default" in conf: |
| is_default = conf.pop("is_default") |
| for x in conf: |
| for y in conf[x].split("\n"): |
| if len(y) == 0: |
| continue |
| full_path = y |
| if base_path is not None: |
| full_path = os.path.join(base_path, full_path) |
| elif not os.path.isabs(full_path): |
| yaml_dir = os.path.dirname(os.path.abspath(yaml_path)) |
| full_path = os.path.abspath(os.path.join(yaml_dir, y)) |
| logging.info("Adding extra search path {} {}".format(x, full_path)) |
| folder_paths.add_model_folder_path(x, full_path, is_default) |
|
|
