[러닝-스크래핑2기] 아이투자 재무제표

import pandas as pd
from selenium import webdriver

id = "YOUR-ID"
pw = "YOUR-PW"

driver = webdriver.Chrome('chromedriver.exe')
driver.get('http://www.itooza.com/')

# 로그인
ui = driver.find_element_by_css_selector("#cid")
ui.send_keys(id)
ui = driver.find_element_by_css_selector("#cpw")
ui.send_keys(pw)

ui = driver.find_element_by_css_selector("#certify > form > div.login-box-wrap > div.btn-login > input[type=image]")
ui.click()

# 재무제표 페이지로 이동
ticker = "006800"
driver.get(f'http://www.itooza.com/vclub/y10_page.php?cmp_cd={ticker}&mode=dy&ss=10&sv=1')

## 하나의 값을 읽기
ui = driver.find_element_by_css_selector("#node-1 > td:nth-of-type(2)")
print(ui.text)

ui = driver.find_element_by_css_selector("#node-1 > td:nth-of-type(3)")
print(ui.text)

# 전체를 데이터 프레임으로 읽기
html = driver.page_source
dfs = pd.read_html(html)
print(dfs[2].head())