网站建设资讯

NEWS

网站建设资讯

封装操作表格的公用类(webdriver)

Table.py

1、可以用通过先定位到整个table表格对象;
2、再通过表格对象定位到行对象;
3、通过行对象再定位单元格对象;
4、通过单元格对象再定位单元格里面的元素对象

#encoding=utf-8
class Table(object):

    __table = None

    def __init__(self,table):
        self.setTable(table)

    def setTable(self,table):
        self.__table = table#此处传入的是整个表格对象

    def getTable(self):
        return self.__table

    def getRowCount(self):
        #返回table对象中所有的行tr标签元素对象
        return len(self.__table.find_elements_by_tag_name("tr"))

    #获取表格中的列数
    def getColumnCount(self):
        return len(self.__table.find_elements_by_tag_name("tr")[0].\
         find_elements_by_tag_name("td"))

    #获取表格中某行某列 单元格对象
    def getElementCell(self,rowNo,colNo):
        try:
            #先找到某一个行对象
            currentRow = self.__table.find_elements_by_tag_name("tr")[rowNo-1]
            #然后根据行找到该行中的某一列
            currentCol = currentRow.find_elements_by_tag_name("td")[colNo - 1]
            #返回该某行某列的单元格
            return currentCol
        except Exception as e:
            raise e

    #获取表格中某行某列 单元格中的元素对象
    def getWebElementInCell(self,rowNo,colNo,by,value):
        #by:定位方式 value:定位表达式
        try:
            currentRow = self.__table.find_elements_by_tag_name("tr")[rowNo - 1]
            currentCol = currentRow.find_elements_by_tag_name("td")[colNo - 1]

            #然后再根据单元格找到定位到元素对象
            element = currentCol.find_element(by,value)

            #返回元素对象
            return element
        except Exception as e:
            raise e

Test_table.py

#encoding=utf-8
from selenium import webdriver
import time,traceback
from Table import Table
import unittest

class TestTable(unittest.TestCase):

    def setUp(self):
        self.driver = webdriver.Chrome(executable_path = "d:\\chromedriver")

    def test_table(self):
        url = "http://127.0.0.1/test_table.html"
        self.driver.get(url)
        try:
            #获取table元素

            table = self.driver.find_element_by_tag_name("table")
            #获取table实例对象
            tableIns = Table(table)

            #获取表格的行数
            print(tableIns.getRowCount())

            #获取表格的列数
            print(tableIns.getColumnCount())

            #获取第2行第3列单元格对象
            cell = tableIns.getElementCell(2,3)
            print(cell)
            print("第2行第3列单元格内容: ",cell.text)

            #获取第2行第3列文本输入框元素对象
            cellInput = tableIns.getWebElementInCell(2,3,"tag name","input")
            cellInput.send_keys("第2行第3列输入框被找到!")
            time.sleep(3)
        except Exception as e:
            print(traceback.print_exc())

    def tearDown(self):
        self.driver.quit()

if __name__== "__main__":
    unittest.main()

文章标题:封装操作表格的公用类(webdriver)
分享链接:http://njwzjz.com/article/goscog.html