乐趣区

前端使用puppeteer 爬虫生成《React.js 小书》PDF并合并

1、puppeteer 是什么?
puppeteer: Google 官方出品的 headless Chrome node 库 puppeteer github 仓库 puppeteer API
官方介绍:
您可以在浏览器中手动执行的大多数操作都可以使用 Puppeteer 完成!生成页面的屏幕截图和 PDF。抓取 SPA 并生成预渲染内容(即“SSR”)。自动化表单提交,UI 测试,键盘输入等。创建最新的自动化测试环境。使用最新的 JavaScript 和浏览器功能直接在最新版本的 Chrome 中运行测试。捕获时间线跟踪 您的网站,以帮助诊断性能问题。测试 Chrome 扩展程序。

2、爬取网站生成 PDF

2.1 安装 puppeteer
// 安装 puppeteer
// 可能会因为网络原因安装失败,可使用淘宝镜像
// npm install -g cnpm –registry=https://registry.npm.taobao.org
npm i puppeteer
# or “yarn add puppeteer”
2.2《React.js 小书》简介

《React.js 小书》简介 关于作者 @胡子大哈这是⼀本关于 React.js 的⼩书。因为⼯作中⼀直在使⽤ React.js,也⼀直以来想总结⼀下⾃⼰关于 React.js 的⼀些知识、经验。于是把⼀些想法慢慢整理书写下来,做成⼀本开源、免费、专业、简单的⼊⻔级别的⼩书,提供给社区。希望能够帮助到更多 React.js 刚⼊⻔朋友。下图是《React.js 小书》部分截图:

2.3 一些可能会用到的 puppeteer API

// 新建 reactMiniBook.js, 运行 node reactMiniBook.js 生成 pdf
const puppeteer = require(‘puppeteer’);

(async () => {
// 启动浏览器
const browser = await puppeteer.launch({
// 无界面 默认为 true, 改成 false, 则可以看到浏览器操作,目前生成 pdf 只支持无界面的操作。
// headless: false,
// 开启开发者调试模式,默认 false, 也就是平时 F12 打开的面版
// devtools: true,
});
// 打开一个标签页
const page = await browser.newPage();
// 跳转到页面 http://huziketang.mangojuice.top/books/react/
await page.goto(‘http://huziketang.com/books/react/’, {waitUntil: ‘networkidle2’});
// path 路径,format 生成 pdf 页面格式
await page.pdf({path: ‘react.pdf’, format: ‘A4’});
// 关闭浏览器
await browser.close();
})();
知道这启动浏览器打开页面关闭浏览器主流程后,再来看几个 API。
const args = 1;
let wh = await page.evaluate((args) => {
// args 可以这样传递给这个函数。
// 类似于 setTimeout(() => {console.log(args);}, 3000, args);
console.log(‘args’, args); // 1
// 这里可以运行 dom 操作等 js
// 返回通过 dom 操作等获取到的数据
return {
width: 1920,
height: document.body.clientHeight,
};
}, args);
// 设置视图大小
await page.setViewport(wh);
// 等待 2s
await page.waitFor(2000);
// 以 iPhone X 执行。
const devices = require(‘puppeteer/DeviceDescriptors’);
const iPhone = devices[‘iPhone X’];
await page.emulate(iPhone);
2.4 知道了以上这些 API 后,就可以开始写主程序了。
简单说下:实现功能和主流程。从上面 React.js 小书截图来看。1、打开浏览器,进入目录页,生成 0. React 小书 目录.pdf2、跳转到 1. React.js 简介页面,获取左侧所有的导航 a 链接的 href, 标题。3、用获取到的 a 链接数组进行 for 循环,这个循环里主要做了如下几件事:
3.1 隐藏左侧导航,便于生成 pdf 3.2 给 React.js 简介等标题 加上序号,便于查看 3.3 设置 docment.title 加上序号, 便于在页眉中使用。3.4 隐藏 传播一下知识也是一个很好的选择 这一个模块(因为页眉页脚中设置了书的链接等信息,就隐藏这个了)3.5 给 分页 上一节,下一节加上序号,便于查看。3.6 最末尾声明下该 pdf 的说明,仅供学习交流,严禁用于商业用途。3.7 返回宽高,用于设置视图大小 3.8 设置视图大小,创建生成 pdf

4、关闭浏览器
具体代码:可以查看这里爬虫生成《React.js 小书》的 pdf 每一小节的代码
// node 执行这个文件
// 笔者这里是:
node src/puppeteer/reactMiniBook.js
即可生成如下图:每一小节(0-46 小节)的 pdf

生成这些后,那么问题来了,就是查看时总不能看一小节,打开一小节来看,这样很不方便。于是接下来就是合并这些 pdf 成为一个 pdf 文件。
3、合并成一个 PDF 文件 pdf-merge

起初,我是使用在线网站 Smallpdf,合并 PDF。合并的效果还是很不错的。这网站还是其他功能。比如 word 转 pdf 等。后来找到社区提供的一个 npm packagepdf merge。(毕竟笔者是写程序的,所以就用代码来实现合并了)
这个 pdf-merge 依赖 pdftk
安装 PDFtkWindows 下载并安装笔者安装后,重启电脑才能使用。
Debian, Ubuntu 安装笔者在 Ubuntu 系统安装后,即可使用。apt-get install pdftk
使用例子
const PDFMerge = require(‘pdf-merge’);

const files = [
`${__dirname}/1.pdf`,
`${__dirname}/2.pdf`,
];

// Buffer (Default)
PDFMerge(files)
.then((buffer) => {…});

// Stream
PDFMerge(files, {output: ‘Stream’})
.then((stream) => {…});

// 笔者这里使用的是这个
// Save as new file
PDFMerge(files, {output: `${__dirname}/3.pdf`})
.then((buffer) => {…});
知道这些后,可以开始写主程序了。简单说下主流程 1、读取到生成的所有 pdf 文件路径,并排序(0-46)2、判断下输出文件夹是否存在,不存在则创建 3、合并这些小节的 pdf 保存到新文件 React 小书(完整版)- 作者:胡子大哈 - 时间戳.pdf
具体代码:可以查看这里爬虫生成《React.js 小书》的 pdf 合并 pdf 的代码
最终合并的 pdf 文件在这里 React 小书(完整版)- 作者:胡子大哈,可供下载。
本想着还可以加下书签和页码,没找到合适的生成方案,那暂时先不加了。如果读者有好的方案,欢迎与笔者交流。
关于
作者:常以轩辕 Rowboat 为名混迹于江湖。前端路上 | PPT 爱好者 | 所知甚少,唯善学。个人博客 segmentfault 个人主页掘金个人主页知乎 github
小结
1、puppeteer 是 Google 官方出品的 headless Chrome node 库,可以在浏览器中手动执行的大多数操作都可以使用 Puppeteer 完成。总之可以用来做很多有趣的事情。2、用 puppeteer 生成每一小节的 pdf,用依赖 pdftk 的 pdf-merge npm 包, 合并成一个新的 pdf 文件。或者使用 Smallpdf 等网站合并。3、《React.js 小书》,推荐给大家。爬虫生成 pdf,应该不会对作者 @胡子大哈有什么影响。作者写书服务社区不易,尽可能多支持作者。
最后推荐几个链接,方便大家学习 puppeteer。puppeteer 入门教程 Puppeteer 初探之前端自动化测试爬虫生成 ES6 标准入门 pdf 大前端神器安利之 Puppeteerpuppeteer API 中文文档

退出移动版