feat: add time_frame parameter for static methods of Limiter

TexteaInc · Oct 27, 2023 · 5943282 · 5943282
1 parent b88c536
commit 5943282
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/backend/funix/decorator/__init__.py b/backend/funix/decorator/__init__.py
@@ -284,12 +284,16 @@ def __init__(
         self.call_history = {}
 
     @staticmethod
-    def ip(max_calls: int):
-        return Limiter(max_calls=max_calls, source=LimitSource.IP)
+    def ip(max_calls: int, time_frame: int = 60):
+        return Limiter(
+            max_calls=max_calls, time_frame=time_frame, source=LimitSource.IP
+        )
 
     @staticmethod
-    def session(max_calls: int):
-        return Limiter(max_calls=max_calls, source=LimitSource.SESSION)
+    def session(max_calls: int, time_frame: int = 60):
+        return Limiter(
+            max_calls=max_calls, time_frame=time_frame, source=LimitSource.SESSION
+        )
 
     def rate_limit(self) -> Optional[Response]:
         call_history = self.call_history