一、将整个注册流程脚本进行模块化-19

标签：name get 19 模块化流程 driver element user self

1、D:\imooc\selenium\register_function.py

# 功能：在register_code.py的基础上，进行二次改造。

# 编码格式
#coding=utf-8

# 添加当前项目路径
import sys
sys.path.append('D:/imooc/selenium')

# 浏览器驱动包
from selenium import webdriver

# 时间包
import time

# 引入随机数生成包
import random

# 使用pip库进行图片解析包/取图片的包
from PIL import Image

# 获取api解析图片包
from ShowapiRequest import ShowapiRequest

# 导入FindElement
from find_element import FindElement

class RegisterFunction(object):

    # 将get_driver()中的driver，放入__init__构造函数中。
    def __init__(self,url):
        # get_driver的时候，需要将url放进去。
        self.driver = self.get_driver(url)

    # 获取driver并打开url
    def get_driver(self,url):
        # Chrome浏览器的驱动
        # driver的实例化
        # driver = webdriver.Edge()
        driver = webdriver.Chrome()
        # driver = webdriver.Firefox()
        # 加载指定url界面
        driver.get(url)
        # 最大化窗口
        driver.maximize_window()
        # 返回driver
        return driver
    
    # 输入用户信息
    def send_user_info(self,key,data):
        self.get_user_element(key).send_keys(data)

    # 定位元素位置，获取element元素值
    def get_user_element(self,key):
        # 调用findelement.py
        find_element = FindElement(self.driver)
        user_element = find_element.get_element(key)
        return user_element

    # 获取随机数
    def get_range_user(self):
        user_info = ''.join(random.sample('1234567890abcdefghijklmn',8))
        return user_info

    # 获取图片
    def get_code_image(self,file_name):
        # 保存整个页面的图片
        # 增加self
        self.driver.save_screenshot(file_name)
        # 通过获取验证码的id，计算code_element的元素，通过get_user_element
        code_element = self.get_user_element("code_image")
        # 通过code_element的元素，计算code_element元素的坐标
        # 获取的格式是：{"x":123,"y":345}
        # print(code_element.location)
        # 左边点的坐标
        left = code_element.location['x']*1.75
        top = code_element.location['y']*1.75
        # 右边点的坐标
        right = code_element.size['width']*1.75+left
        height = code_element.size['height']*1.75+top
        # 打开一张图片，并命名为一个对象
        im = Image.open(file_name)
        # 按照一个坐标进行裁剪
        img = im.crop((left,top,right,height))
        # 保存裁剪后的图片
        img.save(file_name)

    # 解析图片，获取验证码
    def code_online(self,file_name):
        self.get_code_image(file_name)
        # 这里的api不变
        # 需要缴费
        # 修改my_appId = 1182042
        # 修改my_appSecret = 523bfa72c8f64177b7a17f7a2f02c462
        # https://www.showapi.com/apiGateway/view/184?tab=1
        r = ShowapiRequest("http://route.showapi.com/184-1","1182042","523bfa72c8f64177b7a17f7a2f02c462" )
        # 图片存放地址
        r.addFilePara("image", file_name)
        # 代表识别几位数的图片，即多少位数的字。
        r.addBodyPara("typeId", "35")
        # r.addBodyPara("convert_to_jpg", "0")
        res = r.post()
        text = res.json()  #['showapi_res_error']['Result']
        # 返回验证码
        return text

    # 设置value值
    def main(self):
        user_name = self.get_range_user()
        user_email = user_name + "@163.com"
        file_name = "D:/imooc/test01.png"
        code_text = self.code_online(file_name)
        self.send_user_info('user_email',user_email)
        self.send_user_info('user_name',user_name)
        self.send_user_info('user_password',"111111")
        self.send_user_info('code_text',"11111")
        self.get_user_element('register_button').click()
        time.sleep(5)
        self.driver.close()

if __name__ == '__main__':
    register_funtion = RegisterFunction('http://www.5itest.cn/register')
    register_funtion.main()

2、D:\imooc\selenium\config\LocalElement.ini

# 功能：配置文件夹config中的配置文件LocalElement.ini

# 节点名称
[RegisterElement]

# Element id_name and id_value
user_email=id>register_email
user_name=id>register_nickname
user_password=id>register_password
code_image=id>getcode_num
code_text=id>captcha_code
register_button=id>register-btn

3、D:\imooc\selenium\util\read_ini.py

