Python+Selenium实现列表元素的查找定位及删除操作

Python+Selenium实现列表元素的查找及删除

获取列表（单页）全部数据

    # 获取列表单页全部数据
    def get_table_content(self, table_tbody):
        """
        获取列表单页全部数据
        :param table_tbody: 列表定位tbody
        :return:
        """
        time.sleep(1)
        list_1 = []
        # 表格定位路径
        element = self.driver.find_element(By.XPATH, table_tbody)
        # 获取每一行数据tr
        table_tr_list = element.find_elements(By.TAG_NAME, "tr")
        # 按行查询表格的数据，取出的数据是一整行
        for tr in table_tr_list:
            # tr.text获取表格每行的文本内容、切割字符串
            # list_2 = tr.get_attribute('textContent')
            list_2 = (tr.text).split()
            list_1.append(list_2)
        return list_1

删除某行元素

    # 删除某行元素
    def del_table_data(self, table_tbody, row, btn_fixed, del_sucessful_info):
        """
        删除列表元素
        :param table_tbody: 列表定位
        :param row: 删除数据所在行
        :param btn_fixed: （点击删除后，弹出框中区确认删除按钮）
        :param del_sucessful_info: （删除成功后，提示信息）
        :return:
        """
        # 表格定位路径
        element = self.driver.find_element(By.XPATH, table_tbody)
        # 获取每一行数据tr
        table_tr_list = element.find_elements(By.TAG_NAME, "tr")
        try:
        	# 对应行查找“删除”按钮并点击
            table_tr_list[row].find_element(By.XPATH, ".//td//span[text()='删除']").click()
            # 如果有是否确认删除的弹框，点击“确定”按钮
            self.click(btn_fixed, doc="点击确认按钮，删除数据")
            time.sleep(1)
            # 获取删除后的提示信息：“删除成功”
            ret_del_info = self.get_element_text(del_sucessful_info)
            time.sleep(0.5)
        except:
        	# 如果在对应行未找到“删除”按钮，则说明该数据不允许删除
            ret_del_info = "该数据不允许删除，请重新操作！"
        return ret_del_info

查找并删除元素（目前仅支持删除单条数据，循环删除待继续研究）

    # 查找并删除元素
    def find_and_del_table_data(self, table_tbody, queryContent, btn_fixed, del_sucessful_info):
        """
        循环遍历table数据，确定查询数据的位置
        :param table_tbody: 列表表格定位
        :param queryContent: 查询内容
        :param btn_fixed: （点击删除后，弹出框中区确认删除按钮）
        :param del_sucessful_info:（删除成功后，提示信息）
        :return:
        """
        # 调用获取列表（单页）全部数据函数
        arr_data = self.get_table_content(table_tbody)
        ret_info = ""
        del_info = ""
        # 嵌套for循环查找列表中元素的位置
        for i in range(len(arr_data)):
            for j in range(len(arr_data[i])):
                if queryContent == arr_data[i][j]:
                    ret_info = "'{}'坐标位置：（{},{}）".format(queryContent, i + 1, j + 1)
                    ret_row = i
                    print(ret_info)
                    # 调用删除元素函数
                    del_info = self.del_table_data(table_tbody, ret_row, 
                    btn_fixed, del_sucessful_info)
                    print(del_info)
                    time.sleep(2)
        if ret_info != "" and del_info != "该数据不允许删除，请重新操作！":
            ret_info = "'{}'元素存在且删除成功！！！".format(queryContent)
            return ret_info
        elif ret_info != "" and del_info == "该数据不允许删除，请重新操作！":
            ret_info = "'{}'元素不允许删除！！！".format(queryContent)
            return ret_info
        else:
            ret_info = "'{}'元素不存在，请重新查找！！！".format(queryContent)
            return ret_info

结果示例