from selenium import webdriver


def init_driver_edge():
    """Initialize the Selenium WebDriver with Edge edge_options.
    Returns:
        WebDriver: The initialized Selenium WebDriver instance.
    """
    from selenium.webdriver.edge.service import Service
    from selenium.webdriver.edge.options import Options
    service = Service(r"D:\Projects\BUCT\edgedriver\msedgedriver.exe")
    edge_options = Options()
    # edge_options.add_argument("--headless")  # Run in headless mode
    edge_options.add_argument("--no-sandbox")
    edge_options.add_experimental_option("detach", False)
    edge_options.add_argument("--user-data-dir=C:\\Users\\Plane\\AppData\\Local\\Microsoft\\Edge\\User Data")
    # edge_options.add_argument(r"user-data-dir=")
    edge_options.add_argument("--profile-directory=Profile 1")
    driver = webdriver.Edge(options=edge_options, service=service)
    driver.implicitly_wait(5)  # Implicit wait for elements to load
    return driver

def init_driver_chrome():
    """Initialize the Selenium WebDriver with Chrome options.
    Returns:
        WebDriver: The initialized Selenium WebDriver instance.
    """
    from selenium.webdriver.chrome.options import Options
    from selenium.webdriver.chrome.service import Service

    service = Service(executable_path="/usr/bin/chromedriver")
    options = Options()
    options.add_argument("--headless")  # Run in headless mode
    options.add_argument("--no-sandbox")
    # options.add_experimental_option("detach", True)
    driver = webdriver.Chrome(service=service, options=options)
    driver.implicitly_wait(5)  # Implicit wait for elements to load
    return driver

# from selenium import webdriver
# from selenium.webdriver.edge.service import Service
# from time import sleep
#
# ser = Service("E:\\webdriver\\msedgedriver.exe")
# edge_options = webdriver.EdgeOptions()
# edge_options.use_chromium = True
# edge_options.add_argument("user-data-dir=C:\\Users\\Administrator\\AppData\\Local\\Microsoft\\Edge\\User Data")
# edge_options.add_argument("profile-directory=Profile 1")
#
# driver = webdriver.Edge(service = ser,options = edge_options)
# driver.get('https://www.bing.com')
#
# sleep(20)