聊天记录统计功能

message_report/process_message.py

@@ -0,0 +1,28 @@
+import redis
+from datetime import datetime
+import re
+# 连接到Redis
+r = redis.Redis(host='192.168.2.32', port=6379, db=0)
+
+
+def process_message(message):
+    # 示例message字符串
+
+    # 使用正则表达式提取用户名称和聊天室标识
+    user_info_regex = re.compile(r"(\w+)\[([^\]]+)\]")
+    match = user_info_regex.search(message)
+    if match:
+        user_name = match.group(1)
+        chatroom_id = match.group(2)
+        print(f"用户名称: {user_name}")
+        print(f"聊天室标识: {chatroom_id}")
+        current_date = datetime.now().strftime('%Y-%m-%d')
+        # 生成Redis key
+        key = f"{chatroom_id}:{user_name}:{current_date}:count"
+        # 使用Redis哈希（或字符串）增加发言次数
+        print(key)
+        r.hincrby(key, 'count', 1)  # 这里使用哈希，但也可以考虑用字符串的INCR操作
+        # 或者使用字符串：r.incr(key)  # 如果只存储一个整数值，字符串类型可能更简单
+    else:
+        print("Failed to extract user name or chatroom ID.")
+