From 19c94364a4f3b1643e82628abffc952c3b714e7f Mon Sep 17 00:00:00 2001 From: liuwei Date: Thu, 6 Feb 2025 12:19:48 +0800 Subject: [PATCH] =?UTF-8?q?=E5=8A=A0=E5=85=A5=E7=BB=84ID?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- message_storage/message.sql | 1 + message_storage/message_to_db.py | 11 ++++++----- robot.py | 2 +- 3 files changed, 8 insertions(+), 6 deletions(-) diff --git a/message_storage/message.sql b/message_storage/message.sql index 1923465..40e0757 100644 --- a/message_storage/message.sql +++ b/message_storage/message.sql @@ -5,6 +5,7 @@ USE message_archive; CREATE TABLE messages ( id INT AUTO_INCREMENT PRIMARY KEY, + group_id varchar(20) not null , timestamp VARCHAR(20) NOT NULL, sender VARCHAR(255) NOT NULL, content TEXT NULL, diff --git a/message_storage/message_to_db.py b/message_storage/message_to_db.py index da67fad..7a1a497 100644 --- a/message_storage/message_to_db.py +++ b/message_storage/message_to_db.py @@ -9,7 +9,7 @@ db_config = { } -def archive_message(timestamp_str, sender, content, message_type, attachment_url=None): +def archive_message(group_id, timestamp_str, sender, content, message_type, attachment_url=None): # 连接到数据库 connection = pymysql.connect(**db_config) @@ -17,10 +17,10 @@ def archive_message(timestamp_str, sender, content, message_type, attachment_url with connection.cursor() as cursor: # 插入消息信息 sql = """ - INSERT INTO messages (timestamp, sender, content, message_type, attachment_url) - VALUES (%s, %s, %s, %s, %s) + INSERT INTO messages (group_id,timestamp, sender, content, message_type, attachment_url) + VALUES (%s, %s, %s, %s, %s, %s) """ - cursor.execute(sql, (timestamp_str, sender, content, message_type, attachment_url)) + cursor.execute(sql, (group_id, timestamp_str, sender, content, message_type, attachment_url)) # 提交事务 connection.commit() @@ -37,10 +37,11 @@ def archive_message(timestamp_str, sender, content, message_type, attachment_url # 示例用法 if __name__ == "__main__": + group_id ='XXX@123123' timestamp_str = "2025-02-06 11:15:28" sender = "XXX" content = "This is a test message with a string timestamp." message_type = "text" attachment_url = "http://example.com/attachment.pdf" # 可以为None如果没有附件 - archive_message(timestamp_str, sender, content, message_type, attachment_url) \ No newline at end of file + archive_message(group_id,timestamp_str, sender, content, message_type, attachment_url) diff --git a/robot.py b/robot.py index 11c50c0..d0fbbd1 100644 --- a/robot.py +++ b/robot.py @@ -163,7 +163,7 @@ class Robot(Job): # 聊天记录入库动作： try: now_time = str(datetime.now().strftime("%Y-%m-%d %H:%M:%S")) - archive_message(now_time,msg.sender,msg.content,msg.type,msg.extra) + archive_message(msg.roomid,now_time,msg.sender,msg.content,msg.type,msg.extra) except Exception as e: self.LOG.error(f"archive_message error: {e}")