WebTable 扩展

 # coding:utf-8
 """
 页面 table处理
 """

 from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.common.exceptions import NoSuchElementException

 class WebTable(object):

     def __init__(self, webElement):
         self.webTable = webElement

     #得到表格中的行数
     def getRowCount(self):
         rowConunts = self.webTable.find_elements(By.TAG_NAME, 'tr')
         return len(rowConunts)

     #得到指定行的列数
     def getColCount(self, rowIdx):
         try:
             rowCounts = self.webTable.find_elements(By.TAG_NAME, 'tr')

             if len(rowCounts) < rowIdx:
                 raise "当前行数大于表格行数"

             #取得当前的 tr
             rowNum = rowCounts[rowIdx]

             #计算当前行的 td 数
             colCounts = rowNum.find_elements(By.TAG_NAME, 'td')
             return len(colCounts)
         except NoSuchElementException as e:
             raise NoSuchElementException("Failed to get the cell")

     #得到指定单元格内容， 传入指定的行数、列数作为参数
     def getCellText(self, rowIdx, colIdx):
         try:
             rowCounts = self.webTable.find_elements(By.TAG_NAME, 'tr')

             if len(rowCounts) < rowIdx:
                 raise "当前行数大于表格行数"

             #得到对应的行数
             currentRow = rowCounts[rowIdx]
             #获取行中所有的td
             td = currentRow.find_elements(By.TAG_NAME, 'td')

             if len(td) < colIdx:
                 raise "当前列数大于表格列数"

             #取得对应的单元格
             cell = td[colIdx]
             return cell.text
         except NoSuchElementException as e:
             raise NoSuchElementException("Failed to get the cell")

 if __name__ == '__main__':
     driver = webdriver.Firefox()
     driver.get('http://www.w3school.com.cn/tags/tag_table.asp')
     temp_webTable = WebTable(driver.find_element(By.TAG_NAME, 'table'))
     print temp_webTable.getRowCount()
     print temp_webTable.getColCount(3)
     print temp_webTable.getCellText(3, 2)     #行和列的索引从0开始