feat(search): Adds logic to download search results

cl/lib/elasticsearch_utils.py

@@ -105,6 +105,40 @@ def wrapper_func(*args, **kwargs) -> Any:
     return wrapper_func
 
 
+class CSVSerializableDocumentMixin:
+
+    @classmethod
+    def get_csv_headers(cls) -> list[str]:
+        """
+        Returns a list of strings representing the headers for a CSV file.
+
+        This method defines the column headers for a CSV representation of the
+        data associated with this class.
+
+        :return: A list of strings, where each string is a column header.
+        """
+        raise NotImplementedError(
+            "Subclass must implement get_csv_headers method"
+        )
+
+    @classmethod
+    def get_csv_transformations(cls) -> dict[str, Callable[..., Any]]:
+        """
+        Generates a dictionary of transformation functions for CSV export.
+
+        This method defines how specific fields in a data structure should be
+        transformed before being written to a CSV file. It covers
+        transformations for various fields, including those from list of fields
+        with highlights, file paths, URLs, and renamed fields.
+
+        :return: A dictionary where keys are field names and values are lambda
+        functions that define the transformations.
+        """
+        raise NotImplementedError(
+            "Subclass must implement get_csv_transformations method"
+        )
+
+
 def build_numeric_range_query(
     field: str,
     lower_bound: int | float,

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",