Python Selenium 무한스크롤

웹 크롤링을 하다보면 맨 아래까지 스크롤해야 원하는 정보를 얻는 경우가 많다.

 

# 스크롤 전 높이

before_h = driver.execute_script("return window.scrollY")

# 무한 스크롤

while True:

    # 맨 아래로 스크롤을 내린다.

    driver.find_element(By.CSS_SELECTOR, 'body').send_keys(Keys.END)

    # 스크롤 사이 페이지 로딩 시간

    time.sleep(1)

    # 스크롤 후 높이

    after_h = driver.execute_script("return window.scrollY")

    if after_h == before_h:

        break

    before_h = after_h

 

이 코드는 신이고 나는 무적이다.

까먹기 싫어서 보관해두자.