#!/usr/bin/env python
# -*- coding:utf-8 -*- 
# Author: json_steve

from selenium import webdriver
import sys
reload(sys)
sys.setdefaultencoding('utf-8')


def selenium_base_use():

    # 建立瀏覽器物件
    driver = webdriver.PhantomJS()

    # 傳送請求
    driver.get('http://www.baidu.com')

    # 點選新聞按鈕
    driver.find_element_by_name('tj_trnews'
 
).click()

    # 輸入框輸入資料 ：unicode編碼
    driver.find_element_by_id('ww').send_keys(u'json')

    # 點選一下百度一下按鈕
    driver.find_element_by_class_name('btn').click()

    # 點選一條新聞
    driver.find_element_by_xpath('//*[@id="1"]/h3/a').click()

    # 找到新開的頁面,list
    print driver.window_handles

    # 根據下表索引 切換視窗
 

    # driver.switch_to_window(driver.window_handles[1]) # 過去但是可以用
    driver.switch_to.window(driver.window_handles[1])

    # 檢視當前網址
    current_url = driver.current_url
    print current_url

    # 獲取所有cookie
    driver.get_cookies()

    # 快照
    # driver.save_screenshot("1baidu.png")

    # 獲取內容
 

    data = driver.page_source
    data = data.decode('utf-8').encode('gbk')
    with open('baidu.html', 'w') as f:
         f.write(data)


    # 關閉當前的頁面
    # driver.close()
    # 關閉瀏覽器
    # driver.quit()

if __name__ == '__main__':
    selenium_base_use()