# 功能：进行逻辑处理，读取配置文件

# 编码
# coding=utf-8

# 引入读取配置文件的第三方扩展包
import configparser

# class类 
# 将代码按照一定格式封装起来
# class类_name:ReadIni,继承于object
class ReadIni(object):
        
    # 构造函数，获取load_ini函数中的cf对象。
    # 只要运行程序，就要有cf对象。
    def __init__(self,file_name=None,node=None):
        # file_name为非必须传递
        if file_name == None:
            # 设置默认配置文件
            file_name = "D:/imooc/selenium/config/LocalElement.ini"
        # node为非必须传递
        if node == None:
            # 设置默认配置文件的结点名
            self.node = "RegisterElement"
        else:
            self.node = node
        # 获取load_ini函数中的cf对象
        self.cf = self.load_ini(file_name)
    
    # 加载配置文件的函数
    def load_ini(self,file_name):
        # 调用configparser下的ConfigParser，并将调用的内容，设置为cf对象
        cf = configparser.ConfigParser()
        # 读取配置文件file_name，形参传递进来
        cf.read(file_name,encoding="utf-8-sig")
        # 将cf对象返回出去
        return cf
        
    # 获取配置文件中结点值的函数
    def get_value(self,key):
        data = self.cf.get(self.node,key)
        return data

# 主函数运行     
#if __name__ == '__main__':
    # ReadIni()对象实例化为read_init
#    read_init = ReadIni()
    # 输出结果，将user_name传递到get_value中的key
#   print(read_init.get_value("user_email"))

4、D:\imooc\selenium\find_element.py

# 功能：进行逻辑处理，获取配置文件中的元素值

# 编码格式
#coding=utf-8

# 导入util包的读取配置文件代码中的类
from util.read_ini import ReadIni

# 封装class
class FindElement(object):

    # 以构造函数方式传入driver(因为要用，所以必须传)
    def __init__(self,driver):
        self.driver = driver

    # 获取元素值的函数
    def get_element(self,key):

        # 实例化对象
        read_ini = ReadIni()

        # 获取实例化对象指向的值
        data = read_ini.get_value(key)

        # 拆分上述的值，取真正有用的值
        by = data.split('>')[0]
        value = data.split('>')[1]

        # 加容错处理
        try:
            if by == 'id':
                return self.driver.find_element_by_id(value)
            elif by == 'name':
                return self.driver.find_element_by_name(value)
            elif by == 'className':
                return self.driver.find_element_by_class_name(value)
            else:
                return self.driver.find_element_by_xpath(value)
        except:
            return None

5、D:\imooc\selenium\ShowapiRequest.py

import requests
from urllib import parse
#全局请求头
files = {}
headers = {}
body = {}
timeouts = {}
resHeader = {}

class ShowapiRequest:
    def __init__(self, url, my_appId, my_appSecret):
        self.url = url
        self.my_appId = my_appId
        self.my_appSecret = my_appSecret
        body["showapi_appid"] = my_appId
        body["showapi_sign"] = my_appSecret
        headers["User-Agent"] = "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/45.0.2427.7 Safari/537.36"

    def addFilePara(self, key, value_url):
        files[key] = open(r"%s" % (value_url), 'rb')
        return self

    def addHeadPara(self, key, value):
        headers[key] = value
        return self

    def addBodyPara(self, key, value):
        body[key] = value
        return self
    #设置连接时间和读取时间
    def setTimeout(self, connecttimout, readtimeout):
        timeouts["connecttimout"] = connecttimout
        timeouts["readtimeout"] = readtimeout
        return self


    def get(self):
        get_url = self.url + "?" + parse.urlencode(body)
        if not timeouts:
            res = requests.get(get_url, headers=headers)
        else:
            timeout = (timeouts["connecttimout"], timeouts["readtimeout"])
            res = requests.get(get_url, headers=headers, timeout=timeouts)
        return res

    def post(self):
        if not timeouts:
            res = requests.post(self.url, files=files, data=body, headers=headers)
        else:
            timeout = (timeouts["connecttimout"], timeouts["readtimeout"])
            res = requests.post(self.url, files=files, data=body, headers=headers, timeout=timeout)
        return res

标签：name,get,19,模块化,流程,driver,element,user,self
From： https://www.cnblogs.com/jieqiong1755/p/16791468.html

一、将整个注册流程脚本进行模块化-19

相关文章

赞助商

阅读排行