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')