質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

88.63%

PythonのSelenium WebDriver for文で複数の画像が保存できない(最後の画像だけ保存できる)

解決済

回答 1

投稿

  • 評価
  • クリップ 0
  • VIEW 1,070

data-analysis

score 19

for文の中で指定した画像をすべて保存する方法をご教授いただきたいです。

解決したい問題
for文の中のdriver.save_screenshotで画像を保存すると、for文の最後の要素の画像しか保存できない。

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
import time
import re

def init():
    options = Options()
    options.add_argument('--headless')
    driver = webdriver.Chrome(options=options)
    return driver

def go_to_page(url):
    driver.get(url)
    time.sleep(5)

def get_page_prop():
    page_width = driver.execute_script('return document.body.scrollWidth')
    page_height = driver.execute_script('return document.body.scrollHeight')
    page_prop = {'page_width': page_width, 'page_height': page_height}    
    return page_prop

def set_page_prop(page_prop):
    time.sleep(5)
    driver.set_window_size(page_prop['page_width'], page_prop['page_height'])
    time.sleep(5)

def url_to_shapedurl(url):
    shapedurl = re.sub(r'\\|/|:|\*|\?|"|<|>|\|', '_', url)
    return shapedurl

def save_img(save_file):
    driver.save_screenshot(save_file)
    time.sleep(5)

driver = init()

page_list = 'C:/Users/user/Desktop/page_list.txt'
with open(page_list) as f:
    pages = f.readlines()

for url in pages:
    shaped_url = url_to_shapedurl(url)
    go_to_page(url)
    page_prop = get_page_prop()
    set_page_prop(page_prop)
    img_file = 'C:/Users/user/Desktop/' + shaped_url + '.png'
    save_img(img_file)

driver.quit()
https://www.amazon.co.jp/
https://www.apple.com/jp/

コピペさせていただいたページ
https://blog.amedama.jp/entry/2018/07/28/003342

  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 過去に投稿した質問と同じ内容の質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

回答 1

checkベストアンサー

+1

全画面に保存するプログラム

from selenium import webdriver

options = webdriver.ChromeOptions()
options.set_headless()
driver = webdriver.Chrome(chrome_options=options)

urls = ['https://www.amazon.co.jp/', 'https://www.apple.com/jp/']

for i, url in enumerate(urls):

    #print(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("ss{}.png".format(i))

driver.quit()

投稿

編集

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2018/11/30 18:02

    barobaroさん、ご回答ありがとうございます。appleのキャプチャしか保存できませんでした。。

    キャンセル

  • 2018/11/30 18:05

    barobaroさん、ありがとうございます!すべて保存できました。

    キャンセル

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 88.63%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

関連した質問

同じタグがついた質問を見る