You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
2 years ago
|
import os
|
||
|
|
from typing import Optional
|
||
|
|
|
||
|
|
from webdriver_manager.core import utils
|
||
|
|
from webdriver_manager.core.download_manager import DownloadManager
|
||
|
|
from webdriver_manager.drivers.edge import EdgeChromiumDriver
|
||
|
|
from webdriver_manager.drivers.ie import IEDriver
|
||
|
|
from webdriver_manager.core.manager import DriverManager
|
||
|
|
|
||
|
|
|
||
|
|
class IEDriverManager(DriverManager):
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
version: Optional[str] = None,
|
||
|
|
os_type: Optional[str] = None,
|
||
|
|
path: Optional[str] = None,
|
||
|
|
name: str = "IEDriverServer",
|
||
|
|
url: str = "https://github.com/seleniumhq/selenium/releases/download",
|
||
|
|
latest_release_url: str = "https://api.github.com/repos/seleniumhq/selenium/releases",
|
||
|
|
ie_release_tag: str = "https://api.github.com/repos/seleniumhq/selenium/releases/tags/selenium-{0}",
|
||
|
|
cache_valid_range: int = 1,
|
||
|
|
download_manager: Optional[DownloadManager] = None,
|
||
|
|
):
|
||
|
|
super().__init__(path, cache_valid_range, download_manager=download_manager)
|
||
|
|
self.driver = IEDriver(
|
||
|
|
version=version,
|
||
|
|
os_type=os_type,
|
||
|
|
name=name,
|
||
|
|
url=url,
|
||
|
|
latest_release_url=latest_release_url,
|
||
|
|
ie_release_tag=ie_release_tag,
|
||
|
|
http_client=self.http_client,
|
||
|
|
)
|
||
|
|
|
||
|
|
def install(self) -> str:
|
||
|
|
return self._get_driver_path(self.driver)
|
||
|
|
|
||
|
|
|
||
|
|
class EdgeChromiumDriverManager(DriverManager):
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
version: Optional[str] = None,
|
||
|
|
os_type: str = utils.os_type(),
|
||
|
|
path: Optional[str] = None,
|
||
|
|
name: str = "edgedriver",
|
||
|
|
url: str = "https://msedgedriver.azureedge.net",
|
||
|
|
latest_release_url: str = "https://msedgedriver.azureedge.net/LATEST_RELEASE",
|
||
|
|
cache_valid_range: int = 1,
|
||
|
|
download_manager: Optional[DownloadManager] = None,
|
||
|
|
):
|
||
|
|
super().__init__(path, cache_valid_range, download_manager=download_manager)
|
||
|
|
self.driver = EdgeChromiumDriver(
|
||
|
|
version=version,
|
||
|
|
os_type=os_type,
|
||
|
|
name=name,
|
||
|
|
url=url,
|
||
|
|
latest_release_url=latest_release_url,
|
||
|
|
http_client=self.http_client,
|
||
|
|
)
|
||
|
|
|
||
|
|
def install(self) -> str:
|
||
|
|
driver_path = self._get_driver_path(self.driver)
|
||
|
|
os.chmod(driver_path, 0o755)
|
||
|
|
return driver_path
|