fix bugs

Author: shinonomeow

backend/src/main.py

@@ -89,3 +89,4 @@ def index():
         port=settings.program.webui_port,
         log_config=uvicorn_logging_config,
     )
+    

backend/src/module/api/bangumi.py

@@ -3,7 +3,7 @@
 from sqlalchemy.util.concurrency import asyncio
 
 from module.database import Database, engine
-from module.manager import TorrentManager
+from module.manager import BangumiManager
 from module.models import APIResponse, Bangumi, BangumiUpdate, ResponseModel
 from module.security.api import get_current_user
 
@@ -33,7 +33,7 @@ async def get_all_data():
     dependencies=[Depends(get_current_user)],
 )
 async def get_data(bangumi_id: str):
-    resp = TorrentManager().search_one(bangumi_id)
+    resp = BangumiManager().search_one(bangumi_id)
     if resp is None:
         return ResponseModel(
             status_code=406,
@@ -53,7 +53,7 @@ async def update_rule(
     bangumi_id: int,
     data: BangumiUpdate,
 ):
-    resp = await TorrentManager().update_rule(bangumi_id, data)
+    resp = await BangumiManager().update_rule(bangumi_id, data)
     if resp:
         resp = ResponseModel(
             status_code=200,
@@ -78,7 +78,7 @@ async def update_rule(
     dependencies=[Depends(get_current_user)],
 )
 async def delete_rule(bangumi_id: str, file: bool = False):
-    data = await TorrentManager().delete_rule(bangumi_id, file)
+    data = await BangumiManager().delete_rule(bangumi_id, file)
     if data:
         resp = ResponseModel(
             status_code=200,
@@ -104,7 +104,7 @@ async def delete_rule(bangumi_id: str, file: bool = False):
 async def delete_many_rule(bangumi_id: list, file: bool = False):
     tasks = []
     for i in bangumi_id:
-        tasks.append(TorrentManager().delete_rule(i, file))
+        tasks.append(BangumiManager().delete_rule(i, file))
 
     resp = await asyncio.gather(*tasks)
     resp = resp[0]
@@ -117,7 +117,7 @@ async def delete_many_rule(bangumi_id: list, file: bool = False):
     dependencies=[Depends(get_current_user)],
 )
 async def disable_rule(bangumi_id: str, file: bool = False):
-    resp = await TorrentManager().disable_rule(bangumi_id, file)
+    resp = await BangumiManager().disable_rule(bangumi_id, file)
     if resp:
         resp = ResponseModel(
             status_code=200,
@@ -140,10 +140,10 @@ async def disable_rule(bangumi_id: str, file: bool = False):
     response_model=APIResponse,
     dependencies=[Depends(get_current_user)],
 )
-async def disable_many_rule(bangumi_id: list, file: bool = False):
+async def disable_many_rule(bangumi_id: list[int], file: bool = False):
     tasks = []
     for i in bangumi_id:
-        tasks.append(TorrentManager().disable_rule(i, file))
+        tasks.append(BangumiManager().disable_rule(i, file))
     resp = await asyncio.gather(*tasks)
     resp = resp[-1]
     return u_response(resp)
@@ -155,7 +155,7 @@ async def disable_many_rule(bangumi_id: list, file: bool = False):
     dependencies=[Depends(get_current_user)],
 )
 async def enable_rule(bangumi_id: str):
-    resp = await TorrentManager().enable_rule(bangumi_id)
+    resp = await BangumiManager().enable_rule(bangumi_id)
     if resp:
         resp = ResponseModel(
             status_code=200,
@@ -179,7 +179,7 @@ async def enable_rule(bangumi_id: str):
     dependencies=[Depends(get_current_user)],
 )
 async def refresh_poster():
-    resp = await TorrentManager().refresh_poster()
+    resp = await BangumiManager().refresh_poster()
     resp = ResponseModel(
         status_code=200,
         status=True,
@@ -195,7 +195,7 @@ async def refresh_poster():
     dependencies=[Depends(get_current_user)],
 )
 async def refresh_single_poster(bangumi_id: int):
-    resp = await TorrentManager().refind_poster(bangumi_id)
+    resp = await BangumiManager().refind_poster(bangumi_id)
     if resp:
         resp = ResponseModel(
             status_code=200,

backend/src/module/api/program.py

@@ -20,6 +20,14 @@
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     # 启动事件
+    # 首先检查数据库版本并升级
+    logger.info("检查数据库版本兼容性...")
+    from module.database import check_and_upgrade_database
+    
+    if not check_and_upgrade_database():
+        logger.error("数据库升级失败，程序无法启动")
+        raise RuntimeError("数据库升级失败")
+    
     await program.startup()
     yield
     # 关闭事件

backend/src/module/api/rss.py

@@ -180,7 +180,7 @@ async def update_rss(
     dependencies=[Depends(get_current_user)],
 )
 async def refresh_all():
-    await engine.refresh_all_rss()
+    await engine.refresh_all()
     return JSONResponse(
         status_code=200,
         content={
@@ -196,6 +196,7 @@ async def refresh_all():
     dependencies=[Depends(get_current_user)],
 )
 async def refresh_rss(rss_id: int):
+    # TODO: 还没做
     await engine.refresh_rss(rss_id=rss_id)
     return JSONResponse(
         status_code=200,
@@ -233,7 +234,7 @@ async def analysis(rss: RSSItem):
         )
     # 只有非聚合才会用
     for torrent in torrents:
-        data = await analyser.torrent_to_data(torrent, rss)
+        data = await analyser.torrent_to_bangumi(torrent, rss)
         if data:
             return data
 
@@ -245,11 +246,6 @@ async def analysis(rss: RSSItem):
             msg_zh="无法解析此链接。",
         )
     )
-    # data = await analyser.link_to_data(rss)
-    # if isinstance(data, Bangumi):
-    #     return data
-    # else:
-    #     return u_response(data)
 
 
 @router.post(

backend/src/module/core/aiocore.py

@@ -21,6 +21,7 @@ def __init__(self):
         self.services = []
         self._download_monitor = None
         self._rename_monitor = None
+        self._notification_monitor = None
         self._running: bool = False
 
     async def initialize(self) -> None:
@@ -73,6 +74,7 @@ async def _register_event_handlers(self) -> None:
         try:
             from module.downloader.download_monitor import DownloadMonitor
             from module.manager.rename_monitor import RenameMonitor
+            from module.notification.notification_monitor import NotificationMonitor
             from module.utils.events import EventType
 
             # 创建并注册 DownloadMonitor
@@ -92,6 +94,15 @@ async def _register_event_handlers(self) -> None:
             )
             logger.info("[AsyncCore] 已注册 RenameMonitor 事件处理器")
 
+            # 创建并注册 NotificationMonitor
+            self._notification_monitor = NotificationMonitor(event_bus=self.event_bus)
+            await self._notification_monitor.initialize()
+            self.event_bus.subscribe(
+                EventType.NOTIFICATION_REQUEST,
+                self._notification_monitor.handle_notification_request,
+            )
+            logger.info("[AsyncCore] 已注册 NotificationMonitor 事件处理器")
+
             # 为下载服务设置事件总线
             for service in self.services:
                 if service.name == "download" and hasattr(
@@ -154,6 +165,10 @@ async def stop(self) -> None:
         except Exception as e:
             logger.error(f"[AsyncCore] 重命名监控器关闭失败: {e}")
 
+        try:
+            if hasattr(self, "_notification_monitor") and self._notification_monitor:
+                await self._notification_monitor.shutdown()
+                logger.debug("[AsyncCore] 通知监控器已关闭")
         except Exception as e:
             logger.error(f"[AsyncCore] 通知监控器关闭失败: {e}")
 
@@ -192,6 +207,7 @@ def get_status(self) -> dict[str, Any]:
         status["monitors"] = {
             "download_monitor": bool(self._download_monitor),
             "rename_monitor": bool(self._rename_monitor),
+            "notification_monitor": bool(self._notification_monitor),
         }
 
         return status

backend/src/module/core/services.py

@@ -1,9 +1,10 @@
 import logging
-from module.conf import settings
 from abc import ABC, abstractmethod
 from typing import TYPE_CHECKING, Any
+
 from typing_extensions import override
 
+from module.conf import settings
 from module.utils.events import ServiceException
 
 if TYPE_CHECKING:
@@ -15,7 +16,7 @@
 class BaseService(ABC):
     """服务基类"""
 
-    def __init__(self, name: str = None):
+    def __init__(self, name: str|None = None):
         # 如果没有指定名称，使用类名自动生成
         if name is None:
             name = self.__class__.__name__.lower().replace("service", "")
@@ -79,7 +80,6 @@ def get_task_config(self) -> dict[str, Any]:
             "enabled": True,
         }
 
-
     async def execute(self) -> None:
         """执行RSS刷新任务"""
         if not self._engine:
@@ -107,8 +107,7 @@ def __init__(self):
     async def _setup(self) -> None:
         # 预检查下载模块是否可用
         try:
-            from module.downloader import DownloadController
-            from module.downloader import Client
+            from module.downloader import Client, DownloadController
 
             Client.start_login()
 
@@ -143,10 +142,11 @@ async def execute(self) -> None:
             logger.error(f"[DownloadService] 执行失败: {e}")
             raise ServiceException("download", f"执行失败: {e}")
 
-    def set_event_bus(self,event_bus):
+    def set_event_bus(self, event_bus):
         if not self._download_controller:
             raise ServiceException("download", "下载控制器未初始化")
         self._download_controller.set_event_bus(event_bus)
+
     async def cleanup(self) -> None:
         """清理下载客户端"""
         try:

backend/src/module/database/__init__.py

@@ -1,4 +1,5 @@
 from .combine import Database
 from .engine import engine
+from .migration import DatabaseMigration, check_and_upgrade_database
 
-__all__ = ["engine", "Database"]
+__all__ = ["engine", "Database", "DatabaseMigration", "check_and_upgrade_database"]