108 lines
4.0 KiB
Python
108 lines
4.0 KiB
Python
# AAA_Metadata_System/__init__.py
|
|
import importlib.util
|
|
import os
|
|
import sys
|
|
import shutil
|
|
|
|
# Add this import for ComfyUI paths
|
|
try:
|
|
import folder_paths
|
|
except ImportError:
|
|
folder_paths = None
|
|
|
|
# Make this package importable as both its folder name and as 'AAA_Metadata_System'
|
|
# This allows: from AAA_Metadata_System import MetadataService
|
|
sys.modules['AAA_Metadata_System'] = sys.modules.get(__name__, sys.modules[__name__])
|
|
|
|
# Export core functionality for easy access
|
|
from .eric_metadata.service import MetadataService
|
|
from .eric_metadata.handlers.base import BaseHandler
|
|
from .eric_metadata.handlers.xmp import XMPSidecarHandler
|
|
from .eric_metadata.handlers.embedded import EmbeddedMetadataHandler
|
|
from .eric_metadata.handlers.txt import TxtFileHandler
|
|
from .eric_metadata.handlers.db import DatabaseHandler
|
|
|
|
# Initialize node mappings
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
# Get the nodes directory
|
|
def get_nodes_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
|
|
|
|
# Copy web extensions to ComfyUI's main extensions directory
|
|
def install_web_extensions():
|
|
if not folder_paths:
|
|
print("[AAA_Metadata_System] Warning: folder_paths not available, web extensions may not work")
|
|
return
|
|
|
|
try:
|
|
# Get the path to ComfyUI's web/extensions directory
|
|
extension_path = os.path.join(os.path.dirname(folder_paths.__file__), "web", "extensions")
|
|
my_extension_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), "web")
|
|
|
|
# Create MetadataSystem subfolder in ComfyUI extensions
|
|
metadata_extension_path = os.path.join(extension_path, "MetadataSystem")
|
|
os.makedirs(metadata_extension_path, exist_ok=True)
|
|
|
|
# Clean up existing files in the MetadataSystem folder
|
|
if os.path.exists(metadata_extension_path):
|
|
for file in os.listdir(metadata_extension_path):
|
|
file_path = os.path.join(metadata_extension_path, file)
|
|
if os.path.isfile(file_path):
|
|
os.remove(file_path)
|
|
|
|
# Copy our extension files to ComfyUI's extensions/MetadataSystem directory
|
|
if os.path.exists(my_extension_path):
|
|
for file in os.listdir(my_extension_path):
|
|
if file.endswith('.js'):
|
|
src = os.path.join(my_extension_path, file)
|
|
dst = os.path.join(metadata_extension_path, file)
|
|
print(f"[AAA_Metadata_System] Copying extension file: {file}")
|
|
shutil.copy2(src, dst)
|
|
|
|
print(f"[AAA_Metadata_System] Web extensions installed to: {metadata_extension_path}")
|
|
|
|
except Exception as e:
|
|
print(f"[AAA_Metadata_System] Error installing web extensions: {e}")
|
|
|
|
# Install web extensions
|
|
install_web_extensions()
|
|
|
|
# Import nodes from the nodes directory
|
|
nodes_dir = get_nodes_dir("nodes")
|
|
for file in os.listdir(nodes_dir):
|
|
if not file.endswith(".py") or file.startswith("__"):
|
|
continue
|
|
|
|
name = os.path.splitext(file)[0]
|
|
try:
|
|
# Import the module
|
|
imported_module = importlib.import_module(".nodes.{}".format(name), __name__)
|
|
|
|
# Extract and update node mappings
|
|
if hasattr(imported_module, 'NODE_CLASS_MAPPINGS'):
|
|
NODE_CLASS_MAPPINGS.update(imported_module.NODE_CLASS_MAPPINGS)
|
|
|
|
if hasattr(imported_module, 'NODE_DISPLAY_NAME_MAPPINGS'):
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(imported_module.NODE_DISPLAY_NAME_MAPPINGS)
|
|
|
|
print(f"Loaded node module: {name}")
|
|
except Exception as e:
|
|
print(f"Error loading node module {name}: {str(e)}")
|
|
|
|
# Version info
|
|
__version__ = "0.1.0"
|
|
|
|
# Add web directory for UI components if needed
|
|
WEB_DIRECTORY = "./web"
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"] |