Ethanfel
980f406573
Remote-aware LoRA Manager nodes that fetch metadata via HTTP from a remote Docker instance while loading LoRA files from local NFS/SMB mounts. Includes reverse-proxy middleware for transparent web UI access. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
"""Configuration for ComfyUI-LM-Remote."""
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import logging
|
|
import os
|
|
from pathlib import Path
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
_PACKAGE_DIR = Path(__file__).resolve().parent
|
|
_CONFIG_FILE = _PACKAGE_DIR / "config.json"
|
|
|
|
|
|
class RemoteConfig:
|
|
"""Holds remote LoRA Manager connection settings."""
|
|
|
|
def __init__(self):
|
|
self.remote_url: str = ""
|
|
self.timeout: int = 30
|
|
self.path_mappings: dict[str, str] = {}
|
|
self._load()
|
|
|
|
# ------------------------------------------------------------------
|
|
def _load(self):
|
|
# Environment variable takes priority
|
|
env_url = os.environ.get("LM_REMOTE_URL", "")
|
|
env_timeout = os.environ.get("LM_REMOTE_TIMEOUT", "")
|
|
|
|
# Load config.json defaults
|
|
if _CONFIG_FILE.exists():
|
|
try:
|
|
with open(_CONFIG_FILE, "r", encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
self.remote_url = data.get("remote_url", "")
|
|
self.timeout = int(data.get("timeout", 30))
|
|
self.path_mappings = data.get("path_mappings", {})
|
|
except Exception as exc:
|
|
logger.warning("[LM-Remote] Failed to read config.json: %s", exc)
|
|
|
|
# Env overrides
|
|
if env_url:
|
|
self.remote_url = env_url
|
|
if env_timeout:
|
|
self.timeout = int(env_timeout)
|
|
|
|
# Strip trailing slash
|
|
self.remote_url = self.remote_url.rstrip("/")
|
|
|
|
@property
|
|
def is_configured(self) -> bool:
|
|
return bool(self.remote_url)
|
|
|
|
def map_path(self, remote_path: str) -> str:
|
|
"""Apply remote->local path prefix mappings."""
|
|
for remote_prefix, local_prefix in self.path_mappings.items():
|
|
if remote_path.startswith(remote_prefix):
|
|
return local_prefix + remote_path[len(remote_prefix):]
|
|
return remote_path
|
|
|
|
|
|
remote_config = RemoteConfig()
|