聊天记录统计功能

message_report/write_db.py

@@ -37,29 +37,30 @@ def write_to_db():
     conn.close()
 
 
-def generate_and_send_ranking():
+def generate_and_send_ranking(groupId,allContacts: dict):
     # 连接到SQLite数据库（假设数据库文件名为database.db）
     conn = sqlite3.connect('database.db')
     cursor = conn.cursor()
 
     # 编写SQL查询来获取发言数量前20的用户
     query = """
-        SELECT group_id,wx_id, COUNT(*) AS speech_count
-      FROM speech_counts
-      WHERE DATE(date) = DATE('now', '-1 day')
-      GROUP BY group_id,wx_id
-      ORDER BY count DESC
-      LIMIT 20
+      SELECT group_id,wx_id, COUNT(*) AS speech_count
+        FROM speech_counts
+        WHERE DATE(date) = DATE('now', '-1 day')
+        AND group_id = ?
+        GROUP BY group_id,wx_id
+        ORDER BY count DESC
+        LIMIT 20
       """
 
     # 执行查询并获取结果
-    cursor.execute(query)
+    cursor.execute(query,(groupId,))
     results = cursor.fetchall()
 
     # 格式化输出字符串
     ranking_str = "发言数量前20的用户排名:\n"
     for rank, (username, speech_count) in enumerate(results, start=1):
-        ranking_str += f"{rank}. {username}: {speech_count} 次发言\n"
+        ranking_str += f"{rank}. {allContacts[username]}: {speech_count} 次发言\n"
 
     # 关闭数据库连接
     conn.close()