1313import vws_web_tools
1414from dotenv import load_dotenv
1515from selenium .common .exceptions import TimeoutException
16- from tenacity import (
17- retry ,
18- retry_if_exception_type ,
19- stop_after_attempt ,
20- wait_exponential ,
21- )
2216
2317if TYPE_CHECKING :
2418 from selenium .webdriver .remote .webdriver import WebDriver
2519 from vws_web_tools import DatabaseDict , VuMarkDatabaseDict
2620
2721
28- RETRY_ON_TIMEOUT = retry (
29- retry = retry_if_exception_type (exception_types = TimeoutException ),
30- stop = stop_after_attempt (max_attempt_number = 3 ),
31- wait = wait_exponential (multiplier = 2 , min = 5 , max = 30 ),
32- reraise = True ,
33- )
34-
35-
3622def _create_and_get_database_details (
3723 driver : "WebDriver" ,
3824 email_address : str ,
@@ -58,7 +44,7 @@ def _create_and_get_database_details(
5844 license_name = license_name ,
5945 )
6046
61- return RETRY_ON_TIMEOUT ( vws_web_tools .get_database_details ) (
47+ return vws_web_tools .get_database_details (
6248 driver = driver ,
6349 database_name = database_name ,
6450 )
@@ -77,7 +63,7 @@ def _create_and_get_vumark_details(
7763 database_name = vumark_database_name ,
7864 )
7965
80- return RETRY_ON_TIMEOUT ( vws_web_tools .get_vumark_database_details ) (
66+ return vws_web_tools .get_vumark_database_details (
8167 driver = driver ,
8268 database_name = vumark_database_name ,
8369 )
0 commit comments