yzmsb_test.py

识别诺诺金服页面的验证码，并自动登录到后台。

 #导包
 from selenium import webdriver
 from PIL import Image, ImageDraw
 from time import sleep
 import pytesseract
 #定义浏览器句柄，赋值url为诺诺金服后台地址
 driver = webdriver.Chrome()
 driver.implicitly_wait(10)  #隐式等待10s
 url = 'https://gl.axnjf.com/Contents/global/login/login.jsp'
 #窗口最大化
 driver.maximize_window()
 driver.get(url)
 # 诺诺金服后台页面截取验证码，代码如下：
 driver.save_screenshot('E:/test_nuonuojinfu/step_01.png')
 #定位验证码
 imgelement = driver.find_element_by_xpath('//*[@id="img_validator"]')
 #获取验证码x,y轴坐标
 location = imgelement.location
 #获取验证码的长宽
 size = imgelement.size
 #写成我们需要截取的位置坐标
 rangle = (int(location['x']), int(location['y']), int(location['x']+size['width']), int(location['y']+size['height']))
 i = Image.open('E:/test_nuonuojinfu/step_01.png')
 #二次截图
 frame4 = i.crop(rangle)
 #二次截图后命名
 frame4.save('E:/test_nuonuojinfu/step_02.png')
 print('二次截图成功！')

 #对截取后的验证码二值化，代码如下：
 # 二值数组
 t2val = {}
 def twoValue(image, G):
     for y in range(0, image.size[1]):
         for x in range(0, image.size[0]):
             g = image.getpixel((x, y))
             if g > G:
                 t2val[(x, y)] = 1
             else:
                 t2val[(x, y)] = 0

 def saveImage(filename, size):
     image = Image.new("", size)
     draw = ImageDraw.Draw(image)
     for x in range(0, size[0]):
         for y in range(0, size[1]):
             draw.point((x, y), t2val[(x, y)])
     image.save(filename)

 path = 'E:/test_nuonuojinfu/step_02.png'
 image = Image.open(path)
 image = image.convert('L')
 twoValue(image, 125)
 path1 = 'E:/test_nuonuojinfu/step_03.jpg'
 saveImage(path1, image.size)
 print("返回信息：验证码到手，增强成功，春哥附体，定能识别！")
 #将处理后的验证码用Tesseract识别,代码如下：
 #添加pytesseract参数变量
 pytesseract.pytesseract.tesseract_cmd= "C:\Program Files (x86)\Tesseract-OCR/tesseract.exe"
 tessdata_dir_config = '--tessdata-dir "C:\Program Files (x86)\Tesseract-OCR/tessdata" '
 #打开需要识别的验证码图片
 image1 = Image.open("E:/test_nuonuojinfu/step_03.jpg")
 #识别验证码，赋值变量为code
 print('开始识别二维码，请等待...')
 code = pytesseract.image_to_string(image1)
 #打印识别后的验证码
 print('验证码：' + str(code))
 #拿到验证码后，进行登录操作，代码如下：
 driver.find_element_by_xpath('//*[@id="username"]').send_keys("nbiaabc")  #输入账号
 driver.find_element_by_xpath('//*[@id="password"]').send_keys("fBmhicc")  #输入密码
 #输入拿到的验证码
 driver.find_element_by_xpath('//*[@id="validator"]').send_keys(code)  #输入验证码
 # 点击登录按钮
 driver.find_element_by_xpath('//*[@id="button"]').click()
 sleep(1)
 driver.quit()