55 lines
1.5 KiB
Python
Executable File
55 lines
1.5 KiB
Python
Executable File
from selenium import webdriver
|
|
|
|
from msedge.selenium_tools import Edge, EdgeOptions
|
|
from webdriver_manager.firefox import GeckoDriverManager
|
|
from webdriver_manager.chrome import ChromeDriverManager
|
|
from webdriver_manager.microsoft import EdgeChromiumDriverManager
|
|
|
|
|
|
class BrowserDriver:
|
|
@staticmethod
|
|
def get(browser=None, headless=True):
|
|
if browser == "chrome":
|
|
return chrome(headless)
|
|
elif browser == "firefox":
|
|
return firefox(headless)
|
|
elif browser == "edge":
|
|
return edge(headless)
|
|
elif browser == "safari":
|
|
return safari()
|
|
else:
|
|
raise ValueError("'{}' is not a supported browser".format(browser))
|
|
|
|
|
|
def chrome(headless=True):
|
|
options = webdriver.ChromeOptions()
|
|
options.headless = headless
|
|
options.add_argument('--ignore-certificate-errors')
|
|
return webdriver.Chrome(
|
|
ChromeDriverManager().install(),
|
|
options=options)
|
|
|
|
|
|
def firefox(headless=True):
|
|
options = webdriver.FirefoxOptions()
|
|
options.accept_insecure_certs = True
|
|
options.headless = headless
|
|
gecko_driver = GeckoDriverManager().install()
|
|
return webdriver.Firefox(
|
|
executable_path=gecko_driver,
|
|
options=options)
|
|
|
|
|
|
def edge(headless=True):
|
|
options = EdgeOptions()
|
|
options.use_chromium = True
|
|
options.headless = headless
|
|
edge_driver = EdgeChromiumDriverManager().install()
|
|
driver = Edge(edge_driver, options=options)
|
|
return driver
|
|
|
|
|
|
def safari():
|
|
# Because safari driver is bundled with safari
|
|
return webdriver.Safari()
|