自动化测试——保存验证码图片至本地

背景：进行自动化测试时，为实现对验证码进行图像识别，摒弃不完善的对验证码进行截图的方法，采取通过urllib将验证码图片保存至本地，再对其进行图像识别的方法。保存验证码图片至本地代码如下：

import urllib

# 首先对验证码进行定位
pic = driver.find_element_by_css_selector("span.verificationpic.fr > img")
# 获取验证码的src属性
pic_url = pic.get_attribute('src')
# 保存验证码图片到指定路径
urllib.urlretrieve(pic_url, 'C:\yzm.jpg')