Qiitaにログインしてダークテーマを使ってみませんか？🌙

ログインするとOSの設定にあわせたテーマカラーを使用できます！

ログイン新規登録

この記事は最終更新日から5年以上が経過しています。

@ryo2132(Ryo Kawamata)

pythonでweb画面キャプチャ取得

投稿日 2017年02月05日

webデザインまとめサイトを作ろうと思い、web画面キャプチャをお手軽に作成する方法を模索中。とりあえずpythonとseleniumで実装してみた。

前提

max os sierra
python 3.5

準備

seleniumのインストール

pip install selenium

ChromeWebdriverのインストール

こちらよりファイルをダウンロード。
ダウンロードしたファイルのディレクトリで以下を実行。

$ mv chromedriver

import time
import re
from selenium import webdriver

#URLのドメイン抽出パターン作成
pat = r"https?://(www.)?([\w-]+).[\w.]"

# コマンドラインよりURL取得。「,」区切りで複数入力可能
inp = input("URLを入力\n")

# 「,」でリストへ分割。さらに左右の空白消去
URLS = list(map(str.strip,(inp.split(","))))

# プラウザ起動（Chrome）
driver = webdriver.Chrome()

# リストからURLをひとつづつ処理
for url in URLS :
    # ドメインの一部をファイル名として設定
    site_name = re.search(pat,url)
    file_name = "{0}.png".format(site_name.group(2))
    # URLを開く
    driver.get(url)
    # ウィンドウサイズとズームを設定
    driver.set_window_size(1250, 1036)
    driver.execute_script("document.body.style.zoom='90%'")
    # 読み込み待機時間
    time.sleep(2)
    # imagesフォルダにスクリーンショットを保存
    driver.save_screenshot("./images/" + file_name)

# プラウザを閉じる
driver.quit()