45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
|
|
import os
|
|
import shutil
|
|
import inspect
|
|
from server import PromptServer
|
|
|
|
def get_extension_name():
|
|
return "ComfyUI_TextAsset"
|
|
|
|
def is_logging_enabled():
|
|
return True
|
|
|
|
def log(message, type=None, always=False, name=None):
|
|
if not always and not is_logging_enabled():
|
|
return
|
|
|
|
if type is not None:
|
|
message = f"[{type}] {message}"
|
|
|
|
if name is None:
|
|
name = get_extension_name()
|
|
|
|
print(f"({name}) {message}")
|
|
|
|
def get_ext_dir(subpath=None, mkdir=False):
|
|
dir = os.path.dirname(__file__)
|
|
if subpath is not None:
|
|
dir = os.path.join(dir, subpath)
|
|
|
|
dir = os.path.abspath(dir)
|
|
|
|
if mkdir and not os.path.exists(dir):
|
|
os.makedirs(dir)
|
|
return dir
|
|
|
|
def init(check_imports=None):
|
|
if check_imports is not None:
|
|
import importlib.util
|
|
for imp in check_imports:
|
|
spec = importlib.util.find_spec(imp)
|
|
if spec is None:
|
|
log(f"{imp} is required, please check requirements are installed.",
|
|
type="ERROR", always=True)
|
|
return False
|
|
return True |