selenium+headless chrome 自动化测试登陆页面并截图

selenium自动化测试工具
headless-chrome 无头模式
截图保存在桌面

#导包
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
import time

# 声明谷歌浏览器对象
driver = webdriver.Chrome()

#设置浏览器无头模式
option = webdriver.ChromeOptions()
option.add_argument('--headless')
option.add_argument('--disable-gpu')
driver = webdriver.Chrome(chrome_options=option)

#打开浏览器
driver.get('网址')
print(driver.title)

#点击’进入系统‘
button = driver.find_element_by_class_name('n-footer')
button.click()
time.sleep(3)

#将登陆页面截图
driver.get_screenshot_as_file("C:\\Users\\wn\\Desktop\\登陆页面.png")

#输入登录信息
log_in = driver.find_elements_by_class_name('mu-text-field-input')
log_in[0].send_keys('账号')
log_in[1].send_keys('密码')

#点击登陆按钮
button = driver.find_element_by_class_name('mu-ripple-wrapper')
button.click()
time.sleep(3)

#将登陆成功后的页面截图
driver.get_screenshot_as_file("C:\\Users\\wn\\Desktop\\登陆成功后页面.png")
print(driver.current_url)

#退出浏览器
driver.quit()