selenium截长屏

Options = webdriver.ChromeOptions()
#fireFoxOptions.set_headless()
# browser = webdriver.Firefox()
Options.add_argument('-headless')
driver = webdriver.Chrome(Options)                # 开始请求
driver.get('https://www.baidu.com')
driver.maximize_window()
    # 加延时 防止未加载完就截图
time.sleep(1)
# 用js获取页面的宽高，如果有其他需要用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)
time.sleep(1)
driver.get_screenshot_as_file('123.png')