Pythonでsuumoの検索結果を自動でスクショする

概要

suumoの検索結果で出てきた物件の詳細画面を自動でスクショする機能を作りました。

環境/使用技術

• macOS Mojave
• Python 3.7.5
• chromedriver
• selenium

事前準備

seleniumをインストール

pip install selenium

ChromeDriverをインストール

chromedriverをブラウザのバージョンと合わせる必要あります。

pip install chromedriver-binary==94.0.4606.61

参考

実装

ソース

chrome.py

import chromedriver_binary
from selenium import webdriver

driver = webdriver.Chrome()

# suumoの検索結果URLを格納する
# 渋谷区のオススメ上位30件の検索結果URLを格納
url = 'https://suumo.jp/jj/chintai/ichiran/FR301FC001/?ar=030&bs=040&ta=13&sc=13113&cb=0.0&ct=9999999&et=9999999&cn=9999999&mb=0&mt=9999999&shkr1=03&shkr2=03&shkr3=03&shkr4=03&fw2=&srch_navi=1'
driver.get(url)

elem_url=[]

elems = driver.find_elements_by_class_name('js-cassette_link_href')

for elem in elems:
    elem_url.append(elem.get_attribute("href"))

i = 1
for url in elem_url:
    driver.get(url)

    w = driver.execute_script("return document.body.scrollWidth;")
    h = driver.execute_script("return document.body.scrollHeight;")

    driver.set_window_size(w,h)

    driver.save_screenshot('./img/img'+ str(i) +'.png')
    i += 1
    driver.back()

driver.quit()

いざ実行

python chrome.py

chromeが自動で起動し、imgディレクトリに詳細画面のキャプチャが保存されていきます。
これであなたも物件マニア