fix 文件管理的自动重命名缺少集信息

This commit is contained in:
景大侠
2025-07-08 22:37:09 +08:00
parent 5c61b22c2f
commit 8c27b8c33e
+18
View File
@@ -1,6 +1,7 @@
from pathlib import Path from pathlib import Path
from typing import Optional, List, Tuple, Union, Dict, Callable from typing import Optional, List, Tuple, Union, Dict, Callable
from app.chain.tmdb import TmdbChain
from app.core.config import settings from app.core.config import settings
from app.core.context import MediaInfo from app.core.context import MediaInfo
from app.core.meta import MetaBase from app.core.meta import MetaBase
@@ -141,11 +142,28 @@ class FileManagerModule(_ModuleBase):
# 重命名格式 # 重命名格式
rename_format = settings.TV_RENAME_FORMAT \ rename_format = settings.TV_RENAME_FORMAT \
if mediainfo.type == MediaType.TV else settings.MOVIE_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( path = handler.get_rename_path(
template_string=rename_format, template_string=rename_format,
rename_dict=handler.get_naming_dict(meta=meta, rename_dict=handler.get_naming_dict(meta=meta,
mediainfo=mediainfo, mediainfo=mediainfo,
episodes_info=episodes_info,
file_ext=Path(meta.title).suffix) file_ext=Path(meta.title).suffix)
) )
return str(path) return str(path)