59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
import logging
|
|
from datetime import datetime, timedelta
|
|
import pymysql
|
|
from db.connection import DBConnectionManager
|
|
from db.message_storage import MessageStorageDB
|
|
|
|
# 获取数据库连接管理器的单例
|
|
db_manager = DBConnectionManager()
|
|
message_db = MessageStorageDB(db_manager)
|
|
|
|
def write_to_db():
|
|
"""将消息统计写入数据库"""
|
|
try:
|
|
# 获取昨天的日期
|
|
yesterday = (datetime.now() - timedelta(days=1)).strftime("%Y-%m-%d")
|
|
|
|
# 获取昨天的消息统计
|
|
message_counts = message_db.get_message_count_by_date(yesterday)
|
|
|
|
if not message_counts:
|
|
logging.info(f"没有找到 {yesterday} 的消息记录")
|
|
return
|
|
|
|
logging.info(f"成功统计 {yesterday} 的消息记录: {len(message_counts)} 条")
|
|
|
|
# 这里可以添加其他处理逻辑，如发送统计报告等
|
|
|
|
except Exception as e:
|
|
logging.error(f"写入数据库出错: {e}")
|
|
|
|
def generate_and_send_ranking(groupId, allContacts: dict):
|
|
"""生成并发送群聊发言排名"""
|
|
try:
|
|
yesterday = (datetime.now() - timedelta(days=1)).strftime('%Y-%m-%d')
|
|
|
|
# 使用数据库操作类获取排名数据
|
|
results = message_db.get_speech_ranking(yesterday, groupId, limit=20)
|
|
|
|
if not results:
|
|
logging.info(f"没有找到 {yesterday} 的群聊 {groupId} 发言记录")
|
|
return f"{yesterday} 没有发言记录"
|
|
|
|
# 格式化输出字符串
|
|
ranking_str = yesterday + "发言数量前20的用户排名:\n"
|
|
for rank, result in enumerate(results, start=1):
|
|
username = result['wx_id']
|
|
speech_count = result['count']
|
|
ranking_str += f"{rank}. {allContacts.get(username, username)}: {speech_count} 次发言\n"
|
|
|
|
logging.info(f"成功生成 {yesterday} 的群聊 {groupId} 发言排名")
|
|
return ranking_str
|
|
|
|
except Exception as e:
|
|
logging.error(f"生成发言排名出错: {e}")
|
|
return f"生成发言排名出错: {e}"
|
|
|
|
if __name__ == '__main__':
|
|
write_to_db()
|