3．画像一括DL（Selenium）

import os
import requests
import time
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium import webdriver


# definition
path = 'C:/Users/9mlor/OneDrive/デスクトップ/'
###foldername = input('FOLDERNAME: ')
url = input('URL: ')
filename = input('FILENAME: ')






# make directory
os.makedirs(path + filename)

# open page with selenium
driver = webdriver.Chrome()
driver.get(url)
time.sleep(4)
imgs = driver.find_elements(By.TAG_NAME,'img')

# make image list
imglist = []
for img in imgs:
    img = img.get_attribute('src')
    if img.startswith('//'):
        img = 'https:' + img
    if img.endswith('jpg') or img.endswith('png') or img.endswith('webp'):
        imglist.append(img)

# download image
i = 0
for image in imglist:
    root, ext = os.path.splitext(image)
    i += 1
    with open(path + filename + '/' + filename + '-' + str(i) + ext, 'wb') as f:
        f.write(requests.get(image).content)