init commit

core/middleware/stats.py

@@ -0,0 +1,53 @@
+from aiogram import BaseMiddleware
+from aiogram.types import Message
+from typing import Callable, Dict, Any, Awaitable
+import json
+
+class MessageStatsMiddleware(BaseMiddleware):
+    def __init__(self, stats_file: str = 'message_stats.json'):
+        self.stats_file = stats_file
+        self.stats = self.load_stats()
+
+    async def __call__(
+        self,
+        handler: Callable[[Message, Dict[str, Any]], Awaitable[Any]],
+        event: Message,
+        data: Dict[str, Any]
+    ) -> Any:
+        # 只统计群组消息
+        if event.chat.type in ['group', 'supergroup']:
+            chat_id = str(event.chat.id)
+            user_id = str(event.from_user.id if event.from_user else 0)
+
+            # 初始化统计数据
+            if chat_id not in self.stats:
+                self.stats[chat_id] = {
+                    'total_messages': 0,
+                    'users': {},
+                    'chat_title': event.chat.title
+                }
+
+            if user_id not in self.stats[chat_id]['users']:
+                self.stats[chat_id]['users'][user_id] = {
+                    'message_count': 0,
+                    'username': event.from_user.username if event.from_user else 'Unknown',
+                    'first_name': event.from_user.first_name if event.from_user else 'Unknown'
+                }
+
+            # 更新统计
+            self.stats[chat_id]['total_messages'] += 1
+            self.stats[chat_id]['users'][user_id]['message_count'] += 1
+            self.save_stats()
+
+        return await handler(event, data)
+
+    def load_stats(self) -> dict:
+        try:
+            with open(self.stats_file, 'r', encoding='utf-8') as f:
+                return json.load(f)
+        except (FileNotFoundError, json.JSONDecodeError):
+            return {}
+
+    def save_stats(self):
+        with open(self.stats_file, 'w', encoding='utf-8') as f:
+            json.dump(self.stats, f, ensure_ascii=False, indent=2)