测试入库

message_storage/message_to_db.py

@@ -58,14 +58,15 @@ class MessageStorage:
         yesterday = (datetime.now() - timedelta(days=1)).strftime('%Y-%m-%d')
 
         # 遍历Redis中所有与昨天日期相关的key，并写入数据库
-        for key_bytes in redis_conn.keys(f"*:*:{yesterday}:count"):
-            key = key_bytes.decode('utf-8')
+        for key_item in redis_conn.keys(f"*:*:{yesterday}:count"):
+            # 检查key是否为字节类型，如果是则解码
+            key = key_item.decode('utf-8') if isinstance(key_item, bytes) else key_item
             parts = key.split(':')
             group_id, wx_id, _date = parts[0], parts[1], parts[2]  # _date应该是yesterday
 
             # 获取计数值
             count_bytes = redis_conn.hget(key, 'count')
-            count = int(count_bytes) if isinstance(count_bytes, bytes) else 0
+            count = int(count_bytes) if isinstance(count_bytes, bytes) else int(count_bytes) if count_bytes else 0
 
             # 使用MessageStorageDB插入数据
             try: