Selenium处理常见web控件-table

Selenium没有提供table的处理方法,只有根据需要自己编写脚本。主要思路是通过xpath下标和css selector层级的方式,通过循环实现根据行列取数据,根据数据判断所在的行列。
一个简单的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>

大致是这样一个表格:


table简单示例

根据行列取文本值

def row_cell_get_table_text(driver, locator, row, cell):
    """
    根据输入的行列值,获取该行列单元格中的文本。
    :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 = "//table[@id='myTable']"
row_cell_get_table_text(driver, locator, 2, 2)
# '第二行第二列'

根据文本获取所在的行列

方法1:
使用xpath方式,先获取整个列表的tr行数,然后获取所有的td数,tr/rd就是列数。根据行列去遍历整个列表,返回匹配的文本所在的行列。

def text_get_table_row_cell(driver, locator, text):
    """通过xpath定位的方式,根据参数text中的文本返回文本所在的行列"""

    # 获取行数,由于部分表格表头是用th而不是用td,可能会出现计算错误。因此这里先除去表头
    table_tr = driver.find_elements_by_xpath(locator + "/tbody/tr")[1:]
    row = len(table_tr)

    # 获取列数
    table_td = driver.find_elements_by_xpath(locator + "/tbody/tr/td")
    cell = int(len(table_td) / row)

    # 遍历table中的所有文本,并匹配的值返回所在的行列
    # xpath中下标取值从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 = "//table[@id='myTable']"
text_get_table_row_cell(driver,locator,'第一行第二列')
# (2, 2)
# 注意,这里算上表头是第二行,可以根据需要减1.

方法2:
通过css的方式,先查找到所有table内部的tr元素,再找tr内部的所有td标签。同样遍历整个table,获取文本所在的行列。

def text_get_table_row_cell_by_css(driver, csslocator, text):
    """通过css定位的方式,根据参数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)):
        # 查找当前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.

现在表格可能会加多种样式等情况,需要根据实际测试的页面表格情况灵活调整。
再封装一下,封装成一个类:

__author__ = "Nemo"
# =========================================
#     用来处理列表页中的table
# =========================================


class Table(object):

    def __init__(self, driver):
        self.driver = driver

    def row_cell_get_table_text(self, locator, row, cell):
        """
        根据输入的行列值,获取该行列单元格中的文本。
        :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定位的方式,根据参数text中的文本返回文本所在的行列,排除了表头的情况"""

        # 获取行数,除去表头
        table_tr = self.driver.find_elements_by_xpath(locator + "/tbody/tr")[1:]
        row = len(table_tr)

        # 获取列数
        table_td = self.driver.find_elements_by_xpath(locator + "/tbody/tr/td")
        cell = int(len(table_td)/row)

        # 根据行列数遍历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定位的方式,根据参数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 = "//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)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,185评论 6 503
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,652评论 3 393
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,524评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,339评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,387评论 6 391
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,287评论 1 301
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,130评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,985评论 0 275
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,420评论 1 313
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,617评论 3 334
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,779评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,477评论 5 345
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,088评论 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,716评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,857评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,876评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,700评论 2 354

推荐阅读更多精彩内容

  • 问答题47 /72 常见浏览器兼容性问题与解决方案? 参考答案 (1)浏览器兼容问题一:不同浏览器的标签默认的外补...
    _Yfling阅读 13,751评论 1 92
  • 开始注意是因为文章的名字。后来看文知道了是女主内心的inner peace. 写权谋,不够老辣。不过温暖。另,这个...
    叶开开阅读 222评论 0 0
  • 从去年10月份开始,一直到现在,装修还没有完结。从一开始的兴致勃勃,到现在已经接近心死。期间种种几乎出离言语。顶多...
    蒋菱阅读 510评论 0 0
  • 一念秋风起,落叶簌簌。 天蓝的让我忘记想你。心中的快意是那样酣畅淋漓。这个秋日,空旷明净,高而深远。美好的让人不忍...
    一剪芳踪阅读 709评论 0 3
  • 作者:朱金灿来源:http://blog.csdn.net/clever101 这个问题很普遍。最近在研究这个问题...
    舌尖上的大胖阅读 1,795评论 0 3