1. 网页截图

在这篇文章中,我们将研究如何利用Chrome的调试协议来加载网页并截图。通过一个名为的程序包chromedp,一切都可以实现,该程序包使我们可以通过Go代码控制Chrome实例。您还需要安装Chrome或使用类似于chrome/headless-shellDocker映像的工具。

我们将代码中的过程分为:

  • 启动Chrome
  • 运行任务:例如加载网页并截图
  • 将屏幕截图保存到文件
package main

import (
    "context"
    "io/ioutil"
    "log"

    "github.com/chromedp/cdproto/page"
    "github.com/chromedp/chromedp"
)

func main() {

    // Start Chrome
    // Remove the 2nd param if you don't need debug information logged
    ctx, cancel := chromedp.NewContext(context.Background(), chromedp.WithDebugf(log.Printf))
    defer cancel()

    url := "https://golangcode.com/"
    filename := "golangcode.png"

    // Run Tasks
    // List of actions to run in sequence (which also fills our image buffer)
    var imageBuf []byte
    if err := chromedp.Run(ctx, ScreenshotTasks(url, &imageBuf)); err != nil {
        log.Fatal(err)
    }

    // Write our image to file
    if err := ioutil.WriteFile(filename, imageBuf, 0644); err != nil {
        log.Fatal(err)
    }
}

func ScreenshotTasks(url string, imageBuf *[]byte) chromedp.Tasks {
    return chromedp.Tasks{
        chromedp.Navigate(url),
        chromedp.ActionFunc(func(ctx context.Context) (err error) {
            *imageBuf, err = page.CaptureScreenshot().WithQuality(90).Do(ctx)
            return err
        }),
    }
}

topgoer

另外,如果您想将页面另存为pdf而不是图像,则可以将CaptureScreenshot行替换为以下内容:

    *imageBuf, _, err = page.PrintToPDF().WithPrintBackground(false).Do(ctx)

results matching ""

    No results matching ""