Checkpoint

routes/api_routes.py

@@ -4,7 +4,6 @@ import logging
 from aiohttp import web
 from typing import Dict, List
 from ..services.civitai_client import CivitaiClient
-from ..utils.file_utils import update_civitai_metadata, load_metadata
 from ..config import config
 from ..services.lora_scanner import LoraScanner
 from operator import itemgetter

services/file_monitor.py

@@ -62,48 +62,37 @@ class LoraFileHandler(FileSystemEventHandler):
             
             logger.info(f"Processing {len(changes)} file changes")
 
-            async with self.scanner._cache._lock:
+            cache = await self.scanner.get_cached_data()  # 先完成可能的初始化
-                # 获取当前缓存
+            needs_resort = False
-                cache = await self.scanner.get_cached_data()
+            new_folders = set()  # 用于收集新的文件夹
-
+            
-                needs_resort = False
+            for action, file_path in changes:
-                new_folders = set()  # 用于收集新的文件夹
+                try:
-                
+                    if action == 'add':
-                for action, file_path in changes:
+                        # 扫描新文件
-                    try:
+                        lora_data = await self.scanner.scan_single_lora(file_path)
-                        if action == 'add':
+                        if lora_data:
-                            # 扫描新文件
+                            cache.raw_data.append(lora_data)
-                            lora_data = await self.scanner.scan_single_lora(file_path)
+                            new_folders.add(lora_data['folder'])  # 收集新文件夹
-                            if lora_data:
-                                cache.raw_data.append(lora_data)
-                                new_folders.add(lora_data['folder'])  # 收集新文件夹
-                                needs_resort = True
-                                
-                        elif action == 'remove':
-                            # 从缓存中移除
-                            cache.raw_data = [
-                                item for item in cache.raw_data 
-                                if item['file_path'] != file_path
-                            ]
                             needs_resort = True
                             
-                    except Exception as e:
+                    elif action == 'remove':
-                        logger.error(f"Error processing {action} for {file_path}: {e}")
+                        # 从缓存中移除
+                        cache.raw_data = [
+                            item for item in cache.raw_data 
+                            if item['file_path'] != file_path
+                        ]
+                        needs_resort = True
+                        
+                except Exception as e:
+                    logger.error(f"Error processing {action} for {file_path}: {e}")
+            
+            if needs_resort:
+                await cache.resort()
                 
-                if needs_resort:
+                # 更新文件夹列表，包括新添加的文件夹
-                    cache.sorted_by_name = sorted(
+                all_folders = set(cache.folders) | new_folders
-                        self.scanner._cache.raw_data, 
+                cache.folders = sorted(list(all_folders))
-                        key=lambda x: x['model_name'].lower()  # Case-insensitive sort
-                    )
-                    cache.sorted_by_date = sorted(
-                        self.scanner._cache.raw_data, 
-                        key=itemgetter('modified'), 
-                        reverse=True
-                    )
-                    
-                    # 更新文件夹列表，包括新添加的文件夹
-                    all_folders = set(cache.folders) | new_folders
-                    cache.folders = sorted(list(all_folders))
                 
         except Exception as e:
             logger.error(f"Error in process_changes: {e}")

services/lora_scanner.py

@@ -91,28 +91,26 @@ class LoraScanner:
         # 确保缓存已初始化
         cache = await self.get_cached_data()
 
-        async with cache._lock:
+        # Select sorted data based on sort_by parameter
+        data = (cache.sorted_by_date if sort_by == 'date' 
+                else cache.sorted_by_name)
         
-            # Select sorted data based on sort_by parameter
+        # Apply folder filter if specified
-            data = (cache.sorted_by_date if sort_by == 'date' 
+        if folder is not None:
-                    else cache.sorted_by_name)
+            data = [item for item in data if item['folder'] == folder]
-            
+        
-            # Apply folder filter if specified
+        # Calculate pagination
-            if folder is not None:
+        total_items = len(data)
-                data = [item for item in data if item['folder'] == folder]
+        start_idx = (page - 1) * page_size
-            
+        end_idx = min(start_idx + page_size, total_items)
-            # Calculate pagination
+        
-            total_items = len(data)
+        return {
-            start_idx = (page - 1) * page_size
+            'items': data[start_idx:end_idx],
-            end_idx = min(start_idx + page_size, total_items)
+            'total': total_items,
-            
+            'page': page,
-            return {
+            'page_size': page_size,
-                'items': data[start_idx:end_idx],
+            'total_pages': (total_items + page_size - 1) // page_size
-                'total': total_items,
+        }
-                'page': page,
-                'page_size': page_size,
-                'total_pages': (total_items + page_size - 1) // page_size
-            }
 
     def invalidate_cache(self):
         """Invalidate the current cache"""
@@ -176,8 +174,8 @@ class LoraScanner:
         """
         if self._cache is None:
             return False
-            
+
-        return self._cache.update_preview_url(file_path, preview_url)
+        return await self._cache.update_preview_url(file_path, preview_url)
 
     async def scan_single_lora(self, file_path: str) -> Optional[Dict]:
         """Scan a single LoRA file and return its metadata"""