SERVER-111643: add retries to evg_api.tasks_by_build in evergreen_activate_gen_tasks.py (#41950)

GitOrigin-RevId: ce54f888151cb4fc84bc5f9bb7437a43c30c3be1
This commit is contained in:
Tianyu Wang 2025-09-30 10:56:22 -04:00 committed by MongoDB Bot
parent 3988b14c34
commit d73ff660e5
1 changed files with 8 additions and 1 deletions

View File

@ -5,8 +5,10 @@ import os
import sys import sys
import click import click
import requests
import structlog import structlog
from pydantic.main import BaseModel from pydantic.main import BaseModel
from retry.api import retry_call
from urllib3.util import Retry from urllib3.util import Retry
from evergreen.api import ( from evergreen.api import (
@ -93,7 +95,12 @@ def activate_task(expansions: EvgExpansions, evg_api: EvergreenApi) -> None:
tasks_not_activated.append(task.task_id) tasks_not_activated.append(task.task_id)
else: else:
task_list = evg_api.tasks_by_build(expansions.build_id) task_list = retry_call(
evg_api.tasks_by_build,
fargs=[expansions.build_id],
tries=3,
exceptions=requests.exceptions.ChunkedEncodingError,
)
for task in task_list: for task in task_list:
if task.display_name == expansions.task: if task.display_name == expansions.task:
LOGGER.info("Activating task", task_id=task.task_id, task_name=task.display_name) LOGGER.info("Activating task", task_id=task.task_id, task_name=task.display_name)