from selenium import webdriverimport os# chromedriver的路径，设置chrome开启的模式，headless就是无界面模式#一定要使用这个模式，不然截不了全页面，只能截到当前页面chromedriver = r"C:\Google\Chrome\Application\chromedriver.exe"os.environ["webdriver.chrome.driver"] = chromedriverchrome_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()
