feat(update database when the message has been edited)

Author: X1A0CA1

README.md

@@ -24,6 +24,7 @@ Telegram自带搜索对CJK的支持仅限于整句，不支持分词。本项目
 - 消息记录搜索
 - 消息链接定位
 - 支持多个群组
+- 消息编辑后数据库同步更新
 
 ![搜索](https://raw.githubusercontent.com/Taosky/telegram-search-bot/master/preview/search.png)
 
@@ -73,6 +74,9 @@ Telegram自带搜索对CJK的支持仅限于整句，不支持分词。本项目
 - Inline Mode placeholder修改需要重启客户端
 
 ## Update Records
+#### 2022-10-31
+- 支持消息编辑后数据库同步更新
+
 #### 2022-10-30
 - 支持索引频道、匿名管理消息。
 - 修复了一些 BUG
@@ -87,6 +91,9 @@ Telegram自带搜索对CJK的支持仅限于整句，不支持分词。本项目
 - 修复Message ID重复的问题
 - 修复导入历史记录报错的问题
 
+<details>
+<summary>more</summary>
+
 #### 2022-02-17
 - 记录和搜索支持多个群组（数据库有变化，要重新导入历史记录）
 - 搜索时用户名后显示"@群组"用于区分消息来源
@@ -103,9 +110,6 @@ Telegram自带搜索对CJK的支持仅限于整句，不支持分词。本项目
 #### 2022-01-06
 - Docker化
 
-<details>
-<summary>more</summary>
-
 #### 2021-09-20
 - 更新python-telegram-bot库
 - 重构代码，简化操作

user_handlers/msg_store.py

@@ -35,14 +35,52 @@ def insert_message(msg_id, msg_link, msg_text, msg_video, msg_photo, msg_audio,
     session.close()
 
 
+def update_message(from_chat, msg_id, msg_text):
+    session = DBSession()
+    session.query(Message)\
+        .filter(Message.from_chat.is_(from_chat))\
+        .filter(Message.id.is_(msg_id))\
+        .update({"text": msg_text})
+    session.commit()
+    session.close()
+
+
 def store_message(update, context):
+    print(update)
     session = DBSession()
     chat_ids = [chat.id for chat in session.query(Chat) if chat.enable]
-    if update.effective_chat.id not in chat_ids \
-            or update.message.via_bot:
+    if update.effective_chat.id not in chat_ids:
         return
 
     '''
+    判断是否是 Edited 消息，如果是，根据 GroupID 和 MessageID 在数据库中搜索现存的消息并更新。
+    
+    关于图片，音频等媒体的更新我个人并不是很想写，即使更新了在目前来看也没啥特别大的用处，图片并不像文字一样没有良好的分词就无法查询，
+    完全可以使用 TG 自带的图片搜索来解决这个问题。所以我这部分就只更新了文字消息，并不更新其他的任何消息。
+    
+    除此之外，这里还判断了被编辑消息的时间，如果 原消息发布时间 和 编辑消息的时间 差距过大的话则不更新，以避免 userbot 的 dme 炸库。
+    '''
+    if update.edited_message:
+        # 判断被编辑消息的间隔
+        if update.edited_message.edit_date - update.edited_message.date > 120:
+            return
+
+        if update.edited_message.text:
+            msg_text = update.edited_message.text if update.edited_message.text else ''
+        elif update.edited_message.caption:
+            msg_text = update.edited_message.caption if update.edited_message.caption else ''
+        else:
+            return
+
+        msg_id = update.edited_message.message_id
+        chat_id = update.edited_message.chat.id
+
+        update_message(chat_id, msg_id, msg_text)
+        return
+
+    if update.message.via_bot:
+        return
+    '''
     这里的 if 判断发言是用户还是频道或者是 group。
     需要注意的是，这里并不能排除 BOT，因为 Telegram 为了向后兼容，Anon group 实体会附带有一个 from 参数，里面 is_bot 是 true. 如下
     "from": {
@@ -101,4 +139,7 @@ def store_message(update, context):
     update_chat(chat_id, chat_title)
 
 
-handler = MessageHandler(Filters.text | Filters.video | Filters.photo | Filters.audio | Filters.voice, store_message)
+handler = MessageHandler(
+    Filters.text | Filters.video | Filters.photo | Filters.audio | Filters.voice,
+    store_message,
+)