python使用谷歌浏览器打开网页截取全图

from selenium import webdriver
import os
# chromedriver的路径，设置chrome开启的模式，headless就是无界面模式
#一定要使用这个模式，不然截不了全页面，只能截到当前页面
chromedriver = r"C:\Google\Chrome\Application\chromedriver.exe"
os.environ["webdriver.chrome.driver"] = chromedriver
chrome_options = webdriver.ChromeOptions()
chrome_options.add_argument('--headless')
# 启动浏览器
driver = webdriver.Chrome(options=chrome_options)
# 控制浏览器写入并转到链接
driver.get("url地址")
#driver.maximize_window()只能截取到当前页面
# 接下来是全屏的关键，用js获取页面的宽高
width=driver.execute_script("return document.documentElement.scrollWidth")
height=driver.execute_script("return document.documentElement.scrollHeight")
print(width,height)
# 获取浏览器的宽高
driver.set_window_size(width,height)
# 截图并保存
driver.get_screenshot_as_file(r'D:\tupian.png')
#关闭浏览器
driver.quit()