python selenium UI自动化解决验证码的4种方法


Posted in Python onJanuary 05, 2018

本文介绍了python selenium UI自动化解决验证码的4种方法,分享给大家,具体如下:

测试环境

  1. windows7+
  2. firefox50+
  3. geckodriver # firefox浏览器驱动
  4. python3
  5. selenium3

selenium UI自动化解决验证码的4种方法:去掉验证码、设置万能码、验证码识别技术-tesseract、添加cookie登录,本次主要讲解验证码识别技术-tesseract和添加cookie登录。

1. 去掉验证码

去掉验证码,直接通过用户名和密码登陆网站。

2. 设置万能码

设置万能码,就是不管什么情况,输入万能码,都可以成功登录网站。

3. 验证码识别技术-tesseract

准备条件

  1. tesseract,下载地址:https://github.com/parrot-office/tesseract/releases/tag/3.5.1
  2. Python3.x,下载地址:https://www.python.org/downloads/
  3. pillow(Python3图像处理库)

安装好Python,通过pip install pillow安装pillow库。然后将tesseract中的tesseract.exe和testdata文件夹放到测试脚本所在目录下,testdata中默认有eng.traineddata和osd.traineddata,如果要识别汉语,请自行下载对应包。

以下是两个主要文件,TesseractPy3.py是通过python代码去调用tesseract以达到识别验证码的效果。code.py是通过selenium获取验证码图片,进而使用TesseractPy3中的函数得到验证码,实现网站的自动化登陆。

TesseractPy3.py

#coding=utf-8

import os
import subprocess
import traceback
import logging

from PIL import Image # 来源于Pillow库

TESSERACT = 'tesseract' # 调用的本地命令名称
TEMP_IMAGE_NAME = "temp.bmp" # 转换后的临时文件
TEMP_RESULT_NAME = "temp" # 保存识别文字临时文件
CLEANUP_TEMP_FLAG = True # 清理临时文件的标识
INCOMPATIBLE = True # 兼容性标识

def image_to_scratch(image, TEMP_IMAGE_NAME):
  # 将图片处理为兼容格式
  image.save(TEMP_IMAGE_NAME, dpi=(200,200))

def retrieve_text(TEMP_RESULT_NAME):
  # 读取识别内容
  inf = open(TEMP_RESULT_NAME + '.txt','r')
  text = inf.read()
  inf.close()
  return text

def perform_cleanup(TEMP_IMAGE_NAME, TEMP_RESULT_NAME):
  # 清理临时文件
  for name in (TEMP_IMAGE_NAME, TEMP_RESULT_NAME + '.txt', "tesseract.log"):
    try:
      os.remove(name)
    except OSError:
      pass

def call_tesseract(image, result, lang):
  # 调用tesseract.exe,将识读结果写入output_filename中
  args = [TESSERACT, image, result, '-l', lang]
  proc = subprocess.Popen(args)
  retcode = proc.communicate()

def image_to_string(image, lang, cleanup = CLEANUP_TEMP_FLAG, incompatible = INCOMPATIBLE):
  # 假如图片是不兼容的格式并且incompatible = True,先转换图片为兼容格式(本程序将图片转换为.bmp格式),然后获取识读结果;如果cleanup=True,操作之后删除临时文件。
  logging.basicConfig(filename='tesseract.log')
  try:
    try:
      call_tesseract(image, TEMP_RESULT_NAME, lang)
      text = retrieve_text(TEMP_RESULT_NAME)
    except Exception:
      if incompatible:
        image = Image.open(image)
        image_to_scratch(image, TEMP_IMAGE_NAME)
        call_tesseract(TEMP_IMAGE_NAME, TEMP_RESULT_NAME, lang)
        text = retrieve_text(TEMP_RESULT_NAME)
      else:
        raise
    return text
  except: 
    s=traceback.format_exc()
    logging.error(s)
  finally:
    if cleanup:
      perform_cleanup(TEMP_IMAGE_NAME, TEMP_RESULT_NAME)

code.py

#coding=utf-8

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
from selenium.common.exceptions import NoAlertPresentException
from PIL import Image
import unittest, time, re
from TesseractPy3 import *

class lgoin(unittest.TestCase):
  def setUp(self):
    self.driver = webdriver.Ie()
    self.driver.implicitly_wait(30)
    self.base_url = 'http://127.0.0.1:8080/test' # 要测试的链接
    self.title = '某管理平台' # 测试网站的Title
    self.verificationErrors = []
    self.accept_next_alert = True

  def test_lgoin(self):
    driver = self.driver
    driver.get(self.base_url)
    driver.maximize_window()
    driver.save_screenshot('All.png') # 截取当前网页,该网页有我们需要的验证码
    imgelement = driver.find_element_by_class_name('kaptchaImage')
    location = imgelement.location # 获取验证码x,y轴坐标
    size = imgelement.size # 获取验证码的长宽
    rangle = (int(location['x']),int(location['y']),int(location['x']+size['width']),int(location['y']+size['height'])) # 写成我们需要截取的位置坐标
    i = Image.open("All.png") # 打开截图
    result = i.crop(rangle) # 使用Image的crop函数,从截图中再次截取我们需要的区域
    result.save('result.jpg')
    text = image_to_string('result.jpg', 'eng').strip()

    assert self.title in driver.title

    driver.find_element_by_id(u'userCode').clear()
    driver.find_element_by_id(u'userCode').send_keys('XXXXXX') # 用户名
    driver.find_element_by_id(u'password').clear()
    driver.find_element_by_id(u'password').send_keys('XXXXXX') # 密码
    #driver.find_element_by_name('verifyCode').clear()
    driver.find_element_by_name('verifyCode').send_keys(text)
    driver.find_element_by_name('submit').submit()


  def is_element_present(self, how, what):
    try: self.driver.find_element(by=how, value=what)
    except NoSuchElementException as e: return False
    return True

  def is_alert_present(self):
    try: self.driver.switch_to_alert()
    except NoAlertPresentException as e: return False
    return True

  def close_alert_and_get_its_text(self):
    try:
      alert = self.driver.switch_to_alert()
      alert_text = alert.text
      if self.accept_next_alert:
         alert.accept()
      else:
        alert.dismiss()
      return alert_text
    finally: self.accept_next_alert = True

  def tearDown(self):
    #self.driver.quit()
    self.assertEqual([], self.verificationErrors)

