diff --git a/app/modules/filemanager/__init__.py b/app/modules/filemanager/__init__.py index 4a236dab..18bf543b 100644 --- a/app/modules/filemanager/__init__.py +++ b/app/modules/filemanager/__init__.py @@ -1,6 +1,7 @@ from pathlib import Path from typing import Optional, List, Tuple, Union, Dict, Callable +from app.chain.tmdb import TmdbChain from app.core.config import settings from app.core.context import MediaInfo from app.core.meta import MetaBase @@ -141,11 +142,28 @@ class FileManagerModule(_ModuleBase): # 重命名格式 rename_format = settings.TV_RENAME_FORMAT \ if mediainfo.type == MediaType.TV else settings.MOVIE_RENAME_FORMAT + # 获取集信息 + episodes_info: Optional[List[TmdbEpisode]] = None + if mediainfo.type == MediaType.TV: + # 判断注意season为0的情况 + season_num = mediainfo.season + if season_num is None and meta.season_seq: + if meta.season_seq.isdigit(): + season_num = int(meta.season_seq) + # 默认值1 + if season_num is None: + season_num = 1 + episodes_info = TmdbChain().tmdb_episodes( + tmdbid=mediainfo.tmdb_id, + season=season_num, + episode_group=mediainfo.episode_group, + ) # 获取重命名后的名称 path = handler.get_rename_path( template_string=rename_format, rename_dict=handler.get_naming_dict(meta=meta, mediainfo=mediainfo, + episodes_info=episodes_info, file_ext=Path(meta.title).suffix) ) return str(path)