fix

app/chain/__init__.py

@@ -73,6 +73,9 @@ class ChainBase(AbstractSingleton, metaclass=Singleton):
                         tmdbid: int = None) -> Optional[MediaInfo]:
         return self.run_module("recognize_media", meta=meta, mtype=mtype, tmdbid=tmdbid)
 
+    def obtain_image(self, mediainfo: MediaInfo) -> Optional[MediaInfo]:
+        return self.run_module("obtain_image", mediainfo=mediainfo)
+
     def douban_info(self, doubanid: str) -> Optional[dict]:
         return self.run_module("douban_info", doubanid=doubanid)
 
@@ -82,9 +85,6 @@ class ChainBase(AbstractSingleton, metaclass=Singleton):
     def webhook_parser(self, body: Any, form: Any, args: Any) -> Optional[dict]:
         return self.run_module("webhook_parser", body=body, form=form, args=args)
 
-    def obtain_image(self, mediainfo: MediaInfo) -> Optional[MediaInfo]:
-        return self.run_module("obtain_image", mediainfo=mediainfo)
-
     def search_medias(self, meta: MetaBase) -> Optional[List[MediaInfo]]:
         return self.run_module("search_medias", meta=meta)
 
@@ -103,6 +103,9 @@ class ChainBase(AbstractSingleton, metaclass=Singleton):
                  episodes: Set[int] = None) -> Optional[Tuple[Optional[str], str]]:
         return self.run_module("download", torrent_path=torrent_path, cookie=cookie, episodes=episodes)
 
+    def download_added(self, context: Context, torrent_path: Path) -> None:
+        return self.run_module("download_added", context=context, torrent_path=torrent_path)
+
     def list_torrents(self, status: TorrentStatus = None, hashs: Union[list, str] = None) -> Optional[List[dict]]:
         return self.run_module("list_torrents", status=status, hashs=hashs)
 
@@ -112,6 +115,9 @@ class ChainBase(AbstractSingleton, metaclass=Singleton):
     def transfer_completed(self, hashs: Union[str, list], transinfo: dict) -> None:
         return self.run_module("transfer_completed", hashs=hashs, transinfo=transinfo)
 
+    def remove_torrents(self, hashs: Union[str, list]) -> bool:
+        return self.run_module("remove_torrents", hashs=hashs)
+
     def media_exists(self, mediainfo: MediaInfo) -> Optional[dict]:
         return self.run_module("media_exists", mediainfo=mediainfo)
 

app/chain/download.py

@@ -117,7 +117,10 @@ class DownloadChain(ChainBase):
             if _hash:
                 # 下载成功
                 downloaded_list.append(_context)
+                # 发送消息
                 self.post_download_message(meta=_meta, mediainfo=_media, torrent=_torrent, userid=userid)
+                # 下载成功后处理
+                self.download_added(context=_context, torrent_path=_torrent_file)
             else:
                 # 下载失败
                 logger.error(f"{_media.get_title_string()} 添加下载任务失败："

app/chain/user_message.py

@@ -203,6 +203,8 @@ class UserMessageChain(ChainBase):
                         return
                     # 下载成功，发送通知
                     self.downloadchain.post_download_message(meta=meta, mediainfo=self._current_media, torrent=torrent)
+                    # 下载成功后处理
+                    self.download_added(context=context, torrent_path=torrent_file)
 
         elif text.lower() == "p":
             # 上一页