diff --git a/message_storage/message_to_db.py b/message_storage/message_to_db.py index 0c68734..1898088 100644 --- a/message_storage/message_to_db.py +++ b/message_storage/message_to_db.py @@ -36,7 +36,7 @@ def archive_message(group_id, timestamp_str, sender, content, message_type, atta connection.close() -def get_messages(all_contacts: dict): +def get_messages(group_id, all_contacts: dict): # 连接到数据库 connection = pymysql.connect(**db_config) @@ -53,9 +53,9 @@ def get_messages(all_contacts: dict): query = """ SELECT group_id, timestamp, sender, content FROM messages - WHERE timestamp >= %s AND message_type =1 + WHERE timestamp >= %s AND message_type =1 and group_id = %s """ - cursor.execute(query, (eight_hours_ago_str,)) + cursor.execute(query, (eight_hours_ago_str, group_id)) # 提取结果并组成带逗号的字符串 result = [] diff --git a/robot.py b/robot.py index e04c9b3..ff653ef 100644 --- a/robot.py +++ b/robot.py @@ -366,7 +366,7 @@ class Robot(Job): def message_summary_robot(self, sender: str = None): try: - content = get_messages(self.allContacts) + content = get_messages(sender, self.allContacts) summary = message_summary(content) self.sendTextMsg(summary, sender) except Exception as e: