封裝操作表格的公用類（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()