izzzzzi
b08040f46b
- Рефакторинг middlewares с добавлением типизации и улучшенной логики - Оптимизация SessionMiddleware с использованием пула соединений - Улучшение обработки пользователей и логирования в UserMiddleware - Расширение функциональности ThrottlingMiddleware с более гибким управлением - Добавление дополнительных проверок и логирования в AdminMiddleware - Оптимизация функции upsert_user с поддержкой передачи существующей сессии
66 lines
2.5 KiB
Python
66 lines
2.5 KiB
Python
from collections.abc import Awaitable, Callable
|
|
from typing import Any, Dict, Optional, Union, TypedDict
|
|
|
|
from aiogram import BaseMiddleware
|
|
from aiogram.types import CallbackQuery, InlineQuery, Message, TelegramObject, User
|
|
|
|
from logger import logger
|
|
|
|
|
|
class UserInfo(TypedDict):
|
|
user_id: Optional[int]
|
|
username: Optional[str]
|
|
action: Optional[str]
|
|
|
|
|
|
class LoggingMiddleware(BaseMiddleware):
|
|
"""Middleware для логирования действий пользователя."""
|
|
|
|
async def __call__(
|
|
self,
|
|
handler: Callable[[TelegramObject, Dict[str, Any]], Awaitable[Any]],
|
|
event: TelegramObject,
|
|
data: Dict[str, Any],
|
|
) -> Any:
|
|
user_info = self._extract_user_info(event)
|
|
|
|
if user_info["user_id"]:
|
|
logger.info(
|
|
f"Активность пользователя - "
|
|
f"ID пользователя: {user_info['user_id']}, "
|
|
f"Имя пользователя: {user_info['username'] or 'Не указано'}, "
|
|
f"Действие: {user_info['action'] or 'Неизвестно'}"
|
|
)
|
|
|
|
return await handler(event, data)
|
|
|
|
def _extract_user_info(self, event: TelegramObject) -> UserInfo:
|
|
"""Извлекает информацию о пользователе из различных типов событий.
|
|
|
|
Args:
|
|
event: Событие Telegram
|
|
|
|
Returns:
|
|
Словарь с информацией о пользователе
|
|
"""
|
|
result: UserInfo = {
|
|
"user_id": None,
|
|
"username": None,
|
|
"action": None
|
|
}
|
|
|
|
# Обработка общих атрибутов для разных типов событий
|
|
if hasattr(event, "from_user") and isinstance(event.from_user, User):
|
|
result["user_id"] = event.from_user.id
|
|
result["username"] = event.from_user.username
|
|
|
|
# Определение типа действия в зависимости от типа события
|
|
if isinstance(event, Message):
|
|
result["action"] = f"Сообщение: {event.text}"
|
|
elif isinstance(event, CallbackQuery):
|
|
result["action"] = f"Обратный вызов: {event.data}"
|
|
elif isinstance(event, InlineQuery):
|
|
result["action"] = f"Inline запрос: {event.query}"
|
|
|
|
return result
|