fix(database): 过滤聊天搜索索引库

- 将 chat_search_index.db 和 chat_search_index.tmp.db 视为内部索引库
- 避免搜索索引文件被误识别为账号数据库
- 补充数据库扫描和账号信息过滤测试
- 顺手整理系统辅助路由格式

Author: 2977094657

src/wechat_decrypt_tool/database_filters.py

@@ -4,7 +4,12 @@
 
 
 _IGNORED_SOURCE_DATABASE_NAMES = frozenset({"key_info.db"})
-_INDEX_DATABASE_NAMES = frozenset({"chat_search_index.db", "chat_search_index.tmp.db"})
+_INDEX_DATABASE_NAMES = frozenset(
+    {
+        "chat_search_index.db",
+        "chat_search_index.tmp.db",
+    }
+)
 _INDEX_DATABASE_SUFFIXES = ("_fts.db",)
 _INTERNAL_OUTPUT_DATABASE_NAMES = frozenset(
     {

src/wechat_decrypt_tool/routers/system.py

@@ -1,7 +1,10 @@
-from fastapi import APIRouter, HTTPException
-from pydantic import BaseModel
+from __future__ import annotations
+
 import asyncio
 from concurrent.futures import ThreadPoolExecutor
+from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel
+
 from ..img_helper import IMG_HELPER
 from .wechat_detection import check_wechat_status
 
@@ -39,9 +42,7 @@ async def pick_directory(title: str = "请选择目录", initial_dir: str = ""):
 
 @router.get("/api/system/img_helper/status", summary="获取大图下载辅助插件状态")
 async def get_img_helper_status():
-    return {
-        "enabled": IMG_HELPER.is_enabled
-    }
+    return {"enabled": IMG_HELPER.is_enabled}
 
 
 class ImgHelperToggleRequest(BaseModel):

tests/test_database_filters.py

@@ -48,6 +48,8 @@ def test_scan_account_databases_skips_index_databases(self):
             _seed_sqlite(db_storage / "contact_fts.db")
             _seed_sqlite(db_storage / "favorite_fts.db")
             _seed_sqlite(db_storage / "message_fts.db")
+            _seed_sqlite(db_storage / "chat_search_index.db")
+            _seed_sqlite(db_storage / "chat_search_index.tmp.db")
             _seed_sqlite(db_storage / "key_info.db")
 
             result = scan_account_databases_from_path(str(db_storage))
@@ -68,6 +70,8 @@ def test_collect_account_databases_skips_index_databases(self):
             _seed_sqlite(data_dir / "contact_fts.db")
             _seed_sqlite(data_dir / "favorite_fts.db")
             _seed_sqlite(data_dir / "message_fts.db")
+            _seed_sqlite(data_dir / "chat_search_index.db")
+            _seed_sqlite(data_dir / "chat_search_index.tmp.db")
 
             databases = collect_account_databases(str(data_dir), "wxid_demo")
             db_names = sorted(db["name"] for db in databases)
@@ -93,6 +97,7 @@ def test_chat_account_info_hides_index_and_internal_databases(self):
                 _seed_sqlite(account_dir / "session.db", "session_table")
                 _seed_sqlite(account_dir / "message_fts.db")
                 _seed_sqlite(account_dir / "chat_search_index.db")
+                _seed_sqlite(account_dir / "chat_search_index.tmp.db")
                 _seed_sqlite(account_dir / "session_last_message.db")
 
                 result = chat_router.get_chat_account_info("wxid_demo")