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