CollectGoogleImages/collectGooglePictures/collectDoorImages.py

60 lines
1.9 KiB
Python
Raw Normal View History

#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('office door')
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/imagesDoor/door'+str(i)+'.png')
except:
pass
"""
#Bigger Pictures
for i in range(1, 101):
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/imagesDoor/door'+str(i)+'.png')
except:
pass
driver.close()