Python+Selenium自动化脚本实现列表信息的多列内容获取

自动化脚本实现列表信息的多列内容获取（列数目可变）

取出多列信息后显示为一个列表，根据列数目进行列表的拆分，最终展示为多个子列表

    # 获取列表数据多列内容
    def get_list_multicolumn(self, table_tbody, *column):
        """
        :param table_tbody: 表格定位路径
        :param column:获取某几列元素
        :param doc:备注信息
        :return:将列内容返回至列表中
        """
        # 定位表格路径
        element = self.driver.find_element(By.XPATH, table_tbody)
        # 获取每一行的数据tr
        table_tr_list = element.find_elements(By.TAG_NAME, "tr")
        time.sleep(0.5)
        # 设置初始化列表
        table_list = []
        finally_list = []
        # 获取列表数据
        total_numbers = len(table_tr_list)
        if total_numbers >= 1:
            for i in range(1, total_numbers + 1):
            	# len(*column)可变参数的长度
                for j in range(len(*column)):
                    new_column = column[0][j]
                    table_td_list = self.driver.find_elements(By.XPATH, table_tbody + '//tr[' + str(i) + ']//td[' + str(new_column) + ']')[0].text
                    table_list.append(table_td_list)
            # 通过可变参数的长度拆分列表，展示为多个子列表
            for k in range(0, len(table_list), len(*column)):
                new_list_1 = table_list[k: k + len(*column)]
                finally_list.append(new_list_1)
            return finally_list
        else:
            ret_info = "列表数据为空"
            return ret_info

    # @unittest.skip("暂时跳过")
    def test_e_leave_application_alldatas_page1(self):
        """
        请假申请：请假类型
        :return:
        """
        # 调用函数，可变参数为：3、4、5、7、8列
        ret = DOP(self.driver).get_leave_list_multicolumn_applyData(DOL.leave_table_tbody_loc, 3, 4, 5, 7, 8)
        print(ret)
        time.sleep(0.5)

最终结果展示

屏蔽列表拆分

    # 获取列表数据多列内容
    def get_list_multicolumn(self, table_tbody, *column):
        """
        :param table_tbody: 表格定位路径
        :param column:获取某几列元素
        :param doc:备注信息
        :return:将列内容返回至列表中
        """
        # 定位表格路径
        element = self.driver.find_element(By.XPATH, table_tbody)
        # 获取每一行的数据tr
        table_tr_list = element.find_elements(By.TAG_NAME, "tr")
        time.sleep(0.5)
        # 设置初始化列表
        table_list = []
        finally_list = []
        # 获取列表数据
        total_numbers = len(table_tr_list)
        if total_numbers >= 1:
            for i in range(1, total_numbers + 1):
            	# len(*column)可变参数的长度
                for j in range(len(*column)):
                    new_column = column[0][j]
                    table_td_list = self.driver.find_elements(By.XPATH, table_tbody + '//tr[' + str(i) + ']//td[' + str(new_column) + ']')[0].text
                    table_list.append(table_td_list)
            # 屏蔽掉列表拆分
            # 通过可变参数的长度拆分列表，展示为多个子列表
            # for k in range(0, len(table_list), len(*column)):
            #     new_list_1 = table_list[k: k + len(*column)]
            #     finally_list.append(new_list_1)
            return table_list
        else:
            ret_info = "列表数据为空"
            return ret_info

展示结果