# import selenium
from selenium import webdriver


# setup driver's
firefoxdriver = webdriver.Firefox(executable_path="Path to Firefox driver")
# link to driver  "https://github.com/mozilla/geckodriver/releases"

chromedriver = webdriver.Chrome(executable_path="Path to Chrome driver")
# link to driver  "https://sites.google.com/a/chromium.org/chromedriver/downloads"

iedriver = webdriver.IE(executable_path="Path to IEDriverServer.exe")
# link to driver  "http://selenium-release.storage.googleapis.com/index.html"

edgedriver = webdriver.Edge(executable_path="Path to MicrosoftWebDriver.exe")
# link to driver  "https://developer.microsoft.com/en-us/microsoft-edge/tools/webdriver/"
# webdriver.Edge(executable_path = 'C:\\Users\\Downloads\\edgedriver_win32\\msedgedriver.exe')

operadriver = webdriver.Opera(executable_path="Path to operadriver")
# link to driver  "https://github.com/operasoftware/operachromiumdriver/releases"

#  SafariDriver now requires manual installation of the extension prior to automation !!!


# Options, browser arguments
from selenium import webdriver
from selenium.webdriver.chrome.options import Options


options = Options()

options.add_argument("--headless")  # open browser in headless mode. Works in both Chrome and Firefox browsers
options.add_argument("--incognito")  # open private chrome browser 
options.add_argument("--start-maximized")  # start browser maximized to screen. Requires only for Chrome browser. Firefox by default starts maximized
options.add_argument("--disable-notifications")  # disable notifications, works Only in Chrome browser


driver = webdriver.Chrome(chrome_options=options, executable_path="Path to driver")

#  OR
options = Options()
options.add_argument("--incognito", "--start-maximized", "--headless")
driver = webdriver.Chrome(chrome_options=options, executable_path="Path to driver")


#  go to some URL
driver.get("https://www...")  # go to page
driver.back()
driver.forward()
driver.refresh()  # update page

# get browser details
driver.title
driver.window_handles
driver.current_window_handles
driver.current_url
driver.page_source

# get some element or elements
driver.find_element_by_  #  return first element matching the given locator argument
driver.find_elements_by_  # return a list with all elements matching the given locator argument

# by id
# <input id="q" type="text" />
element = driver.find_element_by_id("q")

# by name
# <input id="q" name="search" type="text" />
element = driver.find_element_by_name("search")

# by class_name
# <div class="username" style="display: block;">…</div>
element = driver.find_element_by_class_name("username")

# cklick on checkbox with javascript
driver.execute_script("document.getElementById('personalData').checked = true")

click = driver.find_element_by_class_name('checkbox checkbox-off')
click.click()

# scroll UP page
driver.find_element_by_tag_name('body').send_keys(Keys.CONTROL + Keys.HOME)

# scroll DOWN page
browser.execute_script("window.scrollTo(0,document.body.scrollHeight)")

# driver quit 
driver.quit()  
# or driver.close()


# using Try, Finally
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC


try:
    element = WebDriverWait(driver, 5).until(
    EC.presence_of_element_located((By.ID, "id-of-new-element"))
    )
finally:
    driver.quit()


# disable images
chrome_options = webdriver.ChromeOptions()

prefs = {"profile.managed_default_content_settings.images": 2}

chrome_options.add_experimental_option("prefs", prefs)
driver = webdriver.Chrome(chrome_options=chrome_options)


# get image src
driver.find_element_by_id("element_id").get_attribute("src")

# or
image = driver.find_element(By.XPATH, value='//*[@id="app"]//div[1]/img')
print(image.get_attribute("scr"))


wait = WebDriverWait(browser, 20)
actions = ActionChains(browser)

product_img_xpath = '//div[contains(@class,"s-item")]//img'
wait.until(EC.visibility_of_element_located((By.XPATH, product_img_xpath)))
time.sleep(1)

imgs = browser.find_elements_by_xpath(product_img_xpath)
for img in imgs:
    actions.move_to_element(img).perform()
    print(img.get_attribute('src'))