diff --git a/app/agent/tools/impl/scrape_metadata.py b/app/agent/tools/impl/scrape_metadata.py index 7e640495..1d31ec41 100644 --- a/app/agent/tools/impl/scrape_metadata.py +++ b/app/agent/tools/impl/scrape_metadata.py @@ -91,7 +91,10 @@ class ScrapeMetadataTool(MoviePilotTool): media_chain = MediaChain() scrape_path = Path(path) meta = MetaInfoPath(scrape_path) - mediainfo = await media_chain.async_recognize_by_meta(meta) + mediainfo = await media_chain.async_recognize_by_meta( + meta, + obtain_images=True, + ) if not mediainfo: return json.dumps( diff --git a/app/api/endpoints/media.py b/app/api/endpoints/media.py index e838795d..f4028f5b 100644 --- a/app/api/endpoints/media.py +++ b/app/api/endpoints/media.py @@ -123,7 +123,7 @@ def scrape(fileitem: schemas.FileItem, # 识别媒体信息 scrape_path = Path(fileitem.path) meta = MetaInfoPath(scrape_path) - mediainfo = chain.recognize_by_meta(meta) + mediainfo = chain.recognize_by_meta(meta, obtain_images=True) if not mediainfo: return schemas.Response(success=False, message="刮削失败,无法识别媒体信息") if storage == "local": diff --git a/app/chain/media.py b/app/chain/media.py index f607e4a4..b04fd848 100644 --- a/app/chain/media.py +++ b/app/chain/media.py @@ -451,7 +451,7 @@ class MediaChain(ChainBase, ConfigReloadMixin, metaclass=Singleton): self, metainfo: MetaBase, episode_group: Optional[str] = None, - obtain_images: bool = True, + obtain_images: bool = False, ) -> Optional[MediaInfo]: """ 根据主副标题识别媒体信息 @@ -576,7 +576,7 @@ class MediaChain(ChainBase, ConfigReloadMixin, metaclass=Singleton): self, path: str, episode_group: Optional[str] = None, - obtain_images: bool = True, + obtain_images: bool = False, ) -> Optional[Context]: """ 根据文件路径识别媒体信息 @@ -1386,7 +1386,7 @@ class MediaChain(ChainBase, ConfigReloadMixin, metaclass=Singleton): self, metainfo: MetaBase, episode_group: Optional[str] = None, - obtain_images: bool = True, + obtain_images: bool = False, ) -> Optional[MediaInfo]: """ 根据主副标题识别媒体信息(异步版本) @@ -1510,7 +1510,7 @@ class MediaChain(ChainBase, ConfigReloadMixin, metaclass=Singleton): self, path: str, episode_group: Optional[str] = None, - obtain_images: bool = True, + obtain_images: bool = False, ) -> Optional[Context]: """ 根据文件路径识别媒体信息(异步版本) diff --git a/app/chain/transfer.py b/app/chain/transfer.py index ff76b272..ade6cd6c 100755 --- a/app/chain/transfer.py +++ b/app/chain/transfer.py @@ -1291,10 +1291,10 @@ class TransferChain(ChainBase, ConfigReloadMixin, metaclass=Singleton): # 识别媒体信息 mediainfo = MediaChain().recognize_by_meta( task.meta, - obtain_images=False, + obtain_images=True, ) - # 更新媒体图片 + # 补充图片,确保整理后直接进入刮削的媒体信息已带图片地址。 if mediainfo: self.obtain_images(mediainfo=mediainfo) diff --git a/app/workflow/actions/scrape_file.py b/app/workflow/actions/scrape_file.py index 6a2c1dc1..fc7914f3 100644 --- a/app/workflow/actions/scrape_file.py +++ b/app/workflow/actions/scrape_file.py @@ -67,7 +67,7 @@ class ScrapeFileAction(BaseAction): mediachain = MediaChain() mediainfo = mediachain.recognize_by_meta( meta, - obtain_images=False, + obtain_images=True, ) if not mediainfo: _failed_count += 1