新闻更新

message_report/write_db.py

@@ -27,7 +27,7 @@ def write_to_db():
 
     # 遍历Redis中所有与昨天日期相关的key，并写入数据库
     for key in r.keys(f"*:*:{yesterday}:count"):
-        # print('user key：' + key)
+        print('user key：' + key)
         parts = key.decode('utf-8').split(':')
         group_id, wx_id, _date = parts[0], parts[1], parts[2]  # _date应该是yesterday，但这里为了完整性还是保留
         count = int(r.hget(key, 'count')) if isinstance(r.hget(key, 'count'), bytes) else 0  # 处理可能的None或空值情况
@@ -48,11 +48,11 @@ def generate_and_send_ranking(groupId, allContacts: dict):
     yesterday = (datetime.now() - timedelta(days=1)).strftime('%Y-%m-%d')
     # 编写SQL查询来获取发言数量前20的用户
     query = """
-      SELECT group_id,wx_id, COUNT(*) AS speech_count
+      SELECT 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
+        GROUP BY wx_id
         ORDER BY count DESC
         LIMIT 20
       """
@@ -71,3 +71,6 @@ def generate_and_send_ranking(groupId, allContacts: dict):
     # 这里我们没有实际“发送”排名，只是返回字符串
     # 如果需要发送，可以在此添加发送逻辑
     return ranking_str
+
+if __name__ == '__main__':
+    write_to_db()