在go语言中使用playwright-go实现浏览器网页截图功能代码示例

Go  /  管理员 发布于 1年前   641

Playwright是一个用于自动化浏览器操作的开源工具集。它由微软开发并于2020年发布，旨在提供一种跨浏览器、跨平台的解决方案，可用于测试Web应用程序、编写爬虫、执行自动化任务等。

Playwright支持多种主流浏览器，包括Chrome、Firefox、Safari和Edge，它提供了一组简单易用的API，可以模拟用户与Web页面的交互行为，例如点击、填写表单、导航等。与其他类似工具相比，Playwright的一个重要特点是它的跨浏览器支持，这意味着你可以使用相同的代码在不同浏览器上运行你的自动化任务，而不需要为每个浏览器单独编写代码。

Playwright还提供了强大的调试功能，可以帮助开发人员在自动化过程中定位和解决问题。

它支持截图和录制操作，使得调试变得更加直观和高效。

另外，Playwright还具有一些高级功能，例如可以模拟不同的设备、网络环境和地理位置，以及支持并发执行多个浏览器实例等，

这些功能使得它在编写复杂的自动化任务时非常有用。

下面看看使用go语言进行开发，以playwright的go模块playwright-go的安装使用。

安装

go get -u github.com/playwright-community/playwright-go

安装相关浏览器和操作系统依赖项：

go run github.com/playwright-community/playwright-go/cmd/playwright@latest install --with-deps
# Or
go install github.com/playwright-community/playwright-go/cmd/playwright@latest
playwright install --with-deps

也可以在代码中使用以下代码安装: err := playwright.Install()

通过安装截图可以看出安装了3大主流浏览器和ffmpeg。

官方实例：

接下来我们看一个官方的例子,从Hacker News中抓取当前投票最高的项目。

package main

import (
    "fmt"
    "log"
    "github.com/playwright-community/playwright-go"
)

func main() {
    pw, err := playwright.Run()
    if err != nil {
        log.Fatalf("could not start playwright: %v", err)
    }
    browser, err := pw.Chromium.Launch()
    if err != nil {
        log.Fatalf("could not launch browser: %v", err)
    }
    page, err := browser.NewPage()
    if err != nil {
        log.Fatalf("could not create page: %v", err)
    }
    if _, err = page.Goto("https://news.ycombinator.com"); err != nil {
        log.Fatalf("could not goto: %v", err)
    }
    entries, err := page.Locator(".athing").All()
    if err != nil {
        log.Fatalf("could not get entries: %v", err)
    }
    for i, entry := range entries {
        title, err := entry.Locator("td.title > span > a").TextContent()
    if err != nil {
        log.Fatalf("could not get text content: %v", err)
    }
    fmt.Printf("%d: %s\n", i+1, title)
    }
    if err = browser.Close(); err != nil {
        log.Fatalf("could not close browser: %v", err)
    }
    if err = pw.Stop(); err != nil {
        log.Fatalf("could not stop Playwright: %v", err)
    }
}

运行代码后，输出:

可以看到当前排名最高的30个项目成功输出了。

接下来介绍如何进行网页截图。

截取网站地图页面的截图-示例代码

接下来我们根据 www.zonsgcan.com 的网站地图获取每个页面的截图。

package main

import (
    "encoding/xml"
    "fmt"
    "github.com/playwright-community/playwright-go"
    "io"
    "log"
    "net/http"
    "time"
)

type Urlset struct {
    XMLName xml.Name `xml:"urlset"`
    Xmlns   string   `xml:"xmlns,attr"`
    Urls []Url `xml:"url"`
}

type Url struct {
    Loc     string `xml:"loc"`
    Lastmod string `xml:"lastmod"`
}

func main() {
    sites := formatSiteXml("https://www.zonsgcan.com/sitemap.xml")
    pw, err := playwright.Run()
    if err != nil {
        log.Fatalf("could not launch playwright: %v", err)
    }
    browser, err := pw.Chromium.Launch()
    if err != nil {
        log.Fatalf("could not launch Chromium: %v", err)
    }
    page, err := browser.NewPage()
    if err != nil {
        log.Fatalf("could not create page: %v", err)
    }
    for _, site := range sites {
    if _, err = page.Goto(site, playwright.PageGotoOptions{
        WaitUntil: playwright.WaitUntilStateDomcontentloaded,
        }); err != nil {
        log.Fatalf("could not goto: %v", err)
    }
    if _, err = page.Screenshot(playwright.PageScreenshotOptions{
        Path: playwright.String(fmt.Sprintf("./img/%d.png", time.Now().Unix())),
        }); err != nil {
            log.Fatalf("could not create screenshot: %v", err)
        }
    }
    if err = browser.Close(); err != nil {
        log.Fatalf("could not close browser: %v", err)
    }
    if err = pw.Stop(); err != nil {
        log.Fatalf("could not stop Playwright: %v", err)
    }
}

func formatSiteXml(sitemapURL string) []string {
    resp, err := http.Get(sitemapURL)
    if err != nil {
        log.Fatal(err)
    }
    defer resp.Body.Close()
    bytes, err := io.ReadAll(resp.Body)
    if err != nil {
        log.Fatal(err)
    }
    var urlset Urlset
    err = xml.Unmarshal(bytes, &urlset)
    if err != nil {
        log.Fatal(err)
    }
    urls := make([]string, 0)
    for _, url := range urlset.Urls {
        urls = append(urls, url.Loc)
    }
    return urls
}

ps：

这段代码主要是先获取本站的所有页面，然后对每个页面进行截图。

代码运行完，查看截图时发现截图的大小不是完整的页面内容，只有部分可视大小，

查看源码发现需要设置 FullPage 参数，才能截取完整页面。

相关链接

https://github.com/playwright-community/playwright-go
https://playwright.dev/docs/intro