izzzzzi
0657136ae5
- Упорядочивание и группировка импортов в соответствии с PEP 8 - Удаление неиспользуемых импортов - Перемещение локальных импортов в начало списка - Добавление пустых строк для улучшения читаемости - Систематизация порядка импортов в различных модулях проекта
62 lines
2.4 KiB
Python
62 lines
2.4 KiB
Python
from collections.abc import Awaitable, Callable
|
|
from typing import Any, Dict, Optional, TypedDict, Union
|
|
|
|
from aiogram import BaseMiddleware
|
|
from aiogram.types import CallbackQuery, InlineQuery, Message, TelegramObject, User
|
|
|
|
from logger import logger
|
|
|
|
|
|
class UserInfo(TypedDict):
|
|
user_id: int | None
|
|
username: str | None
|
|
action: str | None
|
|
|
|
|
|
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
|