feat(search): Adds ratelimiter to throttle csv exports per day

freelawproject · Jan 6, 2025 · 725a2e9 · 725a2e9
1 parent f6bc9e1
commit 725a2e9
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cl/lib/ratelimiter.py b/cl/lib/ratelimiter.py
@@ -66,6 +66,7 @@ def get_path_to_make_key(group: str, request: HttpRequest) -> str:
 if "test" in sys.argv:
     ratelimiter_all_2_per_m = lambda func: func
     ratelimiter_unsafe_3_per_m = lambda func: func
+    ratelimiter_unsafe_5_per_d = lambda func: func
     ratelimiter_unsafe_10_per_m = lambda func: func
     ratelimiter_all_10_per_h = lambda func: func
     ratelimiter_unsafe_2000_per_h = lambda func: func
@@ -79,6 +80,11 @@ def get_path_to_make_key(group: str, request: HttpRequest) -> str:
         rate="3/m",
         method=UNSAFE,
     )
+    ratelimiter_unsafe_5_per_d = ratelimit(
+        key=get_ip_for_ratelimiter,
+        rate="5/d",
+        method=UNSAFE,
+    )
     ratelimiter_unsafe_10_per_m = ratelimit(
         key=get_ip_for_ratelimiter,
         rate="10/m",