将每次执行的屏幕截图保存在selenium的不同文件夹中

from selenium import webdriver
import os

def save_screenshot(driver, folder, filename):
    # 创建文件夹
    if not os.path.exists(folder):
        os.makedirs(folder)
    
    # 拼接保存路径
    save_path = os.path.join(folder, filename)
    
    # 执行屏幕截图操作
    driver.get_screenshot_as_file(save_path)

# 创建浏览器驱动
driver = webdriver.Chrome()

# 打开网页
driver.get("https://www.example.com")

# 保存屏幕截图到不同的文件夹
save_screenshot(driver, "folder1", "screenshot1.png")
save_screenshot(driver, "folder2", "screenshot2.png")

# 关闭浏览器
driver.quit()