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

Question

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

Answer 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()

Comments

[退会済みユーザー]

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

[退会済みユーザー]

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