liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

key 设置48小时时效

Browse Source
This commit is contained in:
liuwei
2025-02-05 14:02:00 +08:00
parent 4c13b552d8
commit f0a5cfe092
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
message_report/process_message.py
View File

@@ -13,8 +13,11 @@ def process_message(message:WxMsg):
current_date = datetime.now().strftime('%Y-%m-%d') current_date = datetime.now().strftime('%Y-%m-%d')
# 生成Redis key # 生成Redis key
key = f"{message.roomid}:{message.sender}:{current_date}:count" key = f"{message.roomid}:{message.sender}:{current_date}:count"
# 使用Redis哈希（或字符串）增加发言次数 # 使用Redis哈希（或字符串）增加发言次数
r.hincrby(key, 'count', 1) # 这里使用哈希，但也可以考虑用字符串的INCR操作 r.hincrby(key, 'count', 1) # 这里使用哈希，但也可以考虑用字符串的INCR操作
# 设置时效为48小时
r.expire(key,86400*2)
# 或者使用字符串：r.incr(key) # 如果只存储一个整数值，字符串类型可能更简单 # 或者使用字符串：r.incr(key) # 如果只存储一个整数值，字符串类型可能更简单

3
message_report/write_db.py
View File

@@ -43,6 +43,7 @@ def generate_and_send_ranking(groupId,allContacts: dict):
conn = sqlite3.connect('message_stats.db') conn = sqlite3.connect('message_stats.db')
cursor = conn.cursor() cursor = conn.cursor()
yesterday = (datetime.now() - timedelta(days=1)).strftime('%Y-%m-%d')
# 编写SQL查询来获取发言数量前20的用户 # 编写SQL查询来获取发言数量前20的用户
query = """ query = """
SELECT group_id,wx_id, COUNT(*) AS speech_count SELECT group_id,wx_id, COUNT(*) AS speech_count
@@ -59,7 +60,7 @@ def generate_and_send_ranking(groupId,allContacts: dict):
results = cursor.fetchall() results = cursor.fetchall()
# 格式化输出字符串 # 格式化输出字符串
ranking_str = "发言数量前20的用户排名:\n" ranking_str = yesterday + "发言数量前20的用户排名:\n"
for rank, (username, speech_count) in enumerate(results, start=1): for rank, (username, speech_count) in enumerate(results, start=1):
ranking_str += f"{rank}. {allContacts[username]}: {speech_count} 次发言\n" ranking_str += f"{rank}. {allContacts[username]}: {speech_count} 次发言\n"