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("1", 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()