python selenium下载电子书

有人推荐书籍《御伽草纸》，网上找了很久都找不到下载，估计是被Amazon版权了，但是在网易云阅读看到有书，所以就写个代码下载下来。

由于网易云阅读是js加载，用requests或者下载html的方法都太麻烦（毕竟这本书也才8万字），所以就简单粗暴的用selenium下载，而且还是截图（太懒啦...）

得到的图片被我转换为pdf：

 

附上代码：

#!/usr/bin/python3.4
# -*- coding: utf-8 -*-

from selenium import webdriver
import time
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.action_chains import ActionChains

# http://www.cnblogs.com/fnng/p/3238685.html
# 打开火狐浏览器
browser = webdriver.Chrome()
# 设置浏览器大小
browser.set_window_size(1200, 900)
# 输入网址
browser.get("http://yuedu.163.com/book_reader/ee677a1b42ed4af3b52adbe4c0fb6a23_4")
# 根据各自网速来判断网址加载时间
time.sleep(10)

# 选择阅读方式
browser.find_element_by_class_name("portrait").click()
time.sleep(5)
for i in range(3000):
    # 截图
    browser.save_screenshot("../jpg/txt/" + str(i + 1) + ".png")
    time.sleep(5)

    # 点击右边向右翻页
    try:
        ActionChains(browser).send_keys(Keys.ARROW_RIGHT).perform()
        # ActionChains(browser).click().perform()
        print("111111")
    except:
        ActionChains(browser).send_keys(Keys.ENTER).perform()
        print("2222222222")

browser.quit()

在我的github也放着呀：

御伽草纸