Initial commit: VK Sales Bot project structure

utils/middleware.py

@@ -0,0 +1,71 @@
+from loguru import logger
+from typing import Dict, List
+from datetime import datetime, timedelta
+
+
+class MiddlewareChain:
+    """Простая цепочка мидлваров (логирование, антиспам, аналитика)"""
+    def __init__(self):
+        self.middlewares: List[callable] = []
+
+    def add(self, middleware: callable):
+        self.middlewares.append(middleware)
+
+    def process(self, user_id: int, text: str) -> bool:
+        for mw in self.middlewares:
+            if not mw(user_id, text):
+                return False
+        return True
+
+
+# ==================== Антиспам-фильтр ====================
+
+class SpamFilter:
+    """Простой антиспам-фильтр: блокирует пользователя, если он отправляет
+    более `max_messages` сообщений за `window_seconds` секунд."""
+
+    def __init__(self, max_messages: int = 5, window_seconds: int = 60):
+        self.max_messages = max_messages
+        self.window_seconds = window_seconds
+        # user_id -> [timestamp1, timestamp2, ...]
+        self._messages: Dict[int, List[float]] = {}
+
+    def is_spam(self, user_id: int) -> bool:
+        now = datetime.now().timestamp()
+        window_start = now - self.window_seconds
+
+        # Инициализируем список, если нет
+        if user_id not in self._messages:
+            self._messages[user_id] = []
+
+        # Удаляем старые записи за пределами окна
+        self._messages[user_id] = [
+            ts for ts in self._messages[user_id] if ts > window_start
+        ]
+
+        # Проверяем, не превышен ли лимит
+        if len(self._messages[user_id]) >= self.max_messages:
+            return True
+
+        # Записываем текущее сообщение
+        self._messages[user_id].append(now)
+        return False
+
+    def reset(self, user_id: int):
+        """Сброс счётчика для пользователя (например, при /start)"""
+        self._messages.pop(user_id, None)
+
+
+# ==================== Логирование ====================
+
+def logging_middleware(user_id: int, text: str) -> bool:
+    logger.info(f"Processed message from {user_id}: {text}")
+    return True
+
+
+# ==================== Пример: мидлвар для аналитики ====================
+
+def analytics_middleware(user_id: int, text: str) -> bool:
+    """Заглушка для аналитики — можно расширить позже."""
+    logger.debug(f"Analytics: user {user_id} sent '{text}'")
+    return True