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

feat: Add support for reading SHA256 from .sha256 file in get_file_info function

Browse Source
This commit is contained in:
Will Miao
2025-04-12 11:59:40 +08:00
parent ce30d067a6
commit c3192351da
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
py/utils/file_utils.py
View File

@@ -93,6 +93,17 @@ async def get_file_info(file_path: str, model_class: Type[BaseModelMetadata] = L
logger.debug(f"Using SHA256 from .json file for {file_path}") logger.debug(f"Using SHA256 from .json file for {file_path}")
except Exception as e: except Exception as e:
logger.error(f"Error reading .json file for {file_path}: {e}") logger.error(f"Error reading .json file for {file_path}: {e}")
# If SHA256 is still not found, check for a .sha256 file
if sha256 is None:
sha256_file = f"{os.path.splitext(file_path)[0]}.sha256"
if os.path.exists(sha256_file):
try:
with open(sha256_file, 'r', encoding='utf-8') as f:
sha256 = f.read().strip().lower()
logger.debug(f"Using SHA256 from .sha256 file for {file_path}")
except Exception as e:
logger.error(f"Error reading .sha256 file for {file_path}: {e}")
try: try:
# If we didn't get SHA256 from the .json file, calculate it # If we didn't get SHA256 from the .json file, calculate it