From 596b2e11b814d105ed8e4335e44d4c5b8fa39f2d Mon Sep 17 00:00:00 2001
From: Akimio521 <akimio2333@gmail.com>
Date: Thu, 12 Sep 2024 12:33:00 +0800
Subject: [PATCH] =?UTF-8?q?feat:=E6=9F=A5=E8=AF=A2=E7=A7=8D=E5=AD=90?=
 =?UTF-8?q?=E6=97=B6=E5=8C=B9=E9=85=8D=E9=A6=99=E6=B8=AF=E3=80=81=E5=8F=B0?=
 =?UTF-8?q?=E6=B9=BE=E8=AF=91=E5=90=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/chain/search.py               | 2 ++
 app/core/context.py               | 8 ++++++++
 app/modules/themoviedb/tmdbapi.py | 7 ++++++-
 3 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/app/chain/search.py b/app/chain/search.py
index 78527c17..00c8c17f 100644
--- a/app/chain/search.py
+++ b/app/chain/search.py
@@ -154,6 +154,8 @@ class SearchChain(ChainBase):
             keywords = list(dict.fromkeys([k for k in [mediainfo.title,
                                                        mediainfo.original_title,
                                                        mediainfo.en_title,
+                                                       mediainfo.hk_title,
+                                                       mediainfo.tw_title,
                                                        mediainfo.sg_title] if k]))
 
         # 执行搜索
diff --git a/app/core/context.py b/app/core/context.py
index 35799c2f..56e75264 100644
--- a/app/core/context.py
+++ b/app/core/context.py
@@ -141,6 +141,10 @@ class MediaInfo:
     title: str = None
     # 英文标题
     en_title: str = None
+    # 香港标题
+    hk_title: str = None
+    # 台湾标题
+    tw_title: str = None
     # 新加坡标题
     sg_title: str = None
     # 年份
@@ -386,6 +390,10 @@ class MediaInfo:
         self.original_language = info.get('original_language')
         # 英文标题
         self.en_title = info.get('en_title')
+        # 香港标题
+        self.hk_title = info.get('hk_title')
+        # 台湾标题
+        self.tw_title = info.get('tw_title')
         # 新加坡标题
         self.sg_title = info.get('sg_title')
         if self.type == MediaType.MOVIE:
diff --git a/app/modules/themoviedb/tmdbapi.py b/app/modules/themoviedb/tmdbapi.py
index 15113ecf..83fd3d8a 100644
--- a/app/modules/themoviedb/tmdbapi.py
+++ b/app/modules/themoviedb/tmdbapi.py
@@ -653,9 +653,14 @@ class TmdbApi:
         else:
             en_title = __get_tmdb_lang_title(tmdb_info, "US")
             tmdb_info['en_title'] = en_title or org_title
+        
+        # 查找香港台湾译名
+        tmdb_info['hk_title'] = __get_tmdb_lang_title(tmdb_info, "HK")
+        tmdb_info['tw_title'] = __get_tmdb_lang_title(tmdb_info, "TW")
+
         # 查找新加坡名（用于替代中文名）
         tmdb_info['sg_title'] = __get_tmdb_lang_title(tmdb_info, "SG") or org_title
-
+                                                      
     def __get_movie_detail(self,
                            tmdbid: int,
                            append_to_response: str = "images,"