FilesExpand file tree

master

/

extra_config.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

34 lines (33 loc) · 1.34 KB

master

/

extra_config.py

Copy path

Top

File metadata and controls

• Code
• Blame

34 lines (33 loc) · 1.34 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

import os

import yaml

import folder_paths

import logging

def load_extra_path_config(yaml_path):

with open(yaml_path, 'r', encoding='utf-8') as stream:

config = yaml.safe_load(stream)

yaml_dir = os.path.dirname(os.path.abspath(yaml_path))

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))

if not os.path.isabs(base_path):

base_path = os.path.abspath(os.path.join(yaml_dir, 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:

full_path = os.path.join(base_path, full_path)

elif not os.path.isabs(full_path):

full_path = os.path.abspath(os.path.join(yaml_dir, y))

normalized_path = os.path.normpath(full_path)

logging.info("Adding extra search path {} {}".format(x, normalized_path))

folder_paths.add_model_folder_path(x, normalized_path, is_default)