-
Notifications
You must be signed in to change notification settings - Fork 85
/
Copy pathindex.js
128 lines (112 loc) · 4.75 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
var http = require('http');
var fs = require('fs');
var cheerio = require('cheerio');
var request = require('request');
var inquirer = require('inquirer');
var booksPage = [];
var books = [];
var numOfPages = 1;
var topic = "";
var inquirer = require('inquirer');
inquirer.prompt([{
type: 'input',
name: 'word',
message: '您要找的关键词:'
}]).then(function (answers) {
topic = answers.word;
console.log("你要寻找关于"+topic+"的文件");
var url = "http://www.allitebooks.com/page/1/?s="+topic ;
fetchPage(url);//抓取所有
//startRequest(url);//抓取首页,测试用
});
function fetchPage(x) { //封装了一层函数
http.get(x, function (res) {
var html = ''; //用来存储请求网页的整个html内容
var titles = [];
res.setEncoding('utf-8'); //防止中文乱码
//监听data事件,每次取一块数据
res.on('data', function (chunk) {
html += chunk;
});
//监听end事件,如果整个网页内容的html都获取完毕,就执行回调函数
res.on('end', function () {
var $ = cheerio.load(html); //采用cheerio模块解析html
if( $('a[title="Last Page →"]').text()){//找出页码数量,分为大于5,小于5两种
console.log("超过5页");
numOfPages = $('a[title="Last Page →"]').text().trim();
}else{
console.log("小于5页");
numOfPages = $('div.pagination a');
if(numOfPages[numOfPages.length-1]){
numOfPages = numOfPages[numOfPages.length-1].attribs.title;
}else{
numOfPages = 1;
}
}
console.log(numOfPages);
console.log("总共"+numOfPages+"页");
if(numOfPages<10){
for(let i =1 ; i<=numOfPages; i++){
startRequest("http://www.allitebooks.com/page/"+i+"/?s="+topic);
}
}else{
for(let i =1 ; i<=numOfPages; i++){
var oneSecond = 10000 * i; // 防止反爬虫,每十秒一页
setTimeout(function() {
console.log("==============="+i+"=============");
startRequest("http://www.allitebooks.com/page/"+i+"/?s="+topic);
}, oneSecond);
}
}
}).on('error', function (err) {
console.log(err);
});
});
}
function startRequest(x) {//采用http模块向服务器发起一次get请求
http.get(x, function (res) {//爬取目录中每本书地址
var html = ''; //用来存储请求网页的整个html内容
var titles = [];
res.setEncoding('utf-8'); //防止中文乱码
res.on('data', function (chunk) { //监听data事件,每次取一块数据
html += chunk;
});
res.on('end', function () {//监听end事件,如果整个网页内容的html都获取完毕,就执行回调函数
var $ = cheerio.load(html); //采用cheerio模块解析html
var booksDOM = $('h2.entry-title a');
for(let i =0 ;i<10; i++){
if(booksDOM[i]){
booksPage.push({
link:booksDOM[i].attribs.href,
title:booksDOM[i].children[0].data,
});
console.log(booksDOM[i].attribs.href);
http.get(booksDOM[i].attribs.href, function (res) {//爬取每本书的下载地址
var html = '';
var titles = [];
res.setEncoding('utf-8');
res.on('data', function (chunk) {
html += chunk;
});
res.on('end', function () {
var $ = cheerio.load(html); //采用cheerio模块解析html
if($('i.fa-download').parent()[0]){
var bookDOM = $('i.fa-download').parent()[0].attribs.href;
console.log(bookDOM);
fs.appendFile('./'+topic+'pdf.txt', bookDOM+"\n",function(err){
if(err) console.log('写文件操作失败');
else console.log('写文件操作成功');
});
}
}).on('error', function (err) {
console.log(err);
});
});
}
}
//console.log(booksPage);
}).on('error', function (err) {
console.log(err);
});
});
}