mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-06 16:27:04 +08:00
82 lines
2.7 KiB
Python
82 lines
2.7 KiB
Python
import os
|
|
from typing import Optional
|
|
|
|
import tomllib
|
|
|
|
from pydantic import ValidationError
|
|
import logging
|
|
|
|
from comfy_config.types import (
|
|
ProjectConfig,
|
|
PyProjectConfig,
|
|
)
|
|
|
|
"""
|
|
Extract configuration from a custom node directory's pyproject.toml file.
|
|
|
|
This function reads and parses the pyproject.toml file in the specified directory
|
|
to extract project and ComfyUI-specific configuration information. If no
|
|
pyproject.toml file is found, it creates a minimal configuration using the
|
|
folder name as the project name.
|
|
|
|
Args:
|
|
path (str): Path to the directory containing the pyproject.toml file.
|
|
If pyproject.toml doesn't exist, the folder name will be used
|
|
as the default project name.
|
|
|
|
Returns:
|
|
Optional[PyProjectConfig]: A PyProjectConfig object containing:
|
|
- project: Basic project information (name, version, dependencies, etc.)
|
|
- tool_comfy: ComfyUI-specific configuration (publisher_id, models, etc.)
|
|
Returns None if configuration extraction fails.
|
|
|
|
Notes:
|
|
- If pyproject.toml is missing, creates a default config with folder name
|
|
|
|
Example:
|
|
>>> from comfy_config import config_parser
|
|
>>> custom_node_dir = os.path.dirname(os.path.realpath(__file__))
|
|
>>> project_config = config_parser.extract_node_configuration(custom_node_dir)
|
|
>>> print(project_config.project.name) # "my_custom_node" or name from pyproject.toml
|
|
>>> nodes.EXTENSION_WEB_DIRS[project_config.project.name] = js_dir
|
|
"""
|
|
|
|
|
|
def extract_node_configuration(path) -> Optional[PyProjectConfig]:
|
|
folder_name = os.path.basename(path)
|
|
toml_path = os.path.join(path, "pyproject.toml")
|
|
|
|
if not os.path.isfile(toml_path):
|
|
logging.warning(
|
|
"No pyproject.toml file found in the current directory, will use custom node folder name as project name as default.")
|
|
|
|
try:
|
|
project = ProjectConfig(name=folder_name)
|
|
return PyProjectConfig(project=project)
|
|
except ValidationError as e:
|
|
logging.error(f"Failed to create default configuration: {e}")
|
|
return None
|
|
|
|
try:
|
|
with open(toml_path, "rb") as f:
|
|
data = tomllib.load(f)
|
|
except Exception as e:
|
|
logging.error(f"Failed to read pyproject.toml: {e}")
|
|
return None
|
|
|
|
try:
|
|
config_data = {
|
|
"project": data.get("project", {}),
|
|
"tool_comfy": data.get("tool", {}).get("comfy", {})
|
|
}
|
|
|
|
return PyProjectConfig(**config_data)
|
|
|
|
except ValidationError as e:
|
|
logging.error(f"Validation error while parsing configuration: {e}")
|
|
logging.error(f"Validation details: {e.errors()}")
|
|
return None
|
|
except Exception as e:
|
|
logging.error(f"Unexpected error while parsing configuration: {e}")
|
|
return None
|