|
17 | 17 |
|
18 | 18 | from typing import Optional |
19 | 19 |
|
| 20 | +from selenium.webdriver.chromium.options import ChromiumOptions |
20 | 21 | from selenium.webdriver.chromium.remote_connection import ChromiumRemoteConnection |
| 22 | +from selenium.webdriver.chromium.service import ChromiumService |
21 | 23 | from selenium.webdriver.common.driver_finder import DriverFinder |
22 | 24 | from selenium.webdriver.remote.command import Command |
23 | 25 | from selenium.webdriver.remote.webdriver import WebDriver as RemoteWebDriver |
24 | 26 |
|
25 | | -from .options import ChromiumOptions |
26 | | -from .service import ChromiumService |
27 | | - |
28 | 27 |
|
29 | 28 | class ChromiumDriver(RemoteWebDriver): |
30 | 29 | """Controls the WebDriver instance of ChromiumDriver and allows you to |
31 | 30 | drive the browser.""" |
32 | 31 |
|
33 | 32 | def __init__( |
34 | 33 | self, |
35 | | - browser_name: Optional[str] = None, |
36 | | - vendor_prefix: Optional[str] = None, |
| 34 | + browser_name: str, |
| 35 | + vendor_prefix: str, |
37 | 36 | options: Optional[ChromiumOptions] = None, |
38 | 37 | service: Optional[ChromiumService] = None, |
39 | 38 | keep_alive: bool = True, |
@@ -62,8 +61,8 @@ def __init__( |
62 | 61 |
|
63 | 62 | executor = ChromiumRemoteConnection( |
64 | 63 | remote_server_addr=self.service.service_url, |
65 | | - browser_name=str(browser_name), |
66 | | - vendor_prefix=str(vendor_prefix), |
| 64 | + browser_name=browser_name, |
| 65 | + vendor_prefix=vendor_prefix, |
67 | 66 | keep_alive=keep_alive, |
68 | 67 | ignore_proxy=options._ignore_local_proxy, |
69 | 68 | ) |
|
0 commit comments