#!/usr/bin/env python

# coding: utf-8

# In[12]:

from lib2to3.pgen2 import driver

from selenium import webdriver

import urllib.request

from time import sleep

# url 경로로 다운로드

def downImage(url, name):

dir = 'C:\\Users\\USER\\Sad\\'

urllib.request.urlretrieve(url, dir + name + '.jpg')

#search_term = '우는 사진'

#url= f"https://www.google.com/search?q={search_term}&tbm=isch&ved="

url = "https://www.google.com/search?q=crying+scene&tbm=isch&ved=2ahUKEwjz56aA9f3qAhUMfZQKHeM9AEQQ2-cCegQIABAA&oq=crying+scene&gs_lcp=CgNpbWcQAzIECAAQEzIECAAQEzIECAAQEzIECAAQEzIECAAQEzIECAAQEzIECAAQEzIECAAQEzIECAAQEzIGCAAQHhATOgIIADoFCAAQsQM6BAgAEB5QuUhY_mpgu2xoAnAAeACAAawBiAGSCpIBBDEzLjGYAQCgAQGqAQtnd3Mtd2l6LWltZ8ABAQ&sclient=img&ei=i2wnX7P2Noz60QTj-4CgBA&bih=792&biw=1519&rlz=1C1CAFC_enKR901KR901&hl=ko&hl=ko"

browser = webdriver.Chrome(r"C:\Users\USER\Desktop\ksa_ai\chromedriver.exe")

browser.get(url)

pic_name = browser.find_elements_by_class_name("rg_i")

flg = 0

img_url = {}

# url 경로 모음.

#scroller_with_selenium

last_height = browser.execute_script("return document.body.scrollHeight")

while True:

# Scroll down to bottom

browser.execute_script("window.scrollTo(0, document.body.scrollHeight);")

# Wait to load page

sleep(2)

browser.execute_script("window.scrollTo(0, document.body.scrollHeight-50);")

sleep(2)

# Calculate new scroll height and compare with last scroll height

new_height = browser.execute_script("return document.body.scrollHeight")

if new_height == last_height:

break

last_height = new_height

for idx, value in enumerate(browser.find_elements_by_class_name("rg_i.Q4LuWd")):

print(idx, "번째")

# value.screenshot(search_term + "_" + str(idx) + ".png")

if value.get_attribute("data-src") == None:

img_url[idx] = value.get_attribute("src")

else:

img_url[idx] = value.get_attribute("data-src")

# print(value.get_attribute("data-src"))

# screen_data[idx] = value.get_attribute("data-src")

print(img_url[idx])

downImage(img_url[idx], 'sad08' + str(idx))

#image 갯수

if flg == 1000:

break

flg += 1

# In[ ]: