Selenium沒有提供table的處理方法靶擦，只有根據(jù)需要自己編寫腳本核偿。主要思路是通過xpath下標(biāo)和css selector層級(jí)的方式迹缀，通過循環(huán)實(shí)現(xiàn)根據(jù)行列取數(shù)據(jù)瓦糕，根據(jù)數(shù)據(jù)判斷所在的行列。
一個(gè)簡單的table示例：
<html>
<head>
<meta charset="UTF-8">
<title>Table</title>
</head>
<body>
<table border="1" id="myTable">
<tbody>
<tr>
<th>表頭第一格</th>
<th>表頭第二格</th>
<th>表頭第三格</th>
</tr>
<tr>
<td>第一行第一列</td>
<td>第一行第二列</td>
<td>第一行第三列</td>
</tr>
<tr>
<td>第二行第一列</td>
<td>第二行第二列</td>
<td>第二行第三列</td>
</tr>
</tbody>
</table>
</body>
</html>
大致是這樣一個(gè)表格：
table簡單示例
根據(jù)行列取文本值
def row_cell_get_table_text(driver, locator, row, cell):
"""
根據(jù)輸入的行列值款筑，獲取該行列單元格中的文本智蝠。
:param locator: 定位到table的定位語句
:param row: 行
:param cell: 列
:return: 單元格文本
"""
locator = "%s/tbody/tr[%s]/td[%s]" % (locator, row + 1, cell)
text = driver.find_element_by_xpath(locator).text
return text
from selenium import webdriver
driver = webdriver.Chrome()
driver.get('file:///E:/table.html')
locator = "http://table[@id='myTable']"
row_cell_get_table_text(driver, locator, 2, 2)
# '第二行第二列'
根據(jù)文本獲取所在的行列
方法1：
使用xpath方式腾么，先獲取整個(gè)列表的tr行數(shù)，然后獲取所有的td數(shù)杈湾，tr/rd就是列數(shù)解虱。根據(jù)行列去遍歷整個(gè)列表，返回匹配的文本所在的行列毛秘。
def text_get_table_row_cell(driver, locator, text):
"""通過xpath定位的方式饭寺，根據(jù)參數(shù)text中的文本返回文本所在的行列"""
# 獲取行數(shù)，由于部分表格表頭是用th而不是用td叫挟，可能會(huì)出現(xiàn)計(jì)算錯(cuò)誤艰匙。因此這里先除去表頭
table_tr = driver.find_elements_by_xpath(locator + "/tbody/tr")[1:]
row = len(table_tr)
# 獲取列數(shù)
table_td = driver.find_elements_by_xpath(locator + "/tbody/tr/td")
cell = int(len(table_td) / row)
# 遍歷table中的所有文本，并匹配的值返回所在的行列
# xpath中下標(biāo)取值從1開始抹恳，除去表頭员凝，需要從2開始
for i in range(2, row + 2):
for j in range(1, cell + 1):
tl = locator + "/tbody/tr[" + str(i) + "]/td[" + str(j) + "]"
table_text = driver.find_element_by_xpath(tl).text
if text == table_text:
return i, j
from selenium import webdriver
driver = webdriver.Chrome()
driver.get('file:///E:/table.html')
locator = "http://table[@id='myTable']"
text_get_table_row_cell(driver,locator,'第一行第二列')
# (2, 2)
# 注意，這里算上表頭是第二行奋献，可以根據(jù)需要減1.
方法2：
通過css的方式健霹，先查找到所有table內(nèi)部的tr元素，再找tr內(nèi)部的所有td標(biāo)簽瓶蚂。同樣遍歷整個(gè)table糖埋，獲取文本所在的行列。
def text_get_table_row_cell_by_css(driver, csslocator, text):
"""通過css定位的方式窃这，根據(jù)參數(shù)text中的文本返回文本所在的行列瞳别，排除了表頭的情況"""
# 拼接定位語句
locator = csslocator + ">tbody>tr"
# 去表頭
table_tr_list = driver.find_elements_by_css_selector(locator)[1:]
# 遍歷table中的所有文本
for tr in range(len(table_tr_list)):
# 查找當(dāng)前tr中的所有td
table_td_list = table_tr_list[tr].find_elements_by_tag_name("td")
for td in range(len(table_td_list)):
if text == table_td_list[td].text:
return tr + 1, td + 1
from selenium import webdriver
driver = webdriver.Chrome()
driver.get('file:///E:/table.html')
locator = "#myTable"
text_get_table_row_cell_by_css(driver,locator,'第二行第三列')
# (2, 3)
# 這里排除了表頭的情況，如果需要杭攻，再加1.
現(xiàn)在表格可能會(huì)加多種樣式等情況祟敛，需要根據(jù)實(shí)際測試的頁面表格情況靈活調(diào)整。
再封裝一下兆解，封裝成一個(gè)類：
__author__ = "Nemo"
# =========================================
# 用來處理列表頁中的table
# =========================================
class Table(object):
def __init__(self, driver):
self.driver = driver
def row_cell_get_table_text(self, locator, row, cell):
"""
根據(jù)輸入的行列值馆铁，獲取該行列單元格中的文本。
:param locator: 定位到table的定位語句
:param row: 行
:param cell: 列
:return: 單元格文本
"""
locator = "%s/tbody/tr[%s]/td[%s]" % (locator, row+1, cell)
text = self.driver.find_element_by_xpath(locator).text
return text
def text_get_table_row_cell(self, locator, text):
"""通過xpath定位的方式锅睛，根據(jù)參數(shù)text中的文本返回文本所在的行列埠巨，排除了表頭的情況"""
# 獲取行數(shù)，除去表頭
table_tr = self.driver.find_elements_by_xpath(locator + "/tbody/tr")[1:]
row = len(table_tr)
# 獲取列數(shù)
table_td = self.driver.find_elements_by_xpath(locator + "/tbody/tr/td")
cell = int(len(table_td)/row)
# 根據(jù)行列數(shù)遍歷table中的所有文本衣撬，并獲取匹配的文本所在的行列
for i in range(2, row + 2):
for j in range(1, cell + 1):
tl = locator + "/tbody/tr[" + str(i) + "]/td[" + str(j) + "]"
table_text = self.driver.find_element_by_xpath(tl).text
if text == table_text:
return i, j
def text_get_table_row_cell_by_css(self, csslocator, text):
"""通過css定位的方式乖订，根據(jù)參數(shù)text中的文本返回文本所在的行列，排除了表頭的情況"""
locator = csslocator + ">tbody>tr"
# 去表頭
table_tr_list = self.driver.find_elements_by_css_selector(locator)[1:]
# 遍歷table中的所有文本
for tr in range(len(table_tr_list)):
table_td_list = table_tr_list[tr].find_elements_by_tag_name("td")
for td in range(len(table_td_list)):
if text == table_td_list[td].text:
return tr + 1, td + 1
from selenium import webdriver
driver = webdriver.Chrome()
driver.get('file:///E:/table.html')
table = Table(driver)
css_locator = "#myTable"
xpath_locator = "http://table[@id='myTable']"
table.row_cell_get_table_text(xpath_locator, 2, 2)
# '第二行第二列'
table.text_get_table_row_cell(xpath_locator, '第一行第二列')
# (2, 2)
table.text_get_table_row_cell_by_css(css_locator , '第二行第三列')
# (2, 3)
