eee

utils.py

@@ -0,0 +1,30 @@
+from collections import defaultdict
+import time
+import os
+from dotenv import load_dotenv
+import hashlib
+
+load_dotenv()
+MAX_REQUESTS_PER_HOUR = int(os.getenv("MAX_REQUESTS_PER_HOUR"))
+user_requests = defaultdict(list)
+
+
+def user_is_limited(hashed_id: str) -> bool:
+    current_time = time.time()
+
+    # Prune old requests (older than 1 hour)
+    user_requests[hashed_id] = [
+        ts for ts in user_requests[hashed_id] if current_time - ts < 3600
+    ]
+
+    if len(user_requests[hashed_id]) >= MAX_REQUESTS_PER_HOUR:
+        return True
+
+    user_requests[hashed_id].append(current_time)
+    return False
+
+
+def get_hashed_id(user_id: str) -> str:
+    daily_date = time.strftime("%Y-%m-%d")
+    combined = f"{user_id}{daily_date}"
+    return hashlib.sha256(combined.encode()).hexdigest()[:16]