Python+selenium+PIL实现自动化登录（并通过验证码）

Myth丶恋晨 2022-10-24 01:42 143阅读 0赞

### 文章目录 ###

*  前言
 *  一、前提准备
 *  二、步骤
 *  1.Lib包准备
 *  2.Util包准备
 *  3.实现自动化登录
 *  总结

--------------------

# 前言 #

为了更好的巩固web自动化测试的学习，为此找了个线上网站进行自动化且通过验证码的登录。

# 一、前提准备 #

1.测试网址（https://www.bazhuayu.com/login）  
\#自行注册好账号密码以便登录  
2.第三方识别验证码（http://www.chaojiying.com/）  
\#自行注册好账号，关注微信公众号绑定有1000积分足够测试用  
3.开发工具pycharm

# 二、步骤 #

新建project  
新建包  
Lib（存放接口文档）  
Screenshots（存放截图）  
Testcases（存放测试用例）  
Util（存放自己集成后的相关功能）  
main.py（用于运行）

# 1.Lib包准备 #

http://www.chaojiying.com/api-14.html（超级鹰python demo下载）

#!/usr/bin/env python
    # coding:utf-8
    
    import requests
    from hashlib import md5
    
    class Chaojiying_Client(object):
    
        def __init__(self, username, password, soft_id):
                self.username = username
                password =  password.encode('utf8')
                self.password = md5(password).hexdigest()
                self.soft_id = soft_id
                self.base_params = {
        
                    'user': self.username,
                    'pass2': self.password,
                    'softid': self.soft_id,
                }
                self.headers = {
        
                    'Connection': 'Keep-Alive',
                    'User-Agent': 'Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 5.1; Trident/4.0)',
                }
    
        def PostPic(self, im, codetype):
                """
                im: 图片字节
                codetype: 题目类型 参考 http://www.chaojiying.com/price.html
                """
                params = {
        
                    'codetype': codetype,
                }
                params.update(self.base_params)
                files = {
        'userfile': ('ccc.jpg', im)}
                r = requests.post('http://upload.chaojiying.net/Upload/Processing.php', data=params, files=files, headers=self.headers)
                return r.json()
    
        def ReportError(self, im_id):
                """
                im_id:报错题目的图片ID
                """
                params = {
        
                    'id': im_id,
                }
                params.update(self.base_params)
                r = requests.post('http://upload.chaojiying.net/Upload/ReportError.php', data=params, headers=self.headers)
                return r.json()
    
    
    if __name__ == '__main__':
        chaojiying = Chaojiying_Client('超级鹰用户名', '超级鹰用户名的密码', '96001')  # 用户中心>>软件ID 生成一个替换 96001
        im = open('a.jpg', 'rb').read()  # 本地图片文件路径 来替换 a.jpg 有时WIN系统须要//
        print(chaojiying.PostPic(im, 1902))
    #1902 验证码类型  官方网站>>价格体系 3.4+版 print 后要加()

# 2.Util包准备 #

from PIL import  Image
    from Lib.chaojiying import Chaojiying_Client
    import  random,string,os,time,pickle
    def get_code(drive,id):
        #获取验证码图片
        t=time.time()
        path=os.path.dirname(os.path.dirname(__file__))+'\\Screenshots'
        picture_name1=path+'\\'+str(t)+'.png'
    
        drive.save_screenshot(picture_name1)
        ce=drive.find_element_by_id(id)
    
    
    
        left = ce.location['x'] * 1.25  # 计算左边界
        top = ce.location['y'] * 1.25  # 计算上边界
        right = (ce.location['x'] + ce.size['width']) * 1.25  # 计算右边界 *1.25是根据电脑的缩放与布局
        botom = (ce.location['y'] + ce.size['height']) * 1.25  # 计算下边界
        # 将上下左右边界值放到元祖中（注意顺序：左 上 右  下）
    
        im=Image.open(picture_name1)
        img=im.crop((left, top, right, botom))
    
        t=time.time()
    
        picture_name2=path+'\\'+str(t)+'.png'
        img.save(picture_name2)
    
        # 调用第三方库
        chaojiying = Chaojiying_Client('#', '#', '#')  #自己输入账号，密码，和id（超级鹰个人中心生成）
        im = open(picture_name2, 'rb').read()  # 本地图片文件路径 来替换 a.jpg 有时WIN系统须要//
        res = chaojiying.PostPic(im, 1902)
        data = res['pic_str']
        return data

# 3.实现自动化登录 #

在Testcase包里新建test\_login\_bzy.py

from  selenium import  webdriver
    from  time import  sleep
    from  Util import util
    class Testlogin(object):
        def __init__(self):
            #实例化浏览器
            self.drive=webdriver.Chrome()
            #测试网址
            self.drive.get('https://www.bazhuayu.com/login')
            sleep(2)
        def test_bzy(self):
            #定位用户名/手机号/邮箱输入框
            self.drive.find_element_by_id('LoginName').send_keys('账号')
            #定位密码框
            self.drive.find_element_by_id('Password').send_keys('密码')
            #第三方Api识别
            captcha=util.get_code(self.drive,'captchaImg')
            #定位验证码框
            self.drive.find_element_by_id('Captcha').send_keys(captcha)
            #点击登录
            self.drive.find_element_by_id('login').click()
    
    if __name__ == '__main__':
        Testlogin().test_bzy()

在main.py运行

from Testcases.test_login_bzy import Testlogin
    
    if __name__=="__main__":
    Testlogin().test_bzy()

# 总结 #

1.现在第三方的图片识别比较多，也可以搭建相关依赖进行本地训练，但是为了节约时间，可以选择第三方稳定接口  
2.规范化尤为重要，后期的迭代可以体现  
3.识别验证码其实就是截图后再进行抠图再进行识别返回