if __name__ == "__main__":
  unittest.main()

最后,执行命令python code.py,就可以成功自动登录网站。

注意:

由于受验证码图片质量以及清晰度的影响,并不是每一次都能成功登陆。

4. 添加cookie登录

首先获取网站登陆后的cookie,然后通过添加cookie的方式,实现网站登陆的目的。我们用cook来表示xxxxxx的登录后的cookie。

# coding=utf-8

from selenium import webdriver
import time 

driver = webdriver.Firefox()
driver.get("http://www.xxxxxx.com/") # 要登陆的网站

driver.add_cookie(cook) # 这里添加cookie,有时cookie可能会有多条,需要添加多次
time.sleep(3) 

# 刷新下页面就可以看到登陆成功了
driver.refresh()

注意:

登录时有勾选下次自动登录的请勾选,浏览器提示是否保存用户密码时请选择确定,这样获取的cookie成功登陆的机率比较高

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python列表append和+的区别浅析
Feb 02 Python
Python实现生成简单的Makefile文件代码示例
Mar 10 Python
python模块简介之有序字典(OrderedDict)
Dec 01 Python
python调用摄像头显示图像的实例
Aug 03 Python
Python之Numpy的超实用基础详细教程
Oct 23 Python
Python多线程及其基本使用方法实例分析
Oct 29 Python
python将邻接矩阵输出成图的实现
Nov 21 Python
pytorch实现onehot编码转为普通label标签
Jan 02 Python
如何在Python 游戏中模拟引力
Mar 27 Python
完美解决pyinstaller打包报错找不到依赖pypiwin32或pywin32-ctypes的错误
Apr 01 Python
Python 字节流,字符串,十六进制相互转换实例(binascii,bytes)
May 11 Python
如何通过安装HomeBrew来安装Python3
Dec 23 Python
轻松实现TensorFlow微信跳一跳的AI
Jan 05 #Python
OpenCV-Python实现轮廓检测实例分析
Jan 05 #Python
django2 快速安装指南分享
Jan 05 #Python
Python实现改变与矩形橡胶的线条的颜色代码示例
Jan 05 #Python
用python制作游戏外挂
Jan 04 #Python
Python学习之Anaconda的使用与配置方法
Jan 04 #Python
Windows下Anaconda的安装和简单使用方法
Jan 04 #Python
You might like
PHP脚本的10个技巧(7)
2006/10/09 PHP
从Web查询数据库之PHP与MySQL篇
2009/09/25 PHP
js 绑定带参数的事件以及手动触发事件
2010/04/27 Javascript
document.forms[].submit()使用介绍
2014/02/19 Javascript
一个JavaScript用逗号分割字符串实例
2014/09/22 Javascript
jQuery中trigger()方法用法实例
2015/01/19 Javascript
jquery基本选择器匹配多个元素的实现方法
2016/09/05 Javascript
AngularJS实现的回到顶部指令功能实例
2017/05/17 Javascript
jquery单击文字或图片内容放大并居中显示
2017/06/23 jQuery
利用 spin.js 生成等待效果(js 等待效果)
2017/06/25 Javascript
详解nodejs实现本地上传图片并预览功能(express4.0+)
2017/06/28 NodeJs
基于vue 开发中出现警告问题去除方法
2018/01/25 Javascript
解决vue项目中type=”file“ change事件只执行一次的问题
2018/05/16 Javascript
如何用JavaScript实现功能齐全的单链表详解
2019/02/11 Javascript
Vue如何实现验证码输入交互
2020/12/07 Vue.js
[02:07]2017国际邀请赛中国区预选赛直邀战队前瞻
2017/06/23 DOTA
python中__slots__用法实例
2015/06/04 Python
彻底搞懂Python字符编码
2018/01/23 Python
对python for 文件指定行读写操作详解
2018/12/29 Python
浅谈PySpark SQL 相关知识介绍
2019/06/14 Python
Python Multiprocessing多进程 使用tqdm显示进度条的实现
2019/08/13 Python
利用Python制作动态排名图的实现代码
2020/04/09 Python
Python接口测试环境搭建过程详解
2020/06/29 Python
python快速安装OpenCV的步骤记录
2021/02/22 Python
css3实例教程 一款纯css3实现的发光屏幕旋转特效
2014/12/07 HTML / CSS
飞利浦比利时官方网站:Philips比利时
2016/08/24 全球购物
Set里的元素是不能重复的,那么用什么方法来区分重复与否呢? 是用==还是equals()? 它们有何区别?用contains来区分是否有重复的对象。还是都不用
2013/07/30 面试题
考博自荐信
2013/10/25 职场文书
翻译专业应届生求职信
2013/11/23 职场文书
母亲节感恩活动记录
2014/03/16 职场文书
计算机毕业大学生求职信
2014/06/26 职场文书
党校学习个人总结
2015/02/15 职场文书
小学班主任工作总结2015
2015/04/07 职场文书
合理化建议书范文
2015/09/14 职场文书
2016年优秀班主任先进事迹材料
2016/02/26 职场文书
详解CSS玩转图片Base64编码
2021/05/25 HTML / CSS