From 0270af5b19a378e502a2bda8052d510aceac117c Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Fri, 14 Jun 2024 16:13:21 +0800
Subject: [PATCH] =?UTF-8?q?fix=20=E8=B1=86=E7=93=A3=E8=BD=ACTMDB=E6=90=9C?=
 =?UTF-8?q?=E7=B4=A2=E6=97=B6=E4=B8=A2=E5=A4=B1=E5=AD=A3=E7=9A=84=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/api/endpoints/search.py | 2 ++
 app/chain/media.py          | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/app/api/endpoints/search.py b/app/api/endpoints/search.py
index f2ba22ca..345f0b10 100644
--- a/app/api/endpoints/search.py
+++ b/app/api/endpoints/search.py
@@ -52,6 +52,8 @@ def search_by_id(mediaid: str,
             # 通过豆瓣ID识别TMDBID
             tmdbinfo = MediaChain().get_tmdbinfo_by_doubanid(doubanid=doubanid, mtype=mtype)
             if tmdbinfo:
+                if tmdbinfo.get('season') and not season:
+                    season = tmdbinfo.get('season')
                 torrents = SearchChain().search_by_id(tmdbid=tmdbinfo.get("id"),
                                                       mtype=mtype, area=area, season=season)
             else:
diff --git a/app/chain/media.py b/app/chain/media.py
index a735decc..6ec4fa10 100644
--- a/app/chain/media.py
+++ b/app/chain/media.py
@@ -220,6 +220,8 @@ class MediaChain(ChainBase, metaclass=Singleton):
                     season=meta.begin_season
                 )
                 if tmdbinfo:
+                    # 合季季后返回
+                    tmdbinfo['season'] = meta.begin_season
                     break
         return tmdbinfo