mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-25 07:05:43 -03:00
Remove LoraHashIndex class and related functionality to streamline codebase.
This commit is contained in:
@@ -1,54 +0,0 @@
|
|||||||
from typing import Dict, Optional
|
|
||||||
import logging
|
|
||||||
from dataclasses import dataclass
|
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
|
||||||
|
|
||||||
@dataclass
|
|
||||||
class LoraHashIndex:
|
|
||||||
"""Index for mapping LoRA file hashes to their file paths"""
|
|
||||||
|
|
||||||
def __init__(self):
|
|
||||||
self._hash_to_path: Dict[str, str] = {}
|
|
||||||
|
|
||||||
def add_entry(self, sha256: str, file_path: str) -> None:
|
|
||||||
"""Add or update a hash -> path mapping"""
|
|
||||||
if not sha256 or not file_path:
|
|
||||||
return
|
|
||||||
# Always store lowercase hashes for consistency
|
|
||||||
self._hash_to_path[sha256.lower()] = file_path
|
|
||||||
|
|
||||||
def remove_entry(self, sha256: str) -> None:
|
|
||||||
"""Remove a hash entry"""
|
|
||||||
if sha256:
|
|
||||||
self._hash_to_path.pop(sha256.lower(), None)
|
|
||||||
|
|
||||||
def remove_by_path(self, file_path: str) -> None:
|
|
||||||
"""Remove entry by file path"""
|
|
||||||
for sha256, path in list(self._hash_to_path.items()):
|
|
||||||
if path == file_path:
|
|
||||||
del self._hash_to_path[sha256]
|
|
||||||
break
|
|
||||||
|
|
||||||
def get_path(self, sha256: str) -> Optional[str]:
|
|
||||||
"""Get file path for a given hash"""
|
|
||||||
if not sha256:
|
|
||||||
return None
|
|
||||||
return self._hash_to_path.get(sha256.lower())
|
|
||||||
|
|
||||||
def get_hash(self, file_path: str) -> Optional[str]:
|
|
||||||
"""Get hash for a given file path"""
|
|
||||||
for sha256, path in self._hash_to_path.items():
|
|
||||||
if path == file_path:
|
|
||||||
return sha256
|
|
||||||
return None
|
|
||||||
|
|
||||||
def has_hash(self, sha256: str) -> bool:
|
|
||||||
"""Check if hash exists in index"""
|
|
||||||
if not sha256:
|
|
||||||
return False
|
|
||||||
return sha256.lower() in self._hash_to_path
|
|
||||||
|
|
||||||
def clear(self) -> None:
|
|
||||||
"""Clear all entries"""
|
|
||||||
self._hash_to_path.clear()
|
|
||||||
Reference in New Issue
Block a user