python写页面自动截图

from selenium import webdriver
def take_screenshot(url, file_path):
    driver = webdriver.Chrome()
    driver.get(url)
    driver.save_screenshot(file_path)
    driver.quit()
if __name__ == '__main__':
    take_screenshot('http://baidu.com', 'D:\桌面\wang.png')

要安装selenium还要安装google的驱动