fix group

wechat-server/web.py

@@ -2,6 +2,7 @@
 # -*- coding: utf-8 -*-
 import os
 import time
+from collections import defaultdict
 from functools import lru_cache
 from xml.etree.ElementTree import fromstring
 
@@ -82,14 +83,7 @@ def reply(external_user_id, open_kfid, content):
     return resp.get("errcode", 1) == 0
 
 
-# 企业微信同一条用户的消息会发三次回调，内容都一样
-# 这里加个 lru_cache 挡一挡，纯属偷懒
-@lru_cache()
-def handle_message(message_token: str):
-
-    history = get_chat_history(access_token=access_token(),
-                               message_token=message_token)["msg_list"]
-
+def extract_messages_blocks(history: list):
     text_message_blocks = []
 
     # 这里收集一下 我们能处理的消息
@@ -109,23 +103,47 @@ def handle_message(message_token: str):
             content = message_block["image"]["media_id"]
             print("image block", content)
 
+    grouped_blocks = defaultdict(list)
+
     if len(text_message_blocks) == 0:
-        return
+        return grouped_blocks
 
-    content = "\n".join([message_block["text"]["content"] for message_block in text_message_blocks])
-    last_reply_time[text_message_blocks[-1]["external_userid"]] = text_message_blocks[-1]["send_time"]
+    for message_block in text_message_blocks:
+        grouped_blocks[message_block["external_userid"]].append(message_block)
 
-    # print(open_kfid, external_userid, content)
-    result = task.ask(content, session=False)
-    print("handle_message", result)
-    """
-    {
-        "input": "",
-        "chat_history": "",
-        "text": ""
-    }
-    """
-    reply(external_userid, open_kfid, result["text"])
+    return grouped_blocks
+
+# 企业微信同一条用户的消息会发三次回调，内容都一样
+# 这里加个 lru_cache 挡一挡，纯属偷懒
+
+
+@lru_cache()
+def handle_message(message_token: str):
+
+    history = get_chat_history(access_token=access_token(),
+                               message_token=message_token)
+    # 实际上这里能拿到所有的用户发来的消息，所以这里要做一下分组
+    # TODO 翻页
+    grouped_blocks = extract_messages_blocks(history["msg_list"])
+    if len(grouped_blocks) == 0:
+        return
+
+    for external_userid, text_message_blocks in grouped_blocks.items():
+        content = "\n".join([message_block["text"]["content"] for message_block in text_message_blocks])
+        open_kfid = text_message_blocks[-1].get("open_kfid", None)
+        last_reply_time[text_message_blocks[-1]["external_userid"]] = text_message_blocks[-1]["send_time"]
+
+        # print(open_kfid, external_userid, content)
+        result = task.ask(content, session=False)
+        print("handle_message", result)
+        """
+        {
+            "input": "",
+            "chat_history": "",
+            "text": ""
+        }
+        """
+        reply(external_userid, open_kfid, result["text"])
 
 
 @app.get("/")