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

fix api

Browse Source
This commit is contained in:
jxxghp
2023-07-29 15:07:01 +08:00
parent 5d30921d2f
commit 891eaf2c12
4 changed files with 119 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
app/modules/themoviedb/__init__.py
View File

@ -272,8 +272,7 @@ class TheMovieDbModule(_ModuleBase):
if mediainfo.type == MediaType.MOVIE:
images = self.tmdb.get_movie_images(mediainfo.tmdb_id)
else:
# FIXME tmdbv3api库没有tv.images接口，只能取第1季的
images = self.tmdb.get_tv_images(mediainfo.tmdb_id, season=1)
images = self.tmdb.get_tv_images(mediainfo.tmdb_id)
if not images:
return mediainfo
if isinstance(images, list):
@ -346,6 +345,20 @@ class TheMovieDbModule(_ModuleBase):
"""
return self.tmdb.get_tv_similar(tmdbid=tmdbid)
def movie_recommend(self, tmdbid: int) -> List[dict]:
"""
根据TMDBID查询推荐电影
:param tmdbid: TMDBID
"""
return self.tmdb.get_movie_recommend(tmdbid=tmdbid)
def tv_recommend(self, tmdbid: int) -> List[dict]:
"""
根据TMDBID查询推荐电视剧
:param tmdbid: TMDBID
"""
return self.tmdb.get_tv_recommend(tmdbid=tmdbid)
def movie_credits(self, tmdbid: int) -> List[dict]:
"""
根据TMDBID查询电影演职员表

36
app/modules/themoviedb/tmdbapi.py
View File

@ -1033,7 +1033,7 @@ class TmdbHelper:
print(str(e))
return {}
def get_tv_images(self, tmdbid: int, season: int) -> dict:
def get_tv_images(self, tmdbid: int) -> dict:
"""
获取电视剧的图片
"""
@ -1041,7 +1041,7 @@ class TmdbHelper:
return {}
try:
logger.info(f"正在获取电视剧图片：{tmdbid}...")
return self.season.images(tv_id=tmdbid, season_num=season) or {}
return self.tv.images(tv_id=tmdbid) or {}
except Exception as e:
print(str(e))
return {}
@ -1054,8 +1054,7 @@ class TmdbHelper:
return []
try:
logger.info(f"正在获取相似电影：{tmdbid}...")
info = self.movie.similar(tmdbid) or {}
return info.get('results') or []
info = self.movie.similar(tmdbid) or []
except Exception as e:
print(str(e))
return []
@ -1068,8 +1067,33 @@ class TmdbHelper:
return []
try:
logger.info(f"正在获取相似电视剧：{tmdbid}...")
info = self.tv.similar(tmdbid) or {}
return info.get('results') or []
info = self.tv.similar(tmdbid) or []
except Exception as e:
print(str(e))
return []
def get_movie_recommend(self, tmdbid: int) -> List[dict]:
"""
获取电影的推荐电影
"""
if not self.movie:
return []
try:
logger.info(f"正在获取推荐电影：{tmdbid}...")
info = self.movie.recommendations(tmdbid) or []
except Exception as e:
print(str(e))
return []
def get_tv_recommend(self, tmdbid: int) -> List[dict]:
"""
获取电视剧的推荐电视剧
"""
if not self.tv:
return []
try:
logger.info(f"正在获取推荐电视剧：{tmdbid}...")
info = self.tv.recommendations(tmdbid) or []
except Exception as e:
print(str(e))
return []