Skip to content

Commit 872188b

Browse files
author
dmy.berezovskyi
committed
fix ruff
1 parent 203e956 commit 872188b

File tree

3 files changed

+25
-10
lines changed

3 files changed

+25
-10
lines changed

core/driver.py

Lines changed: 15 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@ def _get_driver_path(driver_type=None):
1818
ErrorHandler.raise_error(ErrorType.UNSUPPORTED_DRIVER_TYPE)
1919

2020
project_dir = os.path.dirname(
21-
os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
21+
os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
22+
)
2223
driver_path = os.path.join(project_dir, "resources", driver_type)
2324

2425
if not os.path.exists(driver_path):
@@ -31,7 +32,9 @@ def _configure_driver(driver, environment):
3132
driver.maximize_window()
3233
driver.implicitly_wait(3)
3334
driver.get(Properties.get_base_url(environment))
34-
log.info(f"Configure driver and base url: {Properties.get_base_url(environment)}")
35+
log.info(
36+
f"Configure driver and base url: {Properties.get_base_url(environment)}"
37+
)
3538

3639

3740
class Driver(ABC):
@@ -58,10 +61,12 @@ def create_driver(self, environment=None, dr_type="local"):
5861
options=options
5962
)
6063
log.info(
61-
f"Created local Chrome driver with session: {driver.session_id}")
64+
f"Created local Chrome driver with session: {driver.session_id}"
65+
)
6266
except Exception as e:
6367
log.error(
64-
f"Failed to create Chrome driver, falling back to local driver: {e}")
68+
f"Failed to create Chrome driver {e}"
69+
)
6570
driver = webdriver.Chrome(
6671
service=ChromeService(_get_driver_path(dr_type)),
6772
options=_init_driver_options(dr_type=dr_type),
@@ -78,22 +83,25 @@ def create_driver(self, environment=None, dr_type=None):
7883
desired_capabilities={"LT:Options": caps}, # noqa
7984
)
8085
log.info(
81-
f"Remote Chrome driver created with session: {driver.session_id}")
86+
f"Remote Chrome driver created with session: {driver.session_id}"
87+
)
8288
return driver
8389

8490

8591
class FirefoxDriver(Driver):
8692
def create_driver(self, environment=None, dr_type=None):
8793
try:
8894
driver = webdriver.Firefox(
89-
options=_init_driver_options(dr_type=dr_type))
95+
options=_init_driver_options(dr_type=dr_type)
96+
)
9097
log.info(f"Created Firefox driver with session: {driver.session_id}")
9198
except Exception as e:
9299
driver = webdriver.Chrome(
93100
service=ChromeService(_get_driver_path(dr_type)),
94101
options=_init_driver_options(dr_type=dr_type),
95102
)
96103
log.error(
97-
f"Failed to create Firefox driver, falling back to Chrome: {e}")
104+
f"Failed to create Firefox driver, falling back to Chrome: {e}"
105+
)
98106
_configure_driver(driver, environment)
99107
return driver

core/driver_factory.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
log = Logger(log_lvl=LogLevel.INFO).get_instance()
66

7+
78
class WebDriverFactory:
89
DRIVER_MAPPING = {
910
"chrome": ChromeRemoteDriver,
@@ -17,6 +18,10 @@ def create_driver(environment=None, driver_type="local"):
1718
driver_type = driver_type.lower()
1819
if driver_type in WebDriverFactory.DRIVER_MAPPING:
1920
driver_class = WebDriverFactory.DRIVER_MAPPING[driver_type]
20-
return driver_class().create_driver(environment, driver_type) # No .value needed
21+
return driver_class().create_driver(
22+
environment, driver_type
23+
) # No .value needed
2124
else:
22-
raise ErrorHandler.raise_error(ErrorType.ENV_ERROR, environment, driver_type)
25+
raise ErrorHandler.raise_error(
26+
ErrorType.ENV_ERROR, environment, driver_type
27+
)

core/driver_options.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,9 @@ def _init_driver_options(dr_type=None):
2626
options = driver_option_mapping.get(dr_type)
2727

2828
if options is None:
29-
raise ErrorHandler.raise_error(ErrorType.UNSUPPORTED_DRIVER_TYPE, dr_type)
29+
raise ErrorHandler.raise_error(
30+
ErrorType.UNSUPPORTED_DRIVER_TYPE, dr_type
31+
)
3032

3133
_shared_driver_options(options)
3234
log.info(f"Driver options {options.arguments}")

0 commit comments

Comments
 (0)