CollectGoogleImages/collectGooglePictures/collectPictureWindowImages.py

#Imports Packages
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import time 
#Opens up web driver and goes to Google Images
#driver = webdriver.Chrome('C:/bin/chromedriver_win32/chromedriver.exe')
driver = webdriver.Chrome()
driver.get('https://www.google.ca/imghp?hl=en&tab=ri&authuser=0&ogbl')

#maximize browser
driver.maximize_window()
driver.find_element_by_xpath('//*[@id="L2AGLb"]').click()
time.sleep(2)
box = driver.find_element_by_xpath('//*[@id="sbtc"]/div/div[2]/input')

box.send_keys('window')
box.send_keys(Keys.ENTER)


#Will keep scrolling down the webpage until it cannot scroll no more
last_height = driver.execute_script('return document.body.scrollHeight')
while True:
    driver.execute_script('window.scrollTo(0,document.body.scrollHeight)')
    time.sleep(2)
    new_height = driver.execute_script('return document.body.scrollHeight')
    try:
        driver.find_element_by_xpath('//*[@id="islmp"]/div/div/div/div/div[5]/input').click()
        time.sleep(2)
    except:
        pass
    if new_height == last_height:
        break
    last_height = new_height

time.sleep(2)
"""
#Small Pictures
for i in range(1, 102):
    try:
        driver.find_element_by_xpath('//*[@id="islrg"]/div[1]/div['+str(i)+']/a[1]/div[1]/img').screenshot('C:/bin/chromedriver_win32/imagesPictureFrame/pictureFrame'+str(i)+'.png')
    except:
        pass
"""

#Bigger Pictures

for i in range(1, 31):
    try:
        driver.find_element_by_xpath('//*[@id="islrg"]/div[1]/div['+str(i)+']/a[1]/div[1]/img').click()
        time.sleep(2)
        picture = driver.find_element_by_xpath('//*[@id="Sva75c"]/div/div/div[3]/div[2]/c-wiz/div/div[1]/div[1]/div[2]/div[1]/a/img')
        picture.screenshot('C:/bin/chromedriver_win32/imagesWindow/pictureWindow'+str(i)+'.png')
    
    except:
        pass
    
driver.close()