python+selenium webdriver 如何处理table

时间：2014-12-18 13:32:00 阅读：2864 评论：0 收藏：0 [点我收藏+]

Table对象是自动化测试中经常需要处理的对象。由于webdriver中没有专门的table类，所以我们需要简单的封装出一个易用易扩展的Table类来帮助简化代码

module EasyWrap
    class EasyWrapError < StandardError;end
    class NotValidElementError < EasyWrapError;end 
    class IncorrectdIndexError < EasyWrapError;end 
  
    class TableBase
        attr_reader :e
        def initialize e
            raise NotValidElementError unless e.is_a?(Selenium::WebDriver::Element)
            @e = e 
        end
  
        def method_missing(m, *params, &blk)
            @e.send(m, *params) if @e.respond_to?(m)
        end
    end #TableBase
    class Table < TableBase
        def initialize e
            super(e)
            __rows
        end
        def __rows
            @rows = @e.find_elements(:css => ‘tr‘)
        end
        private :__rows
  
        def rows
            all_rows = []   
            @rows.each { |r| rows << TableRow.new(r)}
            all_rows
        end
  
        def row_count
            @rows.size
        end
  
        def [](index)
            valid_index? index
            TableRow.new @rows[index]
        end
  
        def valid_index?(index)
            raise IncorrectdIndexError if index.to_i > row_count
        end
    end #Table
  
    class TableRow < TableBase
        def initialize e
            super(e)
            __cells
        end
  
        def __cells 
            @cells = @e.find_elements(:tag_name => ‘td‘) 
            # 如果找不到td那么试着去找th
            @cells = @e.find_elements(:tag_name => ‘th‘) if @cells.empty?
        end
        private :__cells
  
        def cells
            all_cells = []
            @cells.each {|c| all_cells << TableCell.new(c)} 
        end
  
        def cell_count
            @cells.size
        end
  
        def [](index)
            valid_index? index
            TableCell.new @cells[index]
        end
  
        def valid_index?(index)
            raise IncorrectdIndexError if index.to_i > cell_count
        end
    end #TableRow
  
    class TableCell < TableBase
    end #TableCell
end #EasyWrap

EasyWrap定义了3个类

Table类，代表1个table对象，可以通过该对象来访问该table的行和列；
TableRow类代表table的某1行，可以通过其访问table的某一列；
TableCell类代表table的某1个单元格

假如我们需要点击table的第3行第1列，通过EasyWrap，我们可以这样来实现

talbe = EasyWrap::Table(dr.find_element(:id => ‘table_id‘))
table[2][0].click

以下面的html代码为例：

<html>
    <head>
        <title>Table</title>
        <style>
            table {border: 1px solid #ccc}
        </style>
    </head>
    <body>
        <table id = "t">
            <th>C1</th><th>C2</th><th>C3</th>
            <tr>
                <td>v1</td>
                <td>v2</td>
                <td>v3</td>
            </tr>
            <tr>
                <td>k1</td>
                <td>k2</td>
                <td>k3</td>
            </tr>
        </table>
    </body>
</html>

接下来的脚本实现了高亮页面上table元素首行首列的功能

require ‘rubygems‘
require ‘selenium-webdriver‘
require ‘./easy_wrap‘
include EasyWrap
  
dr = Selenium::WebDriver.for :firefox
table_file = ‘file:///‘+File.expand_path(File.join(File.dirname(__FILE__), ‘table.html‘))
dr.get table_file
t = dr.find_element(:id => ‘t‘)
table = Table.new t
p table.row_count
highlight = <<JS
    arguments[0].style.border = "2px solid red"
JS
dr.execute_script highlight, table[0][0].e
# 打印第2行第2列的值
puts table[1][1].text

python+selenium webdriver 如何处理table

标签：style ar color sp for on div bs 代码

原文地址：http://www.cnblogs.com/jane0912/p/4171317.html

踩

(0)

评论一句话评论（0）

分享档案

更多>

2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)

周排行