feat: add rate limit to all endpoint

Author: corentin-hrflow

hrflow/auth/__init__.py

@@ -1,6 +1,7 @@
 import typing as t
 
 from ..core.validation import validate_key, validate_response
+from ..core.rate_limit import rate_limiter
 
 API_SECRET_REGEX = r"^ask[rw]?_[0-9a-f]{32}$"
 
@@ -9,6 +10,7 @@ class Auth:
     def __init__(self, api):
         self.client = api
 
+    @rate_limiter
     def get(self) -> t.Dict[str, t.Any]:
         """
         Try your API Keys. This endpoint allows you to learn how to add the right

hrflow/board/__init__.py

@@ -6,12 +6,14 @@
     validate_response,
     validate_value,
 )
+from ..core.rate_limit import rate_limiter
 
 
 class Board(object):
     def __init__(self, client):
         self.client = client
 
+    @rate_limiter
     def list(self, name=None, page=1, limit=30, sort_by="date", order_by="desc"):
         """
         Search boards for given filters.
@@ -42,6 +44,7 @@ def list(self, name=None, page=1, limit=30, sort_by="date", order_by="desc"):
         response = self.client.get("boards", query_params)
         return validate_response(response)
 
+    @rate_limiter
     def get(self, key=None):
         """
         Get source given a board key.

hrflow/job/asking.py

@@ -6,12 +6,14 @@
     validate_reference,
     validate_response,
 )
+from ..core.rate_limit import rate_limiter
 
 
 class JobAsking:
     def __init__(self, api):
         self.client = api
 
+    @rate_limiter
     def get(
         self,
         board_key: str,

hrflow/job/embedding.py

@@ -1,6 +1,6 @@
 from ..core.validation import validate_response
 from ..core import format_item_payload
-
+from ..core.rate_limit import rate_limiter
 
 class JobEmbedding:
     """Manage embedding related job calls."""
@@ -9,6 +9,7 @@ def __init__(self, api):
         """Init."""
         self.client = api
 
+    @rate_limiter
     def get(self, board_key, key=None, reference=None):
         """
         Retrieve the parsing information.

hrflow/job/scoring.py

@@ -11,6 +11,7 @@
     validate_response,
     validate_value,
 )
+from ..core.rate_limit import rate_limiter
 
 
 class JobScoring:
@@ -20,6 +21,7 @@ def __init__(self, api):
         """Init."""
         self.client = api
 
+    @rate_limiter
     def list(
         self,
         board_keys=None,

hrflow/job/searching.py

@@ -10,6 +10,7 @@
     validate_response,
     validate_value,
 )
+from ..core.rate_limit import rate_limiter
 
 
 class JobSearching:
@@ -19,6 +20,7 @@ def __init__(self, api):
         """Init."""
         self.client = api
 
+    @rate_limiter
     def list(
         self,
         board_keys=None,

hrflow/job/storing.py

@@ -13,6 +13,7 @@
     validate_value,
 )
 from ..core import format_item_payload
+from ..core.rate_limit import rate_limiter
 
 
 class JobStoring:
@@ -28,6 +29,7 @@ def __init__(self, api):
         """
         self.client = api
 
+    @rate_limiter
     def add_json(self, board_key, job_json):
         """This endpoint allows you to Index a Job object.
         Note: If your Job is an unstructured text, make sure to parse it first before
@@ -120,6 +122,7 @@ def add_json(self, board_key, job_json):
         response = self.client.post("job/indexing", json=job_json)
         return validate_response(response)
 
+    @rate_limiter
     def edit(self, board_key, job_json, key=None):
         """
         Edit a job already stored in the given source.
@@ -148,6 +151,7 @@ def edit(self, board_key, job_json, key=None):
         response = self.client.put("job/indexing", json=job_json)
         return validate_response(response)
 
+    @rate_limiter
     def get(self, board_key, key=None, reference=None):
         """
         Retrieve the parsing information.
@@ -168,6 +172,7 @@ def get(self, board_key, key=None, reference=None):
         response = self.client.get("job/indexing", query_params)
         return validate_response(response)
 
+    @rate_limiter
     def archive(self, board_key, key=None, reference=None):
         """
         This method allows to archive (is_archive=1) or unarchive (is_archive=0) a job
@@ -192,6 +197,7 @@ def archive(self, board_key, key=None, reference=None):
         response = self.client.patch("job/indexing/archive", json=payload)
         return validate_response(response)
 
+    @rate_limiter
     def list(
         self,
         board_keys,

hrflow/profile/asking.py

@@ -6,12 +6,13 @@
     validate_reference,
     validate_response,
 )
-
+from ..core.rate_limit import rate_limiter
 
 class ProfileAsking:
     def __init__(self, api):
         self.client = api
 
+    @rate_limiter
     def get(
         self,
         source_key: str,

hrflow/profile/attachment.py

@@ -1,6 +1,7 @@
 from ..core.validation import validate_response
 from ..core import format_item_payload
 
+from ..core.rate_limit import rate_limiter
 
 class ProfileAttachments:
     """Manage documents related profile calls."""
@@ -9,6 +10,7 @@ def __init__(self, api):
         """Init."""
         self.client = api
 
+    @rate_limiter
     def list(self, source_key, key=None, reference=None, email=None):
         """
         Retrieve the interpretability information.

hrflow/profile/embedding.py

@@ -3,6 +3,7 @@
 from ..core.validation import validate_response
 from ..core import format_item_payload
 
+from ..core.rate_limit import rate_limiter
 
 class ProfileEmbedding:
     """Manage embedding related profile calls."""
@@ -11,6 +12,7 @@ def __init__(self, api):
         """Init."""
         self.client = api
 
+    @rate_limiter
     def get(self, source_key, key=None, reference=None, email=None, fields={}):
         """
         Retrieve the interpretability information.