2016-04-07 5 views
1

Я пытаюсь получить доступ к данным на этом сайте: http://surge.srcc.lsu.edu/s1.html. До сих пор у меня есть цикл кода через два раскрывающихся меню, но таблица динамически называется, и у меня возникают проблемы с получением данных из нее. Я пытался получить доступ к данным через класс выше «output_data_table», но у меня были проблемы.Dynamic Table Scraping with selenium in python

# importing libraries 
from selenium import webdriver 
import time 
from selenium.webdriver.support.ui import Select 
import lxml.html 



driver = webdriver.Firefox() 
driver.get("http://surge.srcc.lsu.edu/s1.html") 

# definition for switching frames 
def frame_switch(css_selector): 
    driver.switch_to.frame(driver.find_element_by_css_selector(css_selector)) 

frame_switch("iframe") 

html_source = driver.page_source 
nameSelect = Select(driver.find_element_by_xpath('//select[@id="storm_name"]')) 
stormCount = len(nameSelect.options) 

for i in range(1, stormCount): 
    print("starting loop on option storm " + nameSelect.options[i].text) 
    nameSelect.select_by_index(i) 
    time.sleep(3) 


    yearSelect = Select(driver.find_element_by_xpath('//select[@id="year"]')) 
    yearCount = len(yearSelect.options) 
    for j in range(1, yearCount): 
     print("starting loop on option year " + yearSelect.options[j].text) 
     yearSelect.select_by_index(j) 


     root = lxml.html.fromstring(driver.page_source) 

     #table=driver.find_element_by_id("output_data_table") 

     for row in root.xpath('.//table[@id="output_data_table"]//tr'): 
     # needs dynamic table name 
      cells = row.xpath('.//td/text()') 
      dict_value = {'0th': cells[0], 
        '1st': cells[1], 
        '2nd': cells[2], 
        '3rd': cells[3], 
        '4th': cells[5], 
        '5th': cells[6], 
        '6th': cells[7], 
        '7th': cells[8]} 
      print(dict_value) 
+0

В чем конкретно проблема на данный момент? Благодарю. – alecxe

+0

Похоже, вам нужно подождать, прежде чем называть «root = lxml.html.fromstring (driver.page_source)». Если вы не ждете, вы получите источник html без создания таблицы javascript. Поместите «time.sleep (10)» перед этим –

ответ

0

Похоже, вы должны ждать перед вызовом "корень = lxml.html.fromstring (driver.page_source)".

Если вы не ждете, вы получите источник html без создания таблицы javascript. Поставьте перед ним «time.sleep (10)».

Кажется, что это таблица. Я использовал BeautifulSoup для быстрого примера.

from selenium import webdriver 
import time, re 
from selenium.webdriver.support.ui import Select 
import lxml.html 
from bs4 import BeautifulSoup 

driver = webdriver.Firefox() 
driver.get("http://surge.srcc.lsu.edu/s1.html") 

# definition for switching frames 
def frame_switch(css_selector): 
    driver.switch_to.frame(driver.find_element_by_css_selector(css_selector)) 

frame_switch("iframe") 

html_source = driver.page_source 

nameSelect = Select(driver.find_element_by_xpath('//select[@id="storm_name"]')) 
stormCount = len(nameSelect.options) 

for i in range(1, stormCount): 
    print("starting loop on option storm " + nameSelect.options[i].text) 
    nameSelect.select_by_index(i) 
    time.sleep(3) 


    yearSelect = Select(driver.find_element_by_xpath('//select[@id="year"]')) 
    yearCount = len(yearSelect.options) 
    for j in range(1, yearCount): 
     print("starting loop on option year " + yearSelect.options[j].text) 
     yearSelect.select_by_index(j) 


     time.sleep(10) 

     soup = BeautifulSoup(driver.page_source, 'html.parser') 

     # get the needed table body 
     print soup.find_all("tbody", {"class" : re.compile(".*")})[1].prettify() 


     # print out each column 

     get_table = soup.find_all("tbody", {"class" : re.compile(".*")})[1] 
     columns = get_table.find_all("tr") 

     for column in columns: 
      print column.getText()