Add max retry parameter to the execution (#802)

Co-authored-by: Meroujan.Antonyan <meroujan.antonyan.external@axa.com>
Co-authored-by: Ian Hellen <ianhelle@microsoft.com>
This commit is contained in:
vx3r 2024-10-21 19:28:48 +02:00 коммит произвёл GitHub
Родитель 11ca944de7
Коммит deab7a5e2c
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
1 изменённых файлов: 4 добавлений и 1 удалений

Просмотреть файл

@ -401,6 +401,7 @@ class CybereasonDriver(DriverBase):
page: int = 0,
page_size: int = 2000,
pagination_token: str = None,
max_retry: int = 3,
) -> Dict[str, Any]:
"""
Run query with pagination enabled.
@ -436,7 +437,8 @@ class CybereasonDriver(DriverBase):
headers = {}
params = {"page": page, "itemsPerPage": page_size}
status = None
while status != "SUCCESS":
cur_try = 0
while status != "SUCCESS" and cur_try < max_retry:
response = self.client.post(
self.search_endpoint,
json={**body, **pagination},
@ -446,6 +448,7 @@ class CybereasonDriver(DriverBase):
response.raise_for_status()
json_result = response.json()
status = json_result["status"]
cur_try += 1
return json_result
async def __run_threaded_queries(