Лимит запросов по IP, мелкие правки

user_api/func.py

@@ -1,5 +1,9 @@
 from datetime import datetime
 import pytz
+import asyncio
+import time
+from collections import defaultdict, deque
+from typing import Dict, Deque
 
 from db import user_token_in_db_func
 
@@ -37,4 +41,44 @@ async def append_line_with_limit(text, new_line, max_lines = 5000) -> str:
         lines.pop(0)
 
     # Join lines back together with newlines
-    return '\n'.join(lines)
+    return '\n'.join(lines)
+
+
+class UserRateLimiter:
+    def __init__(self):
+        # Store request timestamps for each IP
+        self.ip_requests: Dict[str, Deque[float]] = defaultdict(deque)
+        self.lock = asyncio.Lock()
+
+    async def is_rate_limited(self, ip: str) -> bool:
+        """
+        Check if user with given IP is making too many requests.
+
+        Args:
+            ip (str): User's IP address
+
+        Returns:
+            bool: True if user makes more than 30 requests per minute, False otherwise
+        """
+        async with self.lock:
+            current_time = time.time()
+
+            # Remove timestamps older than 60 seconds
+            while self.ip_requests[ip] and current_time - self.ip_requests[ip][0] > 60:
+                self.ip_requests[ip].popleft()
+
+            # Add current request
+            self.ip_requests[ip].append(current_time)
+
+            # Log the IP and request count
+            request_count = len(self.ip_requests[ip])
+            print(f"IP: {ip}, Requests in last minute: {request_count}")
+
+            # Return True if rate limit exceeded (more than 30 requests per minute)
+            return request_count > 30
+
+# Global instance
+rate_limiter = UserRateLimiter()
+
+async def is_rate_limited(ip: str) -> bool:
+    return await rate_limiter.is_rate_limited(ip)