【python】seleniumで自動保存。テキストファイルでリスト化で複数キャプチャ【初級】

#キャプチャ自動保存　ファイル名をURLにする
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from webdriver_manager.chrome import ChromeDriverManager #WEBドライバ自動更新
#import chromedriver_binary　＃WEBドライバ自動更新いれたのでいらない
import time

f = open('URL-list.txt', 'r', encoding='UTF-8')
datalist = f.readlines()
f.close()
i = 0
for data in datalist:
#変数にした指定のURLはここに記載
    u = (datalist[i]).strip() #リストの改行削除のため.stripを入れる
#ファイル保存名のための準備
#URLの"."を抜く"/"は"-"にする
    new_u1 = u.replace('.', '') #.を置換
    new_u = new_u1.replace('/', '-') #/を-に置換
# 「https:// 」の後ろにあるだけを抽出したい
    target = 'https://' 
    idx = new_u.find(target)
    r = new_u[len(target): ] # http://以下 ～ 最期まで
    #print('url' + r)

    #webブラウザを取得
    driver = webdriver.Chrome(ChromeDriverManager().install())
    #スクリーンショット
    #変数uから取得したURLをドライバがゲットする
    driver.get(u)
    # Set image(window) size 
    driver.set_window_size(1366, 768) 
    # スクリーンショット保存
    driver.save_screenshot(r + '.png')
    time.sleep(1)
    # ドライバを閉じる
    driver.quit()
    i = i + 1

u = (datalist[i]).strip() 　

from webdriver_manager.chrome import ChromeDriverManager #WEBドライバ自動更新

    #webブラウザを取得
    driver = webdriver.Chrome(ChromeDriverManager().install())