fixbug:调整排行榜写库动作

db/message_storage.py

@@ -51,7 +51,6 @@ class MessageStorageDB(BaseDBOperator):
         """
         return self.execute_query(sql, (date,)) or []
     
-    # 在 MessageStorageDB 类中添加以下方法
     
     def get_speech_ranking(self, date: str, group_id: str, limit: int = 20) -> List[Dict]:
         """获取指定日期和群组的发言排名"""
@@ -67,3 +66,23 @@ class MessageStorageDB(BaseDBOperator):
         params = (date, group_id, limit)
         results = self.execute_query(sql, params)
         return results or []
+    
+    def insert_speech_count(self, group_id: str, wx_id: str, date: str, count: int) -> bool:
+        """插入发言统计数据
+        
+        Args:
+            group_id: 群组ID
+            wx_id: 微信ID
+            date: 日期，格式为 YYYY-MM-DD
+            count: 发言计数
+            
+        Returns:
+            是否成功插入
+        """
+        sql = """
+        INSERT INTO speech_counts (group_id, wx_id, date, count) 
+        VALUES (%s, %s, %s, %s)
+        ON DUPLICATE KEY UPDATE count = VALUES(count)
+        """
+        params = (group_id, wx_id, date, count)
+        return self.execute_update(sql, params)

message_report/write_db.py

@@ -1,32 +1,47 @@
 import logging
 from datetime import datetime, timedelta
-import pymysql
 from db.connection import DBConnectionManager
 from db.message_storage import MessageStorageDB
 
+# 配置日志
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+
 # 获取数据库连接管理器的单例
 db_manager = DBConnectionManager()
 message_db = MessageStorageDB(db_manager)
 
+
 def write_to_db():
-    """将消息统计写入数据库"""
+    """从Redis读取发言统计数据并写入数据库"""
-    try:
+    # 获取Redis连接
-        # 获取昨天的日期
+    redis_conn = db_manager.get_redis_connection()
-        yesterday = (datetime.now() - timedelta(days=1)).strftime("%Y-%m-%d")
 
-        # 获取昨天的消息统计
+    # 获取当前日期的前一天
-        message_counts = message_db.get_message_count_by_date(yesterday)
+    yesterday = (datetime.now() - timedelta(days=1)).strftime('%Y-%m-%d')
 
-        if not message_counts:
+    # 遍历Redis中所有与昨天日期相关的key，并写入数据库
-            logging.info(f"没有找到 {yesterday} 的消息记录")
+    for key_bytes in redis_conn.keys(f"*:*:{yesterday}:count"):
-            return
+        key = key_bytes.decode('utf-8')
+        parts = key.split(':')
+        group_id, wx_id, _date = parts[0], parts[1], parts[2]  # _date应该是yesterday
 
-        logging.info(f"成功统计 {yesterday} 的消息记录: {len(message_counts)} 条")
+        # 获取计数值
+        count_bytes = redis_conn.hget(key, 'count')
+        count = int(count_bytes) if isinstance(count_bytes, bytes) else 0
 
-        # 这里可以添加其他处理逻辑，如发送统计报告等
+        # 使用MessageStorageDB插入数据
+        try:
+            result = message_db.insert_speech_count(group_id, wx_id, yesterday, count)
+            if result:
+                logging.info(f"成功写入发言统计: {group_id}, {wx_id}, {yesterday}, {count}")
+            else:
+                logging.error(f"写入发言统计失败: {group_id}, {wx_id}, {yesterday}, {count}")
+        except Exception as e:
+            logging.error(f"写入发言统计出错: {e}")
 
-    except Exception as e:
-        logging.error(f"写入数据库出错: {e}")
 
 def generate_and_send_ranking(groupId, allContacts: dict):
     """生成并发送群聊发言排名"""
@@ -44,7 +59,7 @@ def generate_and_send_ranking(groupId, allContacts: dict):
         ranking_str = yesterday + "发言数量前20的用户排名:\n"
         for rank, result in enumerate(results, start=1):
             username = result['wx_id']
-            speech_count = result['count']
+            speech_count = result['speech_count']
             ranking_str += f"{rank}. {allContacts.get(username, username)}: {speech_count} 次发言\n"
 
         logging.info(f"成功生成 {yesterday} 的群聊 {groupId} 发言排名")
@@ -54,5 +69,6 @@ def generate_and_send_ranking(groupId, allContacts: dict):
         logging.error(f"生成发言排名出错: {e}")
         return f"生成发言排名出错: {e}"
 
+
 if __name__ == '__main__':
     write_to_db()