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开始