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("")
     except:
         ActionChains(browser).send_keys(Keys.ENTER).perform()
         print("")

 browser.quit()

在我的github也放着呀：

御伽草纸