codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix MediaType

Browse Source
This commit is contained in:
jxxghp 2023-07-07 15:41:53 +08:00
parent 89f2b60e7c
commit 39165a92f0
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app/api/endpoints/subscribe.py
View File

@ -53,7 +53,7 @@ async def create_subscribe(
""" """
# 类型转换 # 类型转换
if subscribe_in.type: if subscribe_in.type:
mtype = MediaType.TV if subscribe_in.type == "电视剧" else MediaType.MOVIE mtype = MediaType(subscribe_in.type)
else: else:
mtype = None mtype = None
# 标题转换 # 标题转换

2
app/api/endpoints/tmdb.py
View File

@ -92,7 +92,7 @@ async def tmdb_info(tmdbid: int, type_name: str,
""" """
根据TMDBID查询themoviedb媒体信息，type_name: 电影/电视剧 根据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) tmdbinfo = TmdbChain().tmdb_info(tmdbid=tmdbid, mtype=mtype)
if not tmdbinfo: if not tmdbinfo:
return schemas.MediaInfo() return schemas.MediaInfo()

4
app/chain/subscribe.py
View File

@ -175,7 +175,7 @@ class SubscribeChain(ChainBase):
meta = MetaInfo(subscribe.name) meta = MetaInfo(subscribe.name)
meta.year = subscribe.year meta.year = subscribe.year
meta.begin_season = subscribe.season or None 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) mediainfo: MediaInfo = self.recognize_media(meta=meta, mtype=meta.type, tmdbid=subscribe.tmdbid)
if not mediainfo: if not mediainfo:
@ -322,7 +322,7 @@ class SubscribeChain(ChainBase):
meta = MetaInfo(subscribe.name) meta = MetaInfo(subscribe.name)
meta.year = subscribe.year meta.year = subscribe.year
meta.begin_season = subscribe.season or None 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) mediainfo: MediaInfo = self.recognize_media(meta=meta, mtype=meta.type, tmdbid=subscribe.tmdbid)
if not mediainfo: if not mediainfo: