From 172bc23b2a0afc882dfc0d24bca18d795510df50 Mon Sep 17 00:00:00 2001
From: Edward <73746306+WangEdward@users.noreply.github.com>
Date: Fri, 12 Apr 2024 14:12:15 +0000
Subject: [PATCH] fix: empty season

---
 app/chain/transfer.py             | 8 ++++++--
 app/modules/themoviedb/scraper.py | 2 +-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/app/chain/transfer.py b/app/chain/transfer.py
index 15677fdc..16e67bd0 100644
--- a/app/chain/transfer.py
+++ b/app/chain/transfer.py
@@ -223,6 +223,8 @@ class TransferChain(ChainBase):
                 # 合并季
                 if season is not None:
                     file_meta.begin_season = season
+                elif file_meta.begin_season is None:
+                    file_meta.begin_season = 1
 
                 if not file_meta:
                     logger.error(f"{file_path} 无法识别有效信息")
@@ -278,9 +280,10 @@ class TransferChain(ChainBase):
 
                 # 获取集数据
                 if file_mediainfo.type == MediaType.TV:
+                    file_mediainfo.season = file_mediainfo.season or file_meta.begin_season
                     episodes_info = self.tmdbchain.tmdb_episodes(
                         tmdbid=file_mediainfo.tmdb_id,
-                        season=1 if file_meta.begin_season is None else file_meta.begin_season
+                        season=file_mediainfo.season
                     )
                 else:
                     episodes_info = None
@@ -357,7 +360,8 @@ class TransferChain(ChainBase):
                 if settings.SCRAP_METADATA:
                     self.scrape_metadata(path=transferinfo.target_path,
                                          mediainfo=file_mediainfo,
-                                         transfer_type=transfer_type)
+                                         transfer_type=transfer_type,
+                                         metainfo=file_meta)
                 # 更新进度
                 processed_num += 1
                 self.progress.update(value=processed_num / total_num * 100,
diff --git a/app/modules/themoviedb/scraper.py b/app/modules/themoviedb/scraper.py
index 5302cfc8..6c6661c5 100644
--- a/app/modules/themoviedb/scraper.py
+++ b/app/modules/themoviedb/scraper.py
@@ -96,7 +96,7 @@ class TmdbScraper:
                             self.__save_image(url=attr_value,
                                               file_path=image_path)
                 # 查询季信息
-                seasoninfo = self.tmdb.get_tv_season_detail(mediainfo.tmdb_id, meta.begin_season)
+                seasoninfo = self.tmdb.get_tv_season_detail(mediainfo.tmdb_id, meta.begin_season or mediainfo.season)
                 if seasoninfo:
                     # 季目录NFO
                     if self._force_nfo or not file_path.with_name("season.nfo").exists():