browser = webdriver.Chrome()
wait = WebDriverWait(browser, 10)
browser.get('https://www.baidu.com')
time.sleep(3)
browser.get_screenshot_as_file('C:\\Users\\zd\\Desktop\\baidu.png')
# 或者
# driver.save_screenshot('full_snap.png')
# page_snap_obj=Image.open('full_snap.png')
注意file中是两个\\
以下是通过截取全屏,再定位到目标(验证码)图片进行截取目标图片并保存。
from PIL import Image
def get_snap(driver): # 对目标网页进行截屏。这里截的是全屏
driver.save_screenshot('full_snap.png')
page_snap_obj=Image.open('full_snap.png')
return page_snap_obj
def get_image(driver): # 对验证码所在位置进行定位,然后截取验证码图片
img = driver.find_element_by_class_name('code')
time.sleep(2)
location = img.location
print(location)
size = img.size
left = location['x']
top = location['y']
right = left + size['width']
bottom = top + size['height']
page_snap_obj = get_snap(driver)
image_obj = page_snap_obj.crop((left, top, right, bottom))
# image_obj.show()
return image_obj # 得到的就是验证码