diff --git a/app/api/endpoints/subscribe.py b/app/api/endpoints/subscribe.py index 3ab451a1..420deb62 100644 --- a/app/api/endpoints/subscribe.py +++ b/app/api/endpoints/subscribe.py @@ -53,7 +53,7 @@ async def create_subscribe( """ # 类型转换 if subscribe_in.type: - mtype = MediaType.TV if subscribe_in.type == "电视剧" else MediaType.MOVIE + mtype = MediaType(subscribe_in.type) else: mtype = None # 标题转换 diff --git a/app/api/endpoints/tmdb.py b/app/api/endpoints/tmdb.py index 2b891d4a..22e92159 100644 --- a/app/api/endpoints/tmdb.py +++ b/app/api/endpoints/tmdb.py @@ -92,7 +92,7 @@ async def tmdb_info(tmdbid: int, type_name: str, """ 根据TMDBID查询themoviedb媒体信息,type_name: 电影/电视剧 """ - mtype = MediaType.MOVIE if type_name == MediaType.MOVIE.value else MediaType.TV + mtype = MediaType(type_name) tmdbinfo = TmdbChain().tmdb_info(tmdbid=tmdbid, mtype=mtype) if not tmdbinfo: return schemas.MediaInfo() diff --git a/app/chain/subscribe.py b/app/chain/subscribe.py index 43a43fea..fc185f8e 100644 --- a/app/chain/subscribe.py +++ b/app/chain/subscribe.py @@ -175,7 +175,7 @@ class SubscribeChain(ChainBase): meta = MetaInfo(subscribe.name) meta.year = subscribe.year meta.begin_season = subscribe.season or None - meta.type = MediaType.MOVIE if subscribe.type == MediaType.MOVIE.value else MediaType.TV + meta.type = MediaType(subscribe.type) # 识别媒体信息 mediainfo: MediaInfo = self.recognize_media(meta=meta, mtype=meta.type, tmdbid=subscribe.tmdbid) if not mediainfo: @@ -322,7 +322,7 @@ class SubscribeChain(ChainBase): meta = MetaInfo(subscribe.name) meta.year = subscribe.year meta.begin_season = subscribe.season or None - meta.type = MediaType.MOVIE if subscribe.type == MediaType.MOVIE.value else MediaType.TV + meta.type = MediaType(subscribe.type) # 识别媒体信息 mediainfo: MediaInfo = self.recognize_media(meta=meta, mtype=meta.type, tmdbid=subscribe.tmdbid) if not mediainfo: