fix images

app/modules/themoviedb/__init__.py

@@ -564,6 +564,45 @@ class TheMovieDbModule(_ModuleBase):
         """
         self.cache.save()
 
+    def obtain_images(self, mediainfo: MediaInfo) -> Optional[MediaInfo]:
+        """
+        补充抓取媒体信息图片
+        :param mediainfo:  识别的媒体信息
+        :return: 更新后的媒体信息
+        """
+        if mediainfo.logo_path \
+                and mediainfo.poster_path \
+                and mediainfo.backdrop_path:
+            # 没有图片缺失
+            return mediainfo
+        # 调用TMDB图片接口
+        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)
+        if not images:
+            return mediainfo
+        # 背景图
+        if not mediainfo.backdrop_path:
+            backdrops = images.get("backdrops")
+            if backdrops:
+                backdrops = sorted(backdrops, key=lambda x: x.get("vote_average"), reverse=True)
+                mediainfo.backdrop_path = backdrops[0].get("file_path")
+        # 标志
+        if not mediainfo.logo_path:
+            logos = images.get("logos")
+            if logos:
+                logos = sorted(logos, key=lambda x: x.get("vote_average"), reverse=True)
+                mediainfo.logo_path = logos[0].get("file_path")
+        # 海报
+        if not mediainfo.poster_path:
+            posters = images.get("posters")
+            if posters:
+                posters = sorted(posters, key=lambda x: x.get("vote_average"), reverse=True)
+                mediainfo.poster_path = posters[0].get("file_path")
+        return mediainfo
+
     def obtain_specific_image(self, mediaid: Union[str, int], mtype: MediaType,
                               image_type: MediaImageType, image_prefix: str = "w500",
                               season: int = None, episode: int = None) -> Optional[str]:

app/modules/themoviedb/tmdb.py

@@ -1019,3 +1019,29 @@ class TmdbHelper:
         except Exception as e:
             print(str(e))
             return []
+
+    def get_movie_images(self, tmdbid: int) -> dict:
+        """
+        获取电影的图片
+        """
+        if not self.movie:
+            return {}
+        try:
+            logger.info(f"正在获取电影图片：{tmdbid}...")
+            return self.movie.images(tmdbid) or {}
+        except Exception as e:
+            print(str(e))
+            return {}
+
+    def get_tv_images(self, tmdbid: int, season: int) -> dict:
+        """
+        获取电视剧的图片
+        """
+        if not self.tv:
+            return {}
+        try:
+            logger.info(f"正在获取电视剧图片：{tmdbid}...")
+            return self.season.images(tv_id=tmdbid, season_num=season) or {}
+        except Exception as e:
+            print(str(e))
+            return {}