Python+selenium 【第九章】封装excel类，封装test_data类 读取测试元素信息

题记：

　　本章节主要讲述的excel的封装，以及测试元素信息的读取，来适配框架的一些操作

使用xlrd3读取数据

由于我们的数据是存储到excel里面的,所以在此我们需要封装一个读取excel的数据的方法

读取baidu_page.xls文件中的内容

• excel图示

示例代码

# -*- coding: utf-8 -*-
# @Time : 2022/1/4 16:15
# @Author : Limusen
# @File : demo_excel_20

import os
import xlrd3

abs_path = os.path.dirname(os.path.abspath(__file__))
# excel的路径
file_path = os.path.join(abs_path, '..', 'files', 'baidu_page.xls')

# 创建excel对象
work_book = xlrd3.open_workbook(file_path)
# 创建表对象
sheet = work_book.sheet_by_name("baidu")

# 获取行
print(sheet.nrows)
# 获取列
print(sheet.ncols)
# 获取数据
print(sheet.cell_value(2, 4))

# 获取总行数
def nrows():
    return sheet.nrows

# 获取总列数
def ncols():
    return sheet.ncols

# 　取出excel的全部数据
all_excel_data = []
for row in range(nrows()):
    row_data = []
    for cols in range(ncols()):
        cell_value = sheet.cell_value(row, cols)
        row_data.append(cell_value)
    all_excel_data.append(row_data)
for i in all_excel_data:
    print(i)

• 读取出来的数据

封装excel类

上面我们讲到了如何将数据读取出来,现在我们将代码封装成一个excel类

　　

　　1.新建common

　　2.common包下新建excel_utils.py文件

示例代码

# -*- coding: utf-8 -*-
# @Time : 2022/1/4 16:49
# @Author : Limusen
# @File : excel_utils

import xlrd3


class ExcelUtils:

    def __init__(self, excel, sheet_name=None):
        self.excel_path = excel
        self.sheet_name = sheet_name
        self.sheet_data = self.__get_all_data()

    def __get_all_data(self):
        """
        创建excel对象
        :return:
        """
        work_book = xlrd3.open_workbook(self.excel_path)
        if self.sheet_name:
            sheet = work_book.sheet_by_name(self.sheet_name)
        else:
            sheet = work_book.sheet_by_index(0)
        return sheet

    @property
    def get_count_rows(self):
        """
        获取excel总行数
        :return:  返回总行数
        """
        row_count = self.sheet_data.nrows
        return row_count

    @property
    def get_count_cols(self):
        """
        获取excel总列数
        :return:  返回总列数
        """
        cols_count = self.sheet_data.ncols
        return cols_count

    def get_sheet_data_by_list(self):
        """
        将数据用列表组成
        :return:
        """
        all_excel_data = []
        for row in range(self.get_count_rows):
            row_data = []
            for col in range(self.get_count_cols):
                cell_value = self.sheet_data.cell_value(row, col)
                row_data.append(cell_value)
            all_excel_data.append(row_data)
        return all_excel_data


if __name__ == '__main__':
    file_path = r"D:\PythonSelenium\sample\files\baidu_page.xls"
    excel = ExcelUtils(file_path)
    print(excel.get_sheet_data_by_list())

代码图示

封装element_excel_data类

上面说到了读取excel的数据,但是那个数据并不是我想要的,我现在想要的是一个字典中包含多个字典的键值对形式数据,只需要输入对应的键名就可以值

 

• 举例:

{

‘login_button’: {‘element_name’: ‘登录按钮’, ‘locator_type’: ‘xpath’, ‘locator_value’: ‘//a[text()=“登录”]’, ‘timeout’: 5.0},

‘user_name_input’: {‘element_name’: ‘用户名输入框’, ‘locator_type’: ‘xpath’, ‘locator_value’: ‘//input[@alt=“填写用户名或邮箱”]’, ‘timeout’: 5.0},

‘password_input’: {‘element_name’: ‘密码输入框’, ‘locator_type’: ‘xpath’, ‘locator_value’: ‘//input[@alt=“填写密码”]’, ‘timeout’: 5.0},

‘switch_login_button’: {‘element_name’: ‘跳转之后,登录按钮’, ‘locator_type’: ‘xpath’, ‘locator_value’: ‘//input[@value=“登录”]’, ‘timeout’: 5.0}

}

• demo_element_excel_22.py

# -*- coding: utf-8 -*-
# @Time : 2022/1/6 15:45
# @Author : Limusen
# @File : demo_element_excel_22


import os
import xlrd3

current_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
excel_path = os.path.join(current_path, '..', 'element_info_datas')


class DemoExcel22:

    def __init__(self, module_name, page_name, element_path=excel_path):
        """
        :param module_name: 模块名,在element_info_datas模块下
        :param page_name: element_info_datas模块下的页面名称
        :param element_path:  excel的地址
        """
        # 指定excel的地址
        self.excel_path = os.path.join(element_path, module_name, page_name)

        # 创建excel对象
        self.work_book = xlrd3.open_workbook(self.excel_path)
        # 创建sheet对象 默认取第一个 一个页面一个excel
        self.sheet = self.work_book.sheet_by_index(0)
        # 获取excel总行数
        self.row_count = self.sheet.nrows

    def get_element_demo(self):
        info_elements = {}
        for info in range(1, self.row_count):
            # 可以发现这里会将不同的行数的数打印出来
            print(self.sheet.cell_value(info,3))
        return info_elements

    def get_element_info(self):
        info_elements = {}
        for info in range(1, self.row_count):
            # 取出每行然后进行拼接处理
            element_info = {}
            element_info['element_name'] = self.sheet.cell_value(info, 1)
            element_info['locator_type'] = self.sheet.cell_value(info, 2)
            element_info['locator_value'] = self.sheet.cell_value(info, 3)
            timeout_value = self.sheet.cell_value(info, 4)
            # 如果超时时间为浮点型则直接使用timeout_value 如果为空则使用数字5
            element_info['timeout'] = timeout_value if isinstance(timeout_value, float) else 5
            info_elements[self.sheet.cell_value(info, 0)] = element_info
        return info_elements


if __name__ == '__main__':
    ex = DemoExcel22("baidu", "baidu_page.xls")
    print(ex.get_element_demo())

代码图示

总结

本章节主要讲述的如何通过python进行excel的读写，并且将数据取出加工成我们想要的数据，到了这里离我们的框架完成已经进行一半啦! 继续坚持~

代码

地址 : https://gitee.com/todayisgoodday/PythonSelenium