puppeteer實現(xiàn)html截圖的示例代碼

更新時間：2019年01月10日 08:59:17 作者：morethink

通過Puppeteer的提供的api直接控制Chrome模擬大部分用戶操作來進行UI Test或者作為爬蟲訪問頁面來收集數(shù)據(jù),這篇文章主要介紹了puppeteer實現(xiàn)截圖的示例代碼，感興趣的可以了解一下

puppeteer是谷歌官方出品的一個通過 DevTools 協(xié)議控制 headless Chrome 的Node庫?？梢酝ㄟ^Puppeteer的提供的api直接控制Chrome模擬大部分用戶操作來進行UI Test或者作為爬蟲訪問頁面來收集數(shù)據(jù)。

安裝

直接運行安裝命令：

npm install puppeteer

如果出現(xiàn)無法安裝的問題，可以使用淘寶鏡像。

puppeteer實現(xiàn)滑動截圖

在我 puppeteer 使用截全屏的過程中發(fā)現(xiàn)有些圖片無法截取到，而實際上是因為有些圖片是懶加載的，如果你沒有滑動到圖片的位置，那么這個圖片是不會加載。

現(xiàn)在我的方式是采用模擬瀏覽器滾動條滑動的方式滑動底部來使圖片加載出來。

代碼如下：

const puppeteer = require('puppeteer');

(async () => {
  const browser = await puppeteer.launch({
    headless: false
  });
  const page = await browser.newPage();
  await page.goto('https://www.cnblogs.com/morethink/p/6525216.html');
  await page.setViewport({
    width: 1200,
    height: 800
  });

  await autoScroll(page);

  await page.screenshot({
    path: '1.png',
    fullPage: true
  });

  await browser.close();
})();


function autoScroll(page) {
  return page.evaluate(() => {
    return new Promise((resolve, reject) => {
      var totalHeight = 0;
      var distance = 100;
      var timer = setInterval(() => {
        var scrollHeight = document.body.scrollHeight;
        window.scrollBy(0, distance);
        totalHeight += distance;
        if (totalHeight >= scrollHeight) {
          clearInterval(timer);
          resolve();
        }
      }, 100);
    })
  });
}

動圖如下：

puppeteer 實現(xiàn) html element 截圖

在某些情況下我們只想要針對html的某個位置進行截圖而不是針對頁面截全屏。

puppeteer提供了ElementHandle.screenshot 方法，該方法參數(shù)和page.screenshot 一樣。而ElementHandle 對象是頁面內(nèi)的Dom對象?？梢詭椭覍?html element進行截圖。這樣的話你想截取頁面的哪部分就截取頁面的哪部分。

代碼如下：

const puppeteer = require('puppeteer');

(async () => {
  const browser = await puppeteer.launch({
    headless: false
  });
  const page = await browser.newPage();
  await page.goto('https://www.cnblogs.com/morethink/p/6525216.html');
  await page.setViewport({
    width: 1200,
    height: 800
  });
  //獲取頁面Dom對象
  let body = await page.$('#cnblogs_post_body');
  //調(diào)用頁面內(nèi)Dom對象的 screenshot 方法進行截圖
  await body.screenshot({
    path: '2.png'
  });
  await browser.close();
})();

參考文檔：

https://github.com/GoogleChrome/puppeteer/blob/v1.11.0/docs/api.md#elementhandlescreenshotoptions

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。