From ad96eacf55edf3bb1df75ef951dabab23da23f51 Mon Sep 17 00:00:00 2001
From: wumode <wumo1999@gmail.com>
Date: Fri, 23 May 2025 17:50:55 +0800
Subject: [PATCH] =?UTF-8?q?imdbsource:=20=E6=8E=A8=E8=8D=90=E7=83=AD?=
 =?UTF-8?q?=E9=97=A8=E7=BA=AA=E5=BD=95=E7=89=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 package.v2.json                   |  1 +
 plugins.v2/imdbsource/__init__.py | 59 +++++++++++++++++++++++++++++--
 2 files changed, 57 insertions(+), 3 deletions(-)

diff --git a/package.v2.json b/package.v2.json
index 75e61f0..3fe26f7 100644
--- a/package.v2.json
+++ b/package.v2.json
@@ -429,6 +429,7 @@
         "author": "wumode",
         "level": 1,
         "history": {
+            "v1.2": "推荐热门纪录片",
             "v1.1": "推荐支持IMDB数据源; 优化海报尺寸，减少卡顿",
             "v1.0": "探索支持IMDb数据源"
         }
diff --git a/plugins.v2/imdbsource/__init__.py b/plugins.v2/imdbsource/__init__.py
index a460602..5ef78d3 100644
--- a/plugins.v2/imdbsource/__init__.py
+++ b/plugins.v2/imdbsource/__init__.py
@@ -42,7 +42,7 @@ class ImdbSource(_PluginBase):
 
     _imdb_helper = None
     _cache = {"discover": [], "trending": [], "trending_in_anime": [], "trending_in_sitcom": [],
-              "imdb_top_250": []}
+              "trending_in_documentary": [], "imdb_top_250": []}
 
     def init_plugin(self, config: dict = None):
         if config:
@@ -342,6 +342,47 @@ class ImdbSource(_PluginBase):
             return MediaType.MOVIE
         return MediaType.UNKNOWN
 
+    def trending_in_documentary(self, apikey: str, page: int = 1, count: int = 30) -> List[schemas.MediaInfo]:
+        if apikey != settings.API_TOKEN:
+            return []
+        if not self._imdb_helper:
+            return []
+        title_types = ("tvSeries", "tvMiniSeries", "tvShort", "tvEpisode", 'movie')
+        first_page = False
+        if page == 1:
+            first_page = True
+            self._cache["trending_in_documentary"] = []  # 清空缓存
+        results = []
+        if len(self._cache["trending_in_documentary"]) >= count:
+            results = self._cache["trending_in_documentary"][:count]
+            self._cache["trending_in_documentary"] = self._cache["trending_in_documentary"][count:]
+        else:
+            results.extend(self._cache["trending_in_documentary"])
+            remaining = count - len(results)
+            self._cache["trending_in_documentary"] = []  # 清空缓存
+            data = self._imdb_helper.advanced_title_search(first_page=first_page,
+                                                           title_types=title_types,
+                                                           sort_by="POPULARITY",
+                                                           sort_order="ASC",
+                                                           interests=("Documentary",)
+                                                           )
+            if not data:
+                new_results = []
+            else:
+                new_results = data.get("edges")
+            if new_results:
+                results.extend(new_results[:remaining])
+                self._cache["trending_in_documentary"] = new_results[remaining:]
+        res = []
+        for item in results:
+            title_type_id = item.get('node').get("title").get("titleType", {}).get("id")
+            mtype = self.title_id_to_mtype(title_type_id)
+            if mtype == MediaType.MOVIE:
+                res.append(self.__movie_to_media(item.get('node').get("title")))
+            elif mtype == MediaType.TV:
+                res.append(self.__series_to_media(item.get('node').get("title")))
+        return res
+
     def imdb_top_250(self, apikey: str, page: int = 1, count: int = 30) -> List[schemas.MediaInfo]:
         if apikey != settings.API_TOKEN:
             return []
@@ -351,7 +392,7 @@ class ImdbSource(_PluginBase):
         first_page = False
         if page == 1:
             first_page = True
-            self._cache["trending_in_sitcom"] = []  # 清空缓存
+            self._cache["imdb_top_250"] = []  # 清空缓存
         results = []
         if len(self._cache["imdb_top_250"]) >= count:
             results = self._cache["imdb_top_250"][:count]
@@ -650,6 +691,13 @@ class ImdbSource(_PluginBase):
                 "methods": ["GET"],
                 "summary": "IMDb Top 250 Movies",
                 "description": "获取 IMDb Top 250 Movies 数据",
+            },
+            {
+                "path": "/trending_in_documentary",
+                "endpoint": self.trending_in_documentary,
+                "methods": ["GET"],
+                "summary": "IMDb Trending in Documentary",
+                "description": "获取 IMDb Trending in Documentary 数据",
             }
         ]
 
@@ -1273,7 +1321,12 @@ class ImdbSource(_PluginBase):
             api_path=f"plugin/ImdbSource/imdb_top_250?apikey={settings.API_TOKEN}",
             type='Movies'
         )
-        trending_source = [imdb_trending, trending_in_anime, trending_in_sitcom, imdb_top_250]
+        imdb_documentary: schemas.RecommendMediaSource = schemas.RecommendMediaSource(
+            name="IMDb Trending in Documentary",
+            api_path=f"plugin/ImdbSource/trending_in_documentary?apikey={settings.API_TOKEN}",
+            type='Rankings'
+        )
+        trending_source = [imdb_trending, trending_in_anime, trending_in_sitcom, imdb_top_250, imdb_documentary]
         if not event_data.extra_sources:
             event_data.extra_sources = trending_source
         else: