alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-24 06:32:12 -03:00
Code Issues Packages Projects Releases Wiki Activity

fix(lora_manager): rename invalid hash folder removal to orphaned folders and update logging

Browse Source
This commit is contained in:
Will Miao
2025-09-18 15:09:32 +08:00
parent bdc86ddf15
commit bc4cd45fcb
1 changed files with 7 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
py/lora_manager.py
View File

@@ -368,7 +368,7 @@ class LoraManager:
total_folders_checked = 0 total_folders_checked = 0
empty_folders_removed = 0 empty_folders_removed = 0
invalid_hash_folders_removed = 0 orphaned_folders_removed = 0
# Scan the example images directory # Scan the example images directory
try: try:
@@ -392,9 +392,8 @@ class LoraManager:
# Check if folder name is a valid SHA256 hash (64 hex characters) # Check if folder name is a valid SHA256 hash (64 hex characters)
if len(folder_name) != 64 or not all(c in '0123456789abcdefABCDEF' for c in folder_name): if len(folder_name) != 64 or not all(c in '0123456789abcdefABCDEF' for c in folder_name):
logger.debug(f"Removing invalid hash folder: {folder_name}") # Skip non-hash folders to avoid deleting other content
await cls._remove_folder_safely(folder_path) logger.debug(f"Skipping non-hash folder: {folder_name}")
invalid_hash_folders_removed += 1
continue continue
# Check if hash exists in any of the scanners # Check if hash exists in any of the scanners
@@ -407,7 +406,7 @@ class LoraManager:
if not hash_exists: if not hash_exists:
logger.debug(f"Removing example images folder for deleted model: {folder_name}") logger.debug(f"Removing example images folder for deleted model: {folder_name}")
await cls._remove_folder_safely(folder_path) await cls._remove_folder_safely(folder_path)
invalid_hash_folders_removed += 1 orphaned_folders_removed += 1
continue continue
except Exception as e: except Exception as e:
@@ -421,11 +420,11 @@ class LoraManager:
return return
# Log final cleanup report # Log final cleanup report
total_removed = empty_folders_removed + invalid_hash_folders_removed total_removed = empty_folders_removed + orphaned_folders_removed
if total_removed > 0: if total_removed > 0:
logger.info(f"Example images cleanup completed: checked {total_folders_checked} folders, " logger.info(f"Example images cleanup completed: checked {total_folders_checked} folders, "
f"removed {empty_folders_removed} empty folders and {invalid_hash_folders_removed} " f"removed {empty_folders_removed} empty folders and {orphaned_folders_removed} "
f"folders for deleted/invalid models (total: {total_removed} removed)") f"folders for deleted models (total: {total_removed} removed)")
else: else:
logger.debug(f"Example images cleanup completed: checked {total_folders_checked} folders, " logger.debug(f"Example images cleanup completed: checked {total_folders_checked} folders, "
f"no cleanup needed") f"no cleanup needed")
@@ -470,11 +469,5 @@ class LoraManager:
try: try:
logger.info("LoRA Manager: Cleaning up services") logger.info("LoRA Manager: Cleaning up services")
# Close CivitaiClient gracefully
civitai_client = await ServiceRegistry.get_service("civitai_client")
if civitai_client:
await civitai_client.close()
logger.info("Closed CivitaiClient connection")
except Exception as e: except Exception as e:
logger.error(f"Error during cleanup: {e}", exc_info=True) logger.error(f"Error during cleanup: {e}", exc_info=True)