命令行批量截圖Node腳本示例代碼

2024-05-06 16:47:53

字體：大中小

來源：轉載

供稿：網友

作為一個軟件工程師，不只是做好自己的本職工作（iOS），而是需要解決項目中的技術問題。這次就是解決自動截圖的問題

早期公司的數據工程師利用 phantomjs 來截圖，后期不斷發現截圖效率低，加之開發者團隊不再維護，因此決定將截圖這部分跟你剝離開來，以后方便開發維護。我就承擔了這個工作

puppeteer

Puppeteer is a Node library which provides a high-level API to control Chrome or Chromium over the DevTools Protocol. Puppeteer runs headless by default, but can be configured to run full (non-headless) Chrome or Chromium.

安裝問題

一開始按照往常的套路（npm install --save puppeteer）好幾次都卡住了，后期查找資料發現切換到國內的鏡像就可以順利下載

PUPPETEER_DOWNLOAD_HOST=https://storage.googleapis.com.cnpmjs.org npm i --save puppeteer

To use Puppeteer in your project, run:

PUPPETEER_DOWNLOAD_HOST=https://storage.googleapis.com.cnpmjs.org npm i --save puppeteer

Install some basic packages

npm install

Usage

const puppeteer = require('puppeteer'),  fs = require('fs'),  path = require('path'),  request = require('request')function mkdirsSync(dirname) { if (fs.existsSync(dirname)) { return true } else { if (mkdirsSync(path.dirname(dirname))) {  fs.mkdirSync(dirname)  return true } }}var snapShotFolerPath = path.join(__dirname, '../snspshot/')mkdirsSync(snapShotFolerPath) function snapShot (taskInfo) { return new Promise(function (resolve, reject) { (async function(){  // 啟動Chromium  const browser = await puppeteer.launch({ignoreHTTPSErrors: true, headless:true, args: ['--no-sandbox']})  // 打開新頁面  const page = await browser.newPage()  // 設置頁面分辨率  await page.setViewport({width: 1920, height: 1080})  // 訪問  await page.goto(taskInfo.websiteUrl, {waitUntil: 'domcontentloaded'}).catch(err => console.log(err))  await page.waitFor(1000)  try {  // 截圖  await page.screenshot({path: snapShotFolerPath + taskInfo.imageName, fullPage:true}).catch(err => {   console.log('截圖失敗: ' + err)  });  await page.waitFor(6000)  } catch (e) {  console.log('failed ' + e)  } finally {  await browser.close()  fs.stat(snapShotFolerPath + taskInfo.imageName, function(err,stats){   if (err) {   reject('fail')   } else {                            if (stats.isFile()) {    resolve('success')   }   }  })  } })() })}module.exports = snapShot

如何安裝 (Linux 、Unix 操作系統)

如果你有翻墻環境

執行 npm install
執行 npm start

如果你沒有翻墻環境

打開 package.json 文件，檢查 dependencies 項目，如果 key 為 puppeteer 的條目，先刪除該條目。
進入工程命令行，輸入 PUPPETEER_DOWNLOAD_HOST=https://storage.googleapis.co... npm i --save puppeteer
執行 npm install 命令
執行 npm start

流程說明

while 循環去調用接口去獲取當前的截圖任務