全栈测试开发系列----WebDriver API及对象识别技术（二）---selenium验证码处理的多种实现方式

pip install pytesseract
pip install Pillow

from PIL import Image

image = Image.open('image.png')
ima_new = image.convert('L')

ima_new.show()

from PIL import Image

image = Image.open('image.png')
ima_new = image.point(lambda x: 0 if x < 143 else 255)
ima_new.show()
get_result = pytesseract.image_to_string(ima_new)
print(get_result)

from PIL import Image
from PIL import ImageEnhance
from pytesseract import pytesseract

image = Image.open('image.png')
ima_new = image.convert('L')
ima_new = image.point(lambda x: 0 if x < 143 else 255)
# ima_new.show()

enhancer = ImageEnhance.Contrast(ima_new)
image2 = enhancer.enhance(4)
get_result = pytesseract.image_to_string(image2)
print(get_result)

# -----------------------------------
'''
@Author:C_N_Candy
@Date  :2023/6/29 14:57
@File  :cookie_login.py
@Desc  :
'''
# ------------------------------------
from selenium import webdriver


class CookieLogin(object):
    def __init__(self):
        self.driver = webdriver.Chrome()
        self.driver.get('http://192.168.235.128/index.php/Admin/login/index.html')
        # 建立session链接
        self.driver.add_cookie({'name': 'login', 'value': 'cookie'})
        # 基于此session执行操作
        self.driver.refresh()  # 刷新
        self.driver.get('http://192.168.235.128/index.php/Admin/login/index.html')


if __name__ == '__main__':
    login = CookieLogin()

# -----------------------------------
'''
@Author:C_N_Candy
@Date  :2023/6/29 15:06
@File  :get_code_image.py
@Desc  :获取页面中的验证码图片
'''
# ------------------------------------
import time

from selenium import webdriver
from selenium.webdriver.common.by import By
from PIL import Image


class GetCodeImage(object):
    def __init__(self):
        url = 'http://192.168.235.128/index.php/Admin/login/index.html'
        self.driver = webdriver.Chrome()
        self.driver.maximize_window()
        self.driver.get(url)
        self.driver.implicitly_wait(10)

    def get_code_image(self):
        self.driver.get_screenshot_as_file('index.png')
        # 需要获取验证码的位置
        code_element = self.driver.find_element(By.ID, 'change_captcha')

        # 这里输出定位到的x,y坐标，根据画图工具查看的坐标，进行相加或者相减，这个数值不是固定的，每个人的电脑型号不同分辨率不同，坐标不同
        # loct = self.driver.find_element(By.ID, 'change_captcha').location
        # print(loct)
        # 输出{'x': 900, 'y': 372}，画图工具显示的坐标是{'x': 1350, 'y': 558}，补上相差的数据450,186

        # 获取验证码图片在截图中的坐标位置和大小
        left = code_element.location['x']+450
        upper = code_element.location['y']+186
        right = code_element.size['width']+52 + left
        bottom = code_element.size['height']+12 + upper

        # 创建image对象，打开之前的截图
        image = Image.open('index.png')
        # 根据坐标点截取图片，另存为
        code_image = image.crop((left, upper, right, bottom))
        code_image.save('code.png')
        print("Success")
        self.driver.quit()



if __name__ == '__main__':
    get_code_png = GetCodeImage()
    get_code_png.get_code_image()

# -----------------------------------
'''
@Author:C_N_Candy
@Date  :2023/6/29 19:04
@File  :Cloud_Code.py
@Desc  :将图片发送给打码平台，返回验证码
'''
# ------------------------------------

import json
import requests
import base64
from io import BytesIO
from PIL import Image
from sys import version_info

"""
version_info是sys模块中的一个函数，主要用于返回你当前所使用的Python版本号。
version_info是一个包含了版本号5个组成部分的元祖，这5个部分分别是主要版本号（major）、次要版本号（minor）、微型版本号（micro）、发布级别（releaselevel）和序列号（serial）
"""


def base64_api(username, password, image):
    # 图片转换成JPEG格式
    img = image.convert('RGB')
    buffered = BytesIO()
    img.save(buffered, format="JPEG")

    # 判断当前Python版本（版本不一样，使用的编码要求不同）
    if version_info.major >= 3:
        b64 = str(base64.b64encode(buffered.getvalue()), encoding='utf-8')
    else:
        b64 = base64.b64encode(buffered.getvalue())

    data = {"username": username, "password": password, "image": b64}
    result = json.loads(requests.post("http://api.ttshitu.com/predict", json=data).text)
    if result["success"]:
        return result["data"]["result"]
    else:
        return result["message"]

    return ""


if __name__ == '__main__':
    img_path = "code.png"
    img = Image.open(img_path)
    result = base64_api(username='图鉴打码平台的账号', password="密码", image=img)
    print(result)

# -----------------------------------
'''
@Author:C_N_Candy
@Date  :2023/6/29 18:44
@File  :Dsmall_Login.py
@Desc  :登录商城后台，验证码处理
'''
# ------------------------------------

from BaseMoudule.Base_Class import BaseClass
from Check_Code.Cloud_Code import base64_api
from selenium.webdriver.common.by import By
from PIL import Image


class Dsmall_Login(BaseClass):
    def __init__(self, url, browserType):
        super().__init__(url, browserType)

    def get_code_image(self):
        self.driver.get_screenshot_as_file('index.png')
        # 需要获取验证码的位置
        code_element = self.driver.find_element(By.ID, 'change_captcha')

        # 获取验证码图片在截图中的坐标位置和大小
        left = code_element.location['x'] + 450
        upper = code_element.location['y'] + 186
        right = code_element.size['width'] + 52 + left
        bottom = code_element.size['height'] + 12 + upper

        # 创建image对象，打开之前的截图
        image = Image.open('index.png')
        # 根据坐标点截取图片，另存为
        code_image = image.crop((left, upper, right, bottom))
        code_image.save('code.png')
        # print("Success")
        # self.driver.quit()

    # 把验证码发送给第三方服务器识别，获取并输入验证码
    def get_code(self, username, password, image):
        get_code = base64_api(username, password, image)
        print(get_code)
        self.driver.find_element(By.NAME, "captcha").send_keys(get_code)


if __name__ == '__main__':
    ds = Dsmall_Login("http://192.168.235.128/index.php/Admin/login/index.html", "Chrome")
    ds.get_code_image()
    ds.get_code(username='打码平台账号', password='密码', image=Image.open("code.png"))