diff --git a/app/api/endpoints/transfer.py b/app/api/endpoints/transfer.py index 00908877..748f8274 100644 --- a/app/api/endpoints/transfer.py +++ b/app/api/endpoints/transfer.py @@ -19,6 +19,7 @@ def manual_transfer(path: str = None, logid: int = None, target: str = None, tmdbid: int = None, + doubanid: str = None, type_name: str = None, season: int = None, transfer_type: str = None, @@ -36,6 +37,7 @@ def manual_transfer(path: str = None, :param target: 目标路径 :param type_name: 媒体类型、电影/电视剧 :param tmdbid: tmdbid + :param doubanid: 豆瓣ID :param season: 剧集季号 :param transfer_type: 转移类型,move/copy 等 :param episode_format: 剧集识别格式 @@ -91,6 +93,7 @@ def manual_transfer(path: str = None, in_path=in_path, target=target, tmdbid=tmdbid, + doubanid=doubanid, mtype=mtype, season=season, transfer_type=transfer_type, diff --git a/app/chain/transfer.py b/app/chain/transfer.py index e6bb8c46..ad5d18da 100644 --- a/app/chain/transfer.py +++ b/app/chain/transfer.py @@ -555,6 +555,7 @@ class TransferChain(ChainBase): def manual_transfer(self, in_path: Path, target: Path = None, tmdbid: int = None, + doubanid: str = None, mtype: MediaType = None, season: int = None, transfer_type: str = None, @@ -566,6 +567,7 @@ class TransferChain(ChainBase): :param in_path: 源文件路径 :param target: 目标路径 :param tmdbid: TMDB ID + :param doubanid: 豆瓣ID :param mtype: 媒体类型 :param season: 季度 :param transfer_type: 转移类型 @@ -575,12 +577,12 @@ class TransferChain(ChainBase): """ logger.info(f"手动转移:{in_path} ...") - if tmdbid: + if tmdbid or doubanid: # 有输入TMDBID时单个识别 # 识别媒体信息 - mediainfo: MediaInfo = self.mediachain.recognize_media(tmdbid=tmdbid, mtype=mtype) + mediainfo: MediaInfo = self.mediachain.recognize_media(tmdbid=tmdbid, doubanid=doubanid, mtype=mtype) if not mediainfo: - return False, f"媒体信息识别失败,tmdbid: {tmdbid}, type: {mtype.value}" + return False, f"媒体信息识别失败,tmdbid:{tmdbid},doubanid:{doubanid},type: {mtype.value}" # 开始进度 self.progress.start(ProgressKey.FileTransfer) self.progress.update(value=0,