56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
import logging
|
||
from typing import Callable, Dict, Any, Awaitable
|
||
from aiogram import BaseMiddleware
|
||
from aiogram.enums import ChatType
|
||
from aiogram.types import Message
|
||
from database import db
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
class MessageCounterMiddleware(BaseMiddleware):
|
||
"""
|
||
Middleware для подсчёта сообщений в группах и супергруппах.
|
||
"""
|
||
|
||
async def __call__(
|
||
self,
|
||
handler: Callable[[Message, Dict[str, Any]], Awaitable[Any]],
|
||
event: Any,
|
||
data: Dict[str, Any]
|
||
) -> Any:
|
||
if not isinstance(event, Message):
|
||
return await handler(event, data)
|
||
|
||
# Проверяем, что сообщение пришло из группового чата и не от бота
|
||
if (event.chat.type in (ChatType.GROUP, ChatType.SUPERGROUP) and
|
||
not event.from_user.is_bot):
|
||
try:
|
||
await self.process_group_message(event)
|
||
except Exception as e:
|
||
logger.error(msg=f"Ошибка при обработке сообщения: {e}", exc_info=True)
|
||
|
||
return await handler(event, data)
|
||
|
||
@staticmethod
|
||
async def process_group_message(message: Message) -> None:
|
||
"""
|
||
Обработка сообщения из группового чата.
|
||
"""
|
||
user_id = message.from_user.id
|
||
message_text = message.text or message.caption or ""
|
||
|
||
# Добавляем пользователя (если его ещё нет)
|
||
await db.add_user(
|
||
user_id=user_id,
|
||
username=message.from_user.username,
|
||
full_name=message.from_user.full_name,
|
||
)
|
||
|
||
# Сохраняем сообщение
|
||
await db.add_message(
|
||
user_id=user_id,
|
||
message_text=message_text,
|
||
created_at=message.date,
|
||
)
|
||
logger.info(f"Сообщение от пользователя {user_id} сохранено в БД")
|