Maxed-Out-99
3ca3859588
Strip stray BOM/hidden characters from multiple Python and web source files to normalize file encoding and avoid lint/parse issues. Also bump package version from 1.8.2 to 1.8.3 in pyproject.toml. No functional code changes were made.
89 lines
2.4 KiB
Python
89 lines
2.4 KiB
Python
import os
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
from .utils import load_json_file, path_exists, save_json_file
|
|
|
|
THIS_DIR = Path(__file__).resolve().parent
|
|
MAXEDOUT_ROOT = THIS_DIR.parent
|
|
USERDATA = MAXEDOUT_ROOT / "userdata" / "loraloader_mxd"
|
|
LEGACY_USERDATA = MAXEDOUT_ROOT.parent / "ComfyUI-LoraLoader-MXD" / "userdata"
|
|
MIGRATION_SENTINEL = USERDATA / ".migrated_from_standalone"
|
|
|
|
|
|
def _log(message: str):
|
|
print(f"[ComfyUI-MaxedOut][LoraLoader-MXD] {message}")
|
|
|
|
|
|
def _migrate_legacy_userdata_once():
|
|
if MIGRATION_SENTINEL.exists():
|
|
return
|
|
if not LEGACY_USERDATA.exists():
|
|
return
|
|
|
|
copied = 0
|
|
skipped = 0
|
|
for src in LEGACY_USERDATA.rglob("*"):
|
|
if not src.is_file():
|
|
continue
|
|
rel = src.relative_to(LEGACY_USERDATA)
|
|
dest = USERDATA / rel
|
|
dest.parent.mkdir(parents=True, exist_ok=True)
|
|
if dest.exists():
|
|
skipped += 1
|
|
continue
|
|
shutil.copy2(src, dest)
|
|
copied += 1
|
|
|
|
USERDATA.mkdir(parents=True, exist_ok=True)
|
|
MIGRATION_SENTINEL.write_text("ok\n", encoding="utf-8")
|
|
_log(f"Migrated legacy userdata files: copied={copied}, skipped_existing={skipped}")
|
|
|
|
|
|
_migrate_legacy_userdata_once()
|
|
|
|
|
|
def read_userdata_file(rel_path: str):
|
|
"""Reads a file from the userdata directory."""
|
|
file_path = clean_path(rel_path)
|
|
if path_exists(file_path):
|
|
with open(file_path, 'r', encoding='UTF-8') as file:
|
|
return file.read()
|
|
return None
|
|
|
|
|
|
def save_userdata_file(rel_path: str, content: str):
|
|
"""Saves a file from the userdata directory."""
|
|
file_path = clean_path(rel_path)
|
|
os.makedirs(os.path.dirname(file_path), exist_ok=True)
|
|
with open(file_path, 'w+', encoding='UTF-8') as file:
|
|
file.write(content)
|
|
|
|
|
|
def delete_userdata_file(rel_path: str):
|
|
"""Deletes a file from the userdata directory."""
|
|
file_path = clean_path(rel_path)
|
|
if os.path.isfile(file_path):
|
|
os.remove(file_path)
|
|
|
|
|
|
def read_userdata_json(rel_path: str):
|
|
"""Reads a json file from the userdata directory."""
|
|
file_path = clean_path(rel_path)
|
|
return load_json_file(file_path)
|
|
|
|
|
|
def save_userdata_json(rel_path: str, data: dict):
|
|
"""Saves a json file from the userdata directory."""
|
|
file_path = clean_path(rel_path)
|
|
return save_json_file(file_path, data)
|
|
|
|
|
|
def clean_path(rel_path: str):
|
|
"""Cleans a relative path by splitting on forward slash and os.path.joining."""
|
|
cleaned = USERDATA
|
|
paths = rel_path.split('/')
|
|
for path in paths:
|
|
cleaned = cleaned / path
|
|
return str(cleaned)
|