From 6cecb3c6a64177c6090647f937a59b031abee87a Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Mon, 18 Mar 2024 20:02:03 +0800
Subject: [PATCH] fix bug

---
 app/chain/media.py | 24 ++++++++++++++++--------
 1 file changed, 16 insertions(+), 8 deletions(-)

diff --git a/app/chain/media.py b/app/chain/media.py
index 56da9fe3..9a5093ac 100644
--- a/app/chain/media.py
+++ b/app/chain/media.py
@@ -235,8 +235,11 @@ class MediaChain(ChainBase, metaclass=Singleton):
         """
         bangumiinfo = self.bangumi_info(bangumiid=bangumiid)
         if bangumiinfo:
-            # 名称
-            name = bangumiinfo.get("name") or bangumiinfo.get("name_cn")
+            # 优先使用原标题匹配
+            if bangumiinfo.get("name"):
+                meta = MetaInfo(title=bangumiinfo.get("name"))
+            else:
+                meta = MetaInfo(title=bangumiinfo.get("name_cn"))
             # 年份
             release_date = bangumiinfo.get("date") or bangumiinfo.get("air_date")
             if release_date:
@@ -245,9 +248,10 @@ class MediaChain(ChainBase, metaclass=Singleton):
                 year = None
             # 使用名称识别TMDB媒体信息
             return self.match_tmdbinfo(
-                name=name,
+                name=meta.name,
                 year=year,
-                mtype=MediaType.TV
+                mtype=MediaType.TV,
+                season=meta.begin_season
             )
         return None
 
@@ -290,8 +294,11 @@ class MediaChain(ChainBase, metaclass=Singleton):
         """
         bangumiinfo = self.bangumi_info(bangumiid=bangumiid)
         if bangumiinfo:
-            # 名称
-            name = bangumiinfo.get("name") or bangumiinfo.get("name_cn")
+            # 优先使用中文标题匹配
+            if bangumiinfo.get("name_cn"):
+                meta = MetaInfo(title=bangumiinfo.get("name_cn"))
+            else:
+                meta = MetaInfo(title=bangumiinfo.get("name"))
             # 年份
             release_date = bangumiinfo.get("date") or bangumiinfo.get("air_date")
             if release_date:
@@ -300,8 +307,9 @@ class MediaChain(ChainBase, metaclass=Singleton):
                 year = None
             # 使用名称识别豆瓣媒体信息
             return self.match_doubaninfo(
-                name=name,
+                name=meta.name,
                 year=year,
-                mtype=MediaType.TV
+                mtype=MediaType.TV,
+                season=meta.begin_season
             )
         return None