//====================================================== // https://www.meitulu.com图片批量下载Node.js爬虫1.05 // 1.00 完成图片爬虫,手动输入页数和目录 // 1.01 改写。 // 1.02 手动输入页面url，然后自动解析 // 1.03 从命令行获得页面url，然后自动解析 // 1.04 解决数量节点位置不固定bug和输入状态不退出bug // 1.05 增加自动模式和手动模式 // 2017年11月7日 //====================================================== // 内置https模块，用来解析页面得到页数（图片总数）和目录（图片所在网络目录） var https=require("https"); // 内置http模块，用来通过地址得到图片文件 var http=require("http"); // 用于解析gzip网页（ungzip，https得到的网页是用gzip进行压缩的） var zlib = require('zlib'); // cheerio模块，提供了类似jQuery的功能，用于从HTML code中查找页数和目录 var cheerio = require("cheerio"); // 内置文件处理模块，用于创建目录和图片文件 var fs=require('fs'); // 请求参数JSON，http和https都有使用 var options; // request请求 var req; //-------------------------------------- // 程序入口 //-------------------------------------- function getInput(){ /*for(var i=30;i<37;i++){ process.stdout.write(" 33["+i+"m 看颜色: 33[39m"); } process.exit(); 31 红，32 绿，33 草黄，34 蓝，35 紫色，36 蓝绿 */ process.stdout.write(" 33[33m 自动模式输入0，手动模式选择1，请输入模式: 33[39m");// 草黄色 process.stdin.resume(); process.stdin.setEncoding('utf8'); process.stdin.on('data',function(text){ if(text.trim()=='1'){ // 手动模式 var endIndex=-1; var folder=""; process.stdout.write(" 33[36m 请输入页数/目录，以,分隔. 33[39m"); // 蓝绿色 process.stdin.on('data',function(text){ var inputText=text.trim();// trim()是必须的！ process.stdin.end(); var arr=inputText.split(","); endIndex=arr[0]; folder=arr[1]; console.log("页数="+endIndex); console.log("目录="+folder); fs.mkdir('./'+folder,function(err){ if(err){ console.log("目录"+folder+"已经存在"); } }); console.log("开始下载(手动模式)..."); // 下载图片 for(var i=1;i<=endIndex;i++){ downloadPic(folder,i); } }); }else{ // 自动模式 process.stdout.write(" 33[35m 请输入页面URL: 33[039m"); //紫色 process.stdin.on('data',function(text){ process.stdin.end();// 退出输入状态 start(text.trim());// trim()是必须的！ }); } }); } //-------------------------------------- // 开始下载 //-------------------------------------- function start(pageUrl){ console.log("开始下载(自动模式)..."); var hostName=""; var Path=""; var arr=pageUrl.split("/"); hostName=arr[2]; Path="/"+arr[3]+"/"+arr[4]; // 初始化options options={ hostname:hostName, port:443, path:Path,// 子路径 method:'GET', agent:false, gzip: true, }; req=https.request(options,function(resp){ var html = []; resp.on("data", function(data) { html.push(data); }) resp.on("end", function() { var buffer = Buffer.concat(html); zlib.gunzip(buffer, function(err, decoded) { var endIndex=-1; var folder=""; if(err){ console.log("不能得到html文本，因为"+err); console.log("请使用 1.手动模式 下载."); }else{ //console.log(decoded.toString());// gzip解压后的html文本 var body=decoded.toString(); var $ = cheerio.load(body); // 查找所有class为c_l的节点下面的p节点 $(".c_l p").each(function(index,element){ var text=$(element).text(); if(text.indexOf("数量")!=-1 && endIndex==-1){ var arr=text.split(" "); endIndex=arr[1]; } }) // 查找所有class为c_l的节点下面的p节点 $(".content center img").each(function(index,element){ if(index==0){ var text=$(element).attr("src"); var arr=text.split("/"); folder=arr[arr.length-2]; } }) console.log("页数="+endIndex); console.log("目录="+folder); fs.mkdir('./'+folder,function(err){ if(err){ console.log("目录"+folder+"已经存在"); } }); } // 下载图片 for(var i=1;i<=endIndex;i++){ downloadPic(folder,i); } }) }).on("error", function() { console.log("获取失败") }) }); // 超时处理 req.setTimeout(5000,function(){ req.abort(); }); // 出错处理 req.on('error',function(err){ if(err.code=="ECONNRESET"){ console.log('socket端口连接超时。'); }else{ console.log('请求发生错误，err.code:'+err.code); } }); // 请求结束 req.end(); } //-------------------------------------- // 下载图片 // folder:图片所在url的目录 // pinctureIndex:图片序号 //-------------------------------------- function downloadPic(folder,pinctureIndex){ console.log("图片:"+pinctureIndex+"下载开始"); // 初始化options options={ hostname:'mtl.ttsqgs.com',// 这里别加http://，否则会出现ENOTFOUND错误 port:80, path:'/images/img/'+folder+'/'+pinctureIndex+'.jpg',// 子路径 method:'GET', }; req=http.request(options,function(resp){ var imgData = ""; resp.setEncoding("binary"); resp.on('data',function(chunk){ imgData+=chunk; }); resp.on('end',function(){ var fileName="./"+folder+"/"+pinctureIndex+".jpg"; fs.writeFile(fileName, imgData, "binary", function(err){ if(err){ console.log("文件"+fileName+"下载失败."); } console.log("文件"+fileName+"下载成功"); }); }); }); // 超时处理 req.setTimeout(5000,function(){ req.abort(); }); // 出错处理 req.on('error',function(err){ if(err.code=="ECONNRESET"){ console.log('socket端口连接超时。'); }else{ console.log('请求发生错误，err.code:'+err.code); } }); // 请求结束 req.end(); } // 调用getInput函数，程序开始 getInput();
2017年11月7日10:08:05