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