Finished new version of rss parser.

src/module/parser/analyser/__init__.py

@@ -1,3 +1,3 @@
 from .raw_parser import raw_parser
 from .torrent_parser import torrent_parser
-from .tmdb_parser import TMDBMatcher
+from .tmdb_parser import tmdb_parser

src/module/parser/analyser/tmdb_parser.py

@@ -9,65 +9,70 @@ from module.conf import TMDB_API
 @dataclass
 class TMDBInfo:
     id: int
-    title_jp: str
-    title_zh: str
+    title: str
+    original_title: str
     season: list[dict]
     last_season: int
-    year_number: int
+    year: str
 
 
-class TMDBMatcher:
-    def __init__(self):
-        self.search_url = lambda e: \
-            f"https://api.themoviedb.org/3/search/tv?api_key={TMDB_API}&page=1&query={e}&include_adult=false"
-        self.info_url = lambda e: \
-            f"https://api.themoviedb.org/3/tv/{e}?api_key={TMDB_API}&language=zh-CN"
+LANGUAGE = {
+    "zh": "zh-CN",
+    "jp": "ja-JP",
+    "en": "en-US"
+}
 
-    def is_animation(self, tv_id) -> bool:
-        url_info = self.info_url(tv_id)
-        with RequestContent() as req:
-            type_id = req.get_json(url_info)["genres"]
-            for type in type_id:
-                if type.get("id") == 16:
-                    return True
-        return False
+search_url = lambda e: \
+        f"https://api.themoviedb.org/3/search/tv?api_key={TMDB_API}&page=1&query={e}&include_adult=false"
+info_url = lambda e, key: \
+        f"https://api.themoviedb.org/3/tv/{e}?api_key={TMDB_API}&language={LANGUAGE[key]}"
 
-    # def get_zh_title(self, id):
-    #     alt_title_url = self.alt_title_url(id)
-    #     titles = self._request.get_content(alt_title_url, content="json")
-    #     for title in titles:
-    #         if title["iso_3166_1"] == "CN":
-    #             return title["title"]
-    #     return None
 
-    @staticmethod
-    def get_season(seasons: list) -> int:
-        for season in seasons:
-            if re.search(r"第 \d 季", season.get("season")) is not None:
-                date = season.get("air_date").split("-")
-                [year, _ , _] = date
-                now_year = time.localtime().tm_year
-                if int(year) == now_year:
-                    return int(re.findall(r"\d", season.get("season"))[0])
+def is_animation(tv_id, language) -> bool:
+    url_info = info_url(tv_id, language)
+    with RequestContent() as req:
+        type_id = req.get_json(url_info)["genres"]
+        for type in type_id:
+            if type.get("id") == 16:
+                return True
+    return False
 
-    def tmdb_search(self, title) -> TMDBInfo:
-        with RequestContent() as req:
-            url = self.search_url(title)
+
+def get_season(seasons: list) -> int:
+    for season in seasons:
+        if re.search(r"第 \d 季", season.get("season")) is not None:
+            date = season.get("air_date").split("-")
+            [year, _ , _] = date
+            now_year = time.localtime().tm_year
+            if int(year) <= now_year:
+                return int(re.findall(r"\d", season.get("season"))[0])
+
+
+def tmdb_parser(title, language) -> TMDBInfo | None:
+    with RequestContent() as req:
+        url = search_url(title)
+        contents = req.get_json(url).get("results")
+        if contents.__len__() == 0:
+            url = search_url(title.replace(" ", ""))
             contents = req.get_json(url).get("results")
-            if contents.__len__() == 0:
-                url = self.search_url(title.replace(" ", ""))
-                contents = req.get_json(url).get("results")
-            # 判断动画
+        # 判断动画
+        if contents:
             for content in contents:
                 id = content["id"]
-                if self.is_animation(id):
+                if is_animation(id, language):
                     break
-            url_info = self.info_url(id)
+            url_info = info_url(id, language)
             info_content = req.get_json(url_info)
+            season = [{"season": s.get("name"), "air_date": s.get("air_date")} for s in info_content.get("seasons")]
+            last_season = get_season(season)
+            original_title = info_content.get("original_name")
+            official_title = info_content.get("name")
+            year_number = info_content.get("first_air_date").split("-")[0]
+            return TMDBInfo(id, official_title, original_title, season, last_season, str(year_number))
+        else:
+            return None
 
-        season = [{"season": s.get("name"), "air_date": s.get("air_date")} for s in info_content.get("seasons")]
-        last_season = self.get_season(season)
-        title_jp = info_content.get("original_name")
-        title_zh = info_content.get("name")
-        year_number = info_content.get("first_air_date").split("-")[0]
-        return TMDBInfo(id, title_jp, title_zh, season, last_season, year_number)
+
+if __name__ == '__main__':
+    title = "鬼灭之刃+刀匠村篇"
+    print(tmdb_parser(title, "zh"))