Claude
529f97e87e
Added one-line module docstrings to config.py, file_utils.py, load_description_node.py, and random_line_from_text_node.py to improve navigability and IDE support. https://claude.ai/code/session_01JBWyFr5shvPXUPmZkX3Nhg
68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
"""ComfyUI custom node for loading text description files from a dropdown."""
|
|
import os
|
|
import time
|
|
|
|
try:
|
|
from .config import DESCRIPTION_PATH, FALLBACK_FILE
|
|
from .file_utils import read_text_file, normalize_description_path, list_all_description_files, _ensure_example_exists
|
|
except ImportError:
|
|
from config import DESCRIPTION_PATH, FALLBACK_FILE
|
|
from file_utils import read_text_file, normalize_description_path, list_all_description_files, _ensure_example_exists
|
|
|
|
# Ensure base folder exists
|
|
os.makedirs(DESCRIPTION_PATH, exist_ok=True)
|
|
|
|
# Create example once at import time if needed
|
|
_ensure_example_exists()
|
|
|
|
|
|
class LoadDescriptionNode:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
files = list_all_description_files()
|
|
if not files:
|
|
files = ["examples/example"]
|
|
default_file = files[0]
|
|
return {
|
|
"required": {
|
|
# Dropdown with all files
|
|
"file_path": (files, {"default": default_file}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("STRING", "STRING")
|
|
RETURN_NAMES = ("description_text", "description_name")
|
|
FUNCTION = "load_description"
|
|
CATEGORY = "Feroc"
|
|
|
|
@staticmethod
|
|
def _resolve_file_path(file_path: str) -> str:
|
|
"""Return file_path or fall back to the first available description file."""
|
|
if not file_path or file_path == "undefined":
|
|
files = list_all_description_files()
|
|
return files[0] if files else FALLBACK_FILE
|
|
return file_path
|
|
|
|
@classmethod
|
|
def IS_CHANGED(cls, file_path=None, **kwargs):
|
|
"""Re-execute when content changes based on mtime and size."""
|
|
try:
|
|
abs_path = normalize_description_path(cls._resolve_file_path(file_path))
|
|
st = os.stat(abs_path)
|
|
return f"{st.st_mtime_ns}:{st.st_size}"
|
|
except Exception:
|
|
return str(time.time())
|
|
|
|
def load_description(self, file_path: str, **kwargs) -> tuple[str, str]:
|
|
"""Load description text from the given file path.
|
|
|
|
Returns:
|
|
Tuple of (file content, filename without extension). Returns ('', '') on error.
|
|
"""
|
|
full_file_path = normalize_description_path(self._resolve_file_path(file_path))
|
|
text = read_text_file(full_file_path)
|
|
if not text and not os.path.exists(full_file_path):
|
|
return ("", "")
|
|
name = os.path.splitext(os.path.basename(full_file_path))[0]
|
|
return (text, name)
|