From 5a07732712b0320c9f80083e15155c8ad8a7b14d Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Mon, 13 Nov 2023 20:25:57 +0800
Subject: [PATCH] =?UTF-8?q?fix=20=E8=B1=86=E7=93=A3=E5=B7=B2=E8=AE=A2?=
 =?UTF-8?q?=E9=98=85=E6=98=BE=E7=A4=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/api/endpoints/subscribe.py | 4 +++-
 app/db/models/subscribe.py     | 5 ++++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/app/api/endpoints/subscribe.py b/app/api/endpoints/subscribe.py
index bf331622..c4ddd20e 100644
--- a/app/api/endpoints/subscribe.py
+++ b/app/api/endpoints/subscribe.py
@@ -142,7 +142,9 @@ def subscribe_mediaid(
 
     if not result and title:
         meta = MetaInfo(title)
-        result = Subscribe.get_by_title(db, meta.name)
+        if season:
+            meta.begin_season = season
+        result = Subscribe.get_by_title(db, title=meta.name, season=meta.begin_season)
 
     if result and result.sites:
         result.sites = json.loads(result.sites)
diff --git a/app/db/models/subscribe.py b/app/db/models/subscribe.py
index c478a426..b9699e36 100644
--- a/app/db/models/subscribe.py
+++ b/app/db/models/subscribe.py
@@ -97,7 +97,10 @@ class Subscribe(Base):
 
     @staticmethod
     @db_query
-    def get_by_title(db: Session, title: str):
+    def get_by_title(db: Session, title: str, season: int = None):
+        if season:
+            return db.query(Subscribe).filter(Subscribe.name == title,
+                                              Subscribe.season == season).first()
         return db.query(Subscribe).filter(Subscribe.name == title).first()
 
     @staticmethod