테이블 행 및 셀 배치하기

# -*- coding:utf-8 -*-



from selenium import webdriver

from selenium.webdriver.common.by import By

import unittest



class TableTest(unittest.TestCase):

    def setUp(self):

        self.driver = webdriver.Firefox()

        

        

    def test_table(self):

        driver = self.driver

        driver.get('http://www.w3school.com.cn/html/html_tables.asp')

        #      tr   

        rows = driver.find_elements_by_css_selector('.dataintable>tbody>tr')

        print 'Table   : ', len(rows)

        

        for tr in rows:

            tds = tr.find_elements_by_tag_name('td')        #     tr    td   

            if tds:

                for td in tds:

                    print td.text                           #   td     

    

    def tearDown(self):

        self.driver.quit()

        

if __name__ == '__main__':

    unittest.main()