requests從selenium獲取cookies

# coding:utf-8
# 用webdriver登錄並獲取cookies，並用requests發送請求，以豆瓣為例
from selenium import webdriver
import requests
import time
import json
import sys
reload(sys)
sys.setdefaultencoding('utf-8')

def main():
    # 從命令行參數獲取登錄用戶名和密碼
    user_name = sys.argv[1]
    password = sys.argv[2]

    # 豆瓣登錄頁面URL
    login_url = 'https://www.douban.com/accounts/login'

    # 獲取chrome的配置
    opt = webdriver.ChromeOptions()
    # 在運行的時候不彈出瀏覽器窗口
    # opt.set_headless()

    # 獲取driver對象
    driver = webdriver.Chrome(chrome_options = opt)
    # 打開登錄頁面
    driver.get(login_url)

    print 'opened login page...'
    
    # 向瀏覽器發送用戶名、密碼，並點擊登錄按鈕
    driver.find_element_by_name('form_email').send_keys(user_name)
    driver.find_element_by_name('form_password').send_keys(password)
    # 多次登錄需要輸入驗證碼，這里給一個手工輸入驗證碼的時間
    time.sleep(6)
    driver.find_element_by_class_name('btn-submit').submit()
    print 'submited...'
    # 等待2秒鍾
    time.sleep(2)

    # 創建一個requests session對象
    s = requests.Session()
    # 從driver中獲取cookie列表（是一個列表，列表的每個元素都是一個字典）
    cookies = driver.get_cookies()
    # 把cookies設置到session中
    for cookie in cookies:
        s.cookies.set(cookie['name'],cookie['value'])
    # 關閉driver
    driver.close()

    # 需要登錄才能看到的頁面URL
    page_url = 'https://www.douban.com/accounts/'
    # 獲取該頁面的HTML
    resp = s.get(page_url)
    resp.encoding = 'utf-8'
    print 'status_code = {0}'.format(resp.status_code)
    # 將網頁內容存入文件
    with open('html.txt','w+') as  fout:
        fout.write(resp.text)
    
    print 'end'

if __name__ == '__main__':
    main()