From 3eabf11ba50eaa098c678bfab02e9ab125ef0500 Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@163.com>
Date: Sat, 10 Jun 2023 14:11:26 +0800
Subject: [PATCH] fix message

---
 app/chain/user_message.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/app/chain/user_message.py b/app/chain/user_message.py
index 7fda4dd7..f2dc5d29 100644
--- a/app/chain/user_message.py
+++ b/app/chain/user_message.py
@@ -81,13 +81,17 @@ class UserMessageChain(ChainBase):
             if cache_type == "Search":
                 mediainfo: MediaInfo = cache_list[int(text) + self._current_page * self._page_size - 1]
                 self._current_media = mediainfo
-                # 检查是否已存在
-                exists: dict = self.media_exists(mediainfo=mediainfo)
-                if exists:
-                    # 已存在
-                    self.post_message(
-                        title=f"{mediainfo.type.value} {mediainfo.get_title_string()} 媒体库中已存在", userid=userid)
+                # 查询缺失的媒体信息
+                exist_flag, no_exists = self.downloadchain.get_no_exists_info(mediainfo=self._current_media)
+                if exist_flag:
+                    self.post_message(title=f"{self._current_media.get_title_string()} 媒体库中已存在",
+                                      userid=userid)
                     return
+                # 发送缺失的媒体信息
+                if no_exists:
+                    messages = [f"第 {no_exist.get('season')} 季缺失 {no_exist.get('total_episodes')} 集"
+                                for no_exist in no_exists.get(mediainfo.tmdb_id)]
+                    self.post_message(title=f"{mediainfo.get_title_string()}：\n" + "\n".join(messages))
                 logger.info(f"{mediainfo.get_title_string()} 媒体库中不存在，开始搜索 ...")
                 self.post_message(
                     title=f"开始搜索 {mediainfo.type.value} {mediainfo.get_title_string()} ...", userid=userid)