diff --git a/main.py b/main.py index efc02bf..6d1f852 100644 --- a/main.py +++ b/main.py @@ -58,8 +58,10 @@ def main(chat_type: int): robot.onEveryTime("10:30", robot.sendEpicFreeGames) - # message report + # message report 1:数据自动从redis 转到sqllite robot.onEveryTime("00:30", robot.messageCountToDB) + # 从db中提取并发送给相关群 + robot.onEveryTime("09:30", robot.generateAndSendRanking) # 让机器人一直跑 robot.keepRunningAndBlockProcess() diff --git a/message_report/db_init.sql b/message_report/db_init.sql index 19c27cb..893c9d7 100644 --- a/message_report/db_init.sql +++ b/message_report/db_init.sql @@ -1,7 +1,9 @@ -CREATE TABLE IF NOT EXISTS speech_counts ( - group_id TEXT, - wx_id TEXT, - date TEXT, - count INTEGER, - PRIMARY KEY (group_id, wx_id, date) - ) \ No newline at end of file +CREATE TABLE IF NOT EXISTS speech_counts +( + group_id TEXT, + wx_id TEXT, + date TEXT, + count INTEGER, + PRIMARY KEY (group_id, wx_id, date) +) + diff --git a/message_report/process_message.py b/message_report/process_message.py index 29c0803..e1d6ff0 100644 --- a/message_report/process_message.py +++ b/message_report/process_message.py @@ -8,7 +8,7 @@ from wcferry import WxMsg r = redis.Redis(host='192.168.2.32', port=6379, db=0) -def process_message(message:WxMsg): +def process_message(message: WxMsg): # 示例message字符串 current_date = datetime.now().strftime('%Y-%m-%d') # 生成Redis key @@ -17,7 +17,5 @@ def process_message(message:WxMsg): # 使用Redis哈希(或字符串)增加发言次数 r.hincrby(key, 'count', 1) # 这里使用哈希,但也可以考虑用字符串的INCR操作 # 设置时效为48小时 - r.expire(key,86400*2) + r.expire(key, 86400 * 2) # 或者使用字符串:r.incr(key) # 如果只存储一个整数值,字符串类型可能更简单 - - diff --git a/message_report/write_db.py b/message_report/write_db.py index aaaf862..f045d01 100644 --- a/message_report/write_db.py +++ b/message_report/write_db.py @@ -3,9 +3,11 @@ import time import sqlite3 from datetime import datetime, timedelta import redis + # 连接到Redis r = redis.Redis(host='192.168.2.32', port=6379, db=0) + def write_to_db(): # 连接到SQLite数据库 conn = sqlite3.connect('message_stats.db') @@ -25,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或空值情况 @@ -38,7 +40,7 @@ def write_to_db(): conn.close() -def generate_and_send_ranking(groupId,allContacts: dict): +def generate_and_send_ranking(groupId, allContacts: dict): # 连接到SQLite数据库(假设数据库文件名为database.db) conn = sqlite3.connect('message_stats.db') cursor = conn.cursor() @@ -56,19 +58,16 @@ def generate_and_send_ranking(groupId,allContacts: dict): """ # 执行查询并获取结果 - cursor.execute(query,(groupId,)) + cursor.execute(query, (groupId,)) results = cursor.fetchall() # 格式化输出字符串 ranking_str = yesterday + "发言数量前20的用户排名:\n" 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" # 关闭数据库连接 conn.close() # 这里我们没有实际“发送”排名,只是返回字符串 # 如果需要发送,可以在此添加发送逻辑 return ranking_str - - diff --git a/message_sign/readme.md b/message_sign/readme.md new file mode 100644 index 0000000..c9aaf2b --- /dev/null +++ b/message_sign/readme.md @@ -0,0 +1,2 @@ +# 签到功能 +## 用户每日签到,可以汇报签到次数 diff --git a/task/task_list.py b/task/task_list.py index 62047dd..3768b08 100644 --- a/task/task_list.py +++ b/task/task_list.py @@ -24,7 +24,7 @@ def add_task(description, time_str): conn = sqlite3.connect(DB_FILE) c = conn.cursor() time_obj = datetime.datetime.strptime(time_str, '%Y-%m-%d %H:%M:%S') - c.execute("INSERT INTO tasks (description, reminder_time) VALUES (?, ?)", + c.execute("INSERT INTO tasks (des cription, reminder_time) VALUES (?, ?)", (description, time_obj.isoformat())) conn.commit() conn.close()