Python+Selenium_UI自动化操作(5)——浏览器截屏

UI自动化——浏览器截屏

练习：
用chrome浏览器打开不同的网页，并截屏。
语法：
get_screenshot_as_file(“图片名.格式”)
数据准备：
将网页的域名www.baidu.com/www.sohu.com，写入文件，存放在f:\weburl.txt

#encoding=utf-8
from selenium import webdriver
import time ,re

driver=webdriver.Chrome()
with open("f:\\weburl.txt") as fp : 
    for url_date in fp:
        url=url_date.strip()
        # 使用正则生成图片的名称，即取网页的名称，如baidu、sohu
        pic_name=re.search(r"www.(.*).com",url).group(1)
        driver.get(url)
        time.sleep(3)
        # 将调用get_screenshot_as_file方法，将截屏的图片，存放在对应路径，以png格式保存。
        driver.get_screenshot_as_file(r"f:\\测试截屏\\"+pic_name+".png")

driver.quit()

执行后，图片存放在对应路径