Skip to content

Commit 1dd41f6

Browse files
committed
Simplify VuMark retry function usage
1 parent 3de872f commit 1dd41f6

File tree

1 file changed

+7
-18
lines changed

1 file changed

+7
-18
lines changed

admin/create_secrets_files.py

Lines changed: 7 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,12 @@ def _create_and_get_database_details(
5656
)
5757

5858

59+
@retry(
60+
retry=retry_if_exception_type(exception_types=TimeoutException),
61+
stop=stop_after_attempt(max_attempt_number=3),
62+
wait=wait_exponential(multiplier=2, min=5, max=30),
63+
reraise=True,
64+
)
5965
def _create_and_get_vumark_details(
6066
driver: "WebDriver",
6167
vumark_database_name: str,
@@ -100,23 +106,6 @@ def _create_and_get_database_details_with_retries(
100106
)
101107

102108

103-
@retry(
104-
retry=retry_if_exception_type(exception_types=TimeoutException),
105-
stop=stop_after_attempt(max_attempt_number=3),
106-
wait=wait_exponential(multiplier=2, min=5, max=30),
107-
reraise=True,
108-
)
109-
def _create_and_get_vumark_details_with_retries(
110-
driver: "WebDriver",
111-
vumark_database_name: str,
112-
) -> "VuMarkDatabaseDict":
113-
"""Create a VuMark database and return details with retries on timeout."""
114-
return _create_and_get_vumark_details(
115-
driver=driver,
116-
vumark_database_name=vumark_database_name,
117-
)
118-
119-
120109
def _generate_secrets_file_content(
121110
database_details: "DatabaseDict",
122111
vumark_details: "VuMarkDatabaseDict",
@@ -195,7 +184,7 @@ def main() -> None:
195184
continue
196185

197186
try:
198-
vumark_details = _create_and_get_vumark_details_with_retries(
187+
vumark_details = _create_and_get_vumark_details(
199188
driver=driver,
200189
vumark_database_name=vumark_database_name,
201190
)

0 commit comments

Comments
 (0)