使用node调用chrome(基于selenium-webdriver包)

下载测试版chrome和chromedriver

https://googlechromelabs.github.io/chrome-for-testing/

把chromedriver复制到chrome的文件里

设置环境变量

编写代码

javascript 复制代码
const { Builder, Browser, By, Key, until } = require('selenium-webdriver')
const puppeteer = require('puppeteer')

//查询
const chromiumSearch = async (req, res) => {
  const { type = '' } = req.body

  if (type === 'puppeteer') {
    async function pageScreenshot(url, path = 'screenshot.png') {
      const browser = await puppeteer.launch()
      const page = await browser.newPage()
      await page.goto(url)
      await page.screenshot({ path })
      await browser.close()
    }

    // 使用函数将指定网页转换为图片
    pageScreenshot('https://www.baidu.com', '/temp/ai/puppeter/example.png')

    res.send({
      code: 200,
      data: {},
      message: '成功',
    })
  } else if (type === 'selenium-webdriver') {
    let driver = await new Builder().forBrowser(Browser.CHROME).build()

    try {
      //await driver.get('https://www.baidu.com')
      await driver.get('http://localhost:85/test1.html')

      // await driver.findElement(By.name('q')).sendKeys('webdriver', Key.RETURN)
      // await driver.wait(until.titleIs('webdriver - Google Search'), 3000)
      // await driver.wait(until.titleIs('webdriver - Google Search'), 1000)

    // 等待页面加载完成
    // await driver.wait(until.titleIs('Example Domain'), 10000);
 
    // 滚动到页面顶部
    await driver.executeScript('window.scrollTo(0, 0);');
 
    // 等待一会,确保页面加载和渲染完成
    await new Promise(resolve => setTimeout(resolve, 3000));
 
    // 获取页面截图并保存为图片
    const screenshot = await driver.takeScreenshot();
    const base64Data = screenshot.replace(/^data:image\/png;base64,/, '');
    const fs = require('fs');
    let now = Date.now()
    fs.writeFileSync(`/temp/ai/chromium/${now}.png`, base64Data, 'base64');
 
    console.log('网页已转换成图片: screenshot.png');


    } finally {
      await driver.quit()
    }

    res.send({
      code: 200,
      data: {},
      message: '成功',
    })
  } else {
    res.send({
      code: 400,
      data: {},
      message: '参数错误:type',
    })
  }
}

module.exports = {
  chromiumSearch,
}

参考链接:

https://blog.csdn.net/qq_52867389/article/details/132197894

人工智能学习网站

https://chat.xutongbao.top

相关推荐
奇奇怪怪的11 分钟前
Embedding 模型 10+ 横向评测
前端
陈广亮14 分钟前
Monorepo 从 0 到 1 实操指南 2026 版:pnpm catalogs + Turborepo 2.x + changesets 全链路
前端
子兮曰14 分钟前
OpenMontage 深度解剖:你的 AI 编程助手,其实是个视频工作室
前端·后端·ai编程
敲代码的鱼15 分钟前
PDF 预览与签名批注写回 支持安卓 iOS 鸿蒙 UTS插件
android·前端·ios
子兮曰22 分钟前
前端工具链的「Rust 化」:一场没有赢家的军备竞赛?
前端·后端·rust
Hyyy1 小时前
Function Calling / Tool Use的原理和实现模式
前端·llm·ai编程
爱勇宝1 小时前
从 Ctrl+CV 到 Enter:程序员正在失去什么
前端·后端·程序员
徐小夕2 小时前
我们开源了一款“框架无关”的思维导图编辑器,3分钟集成到任意系统
前端·javascript·github
PBitW2 小时前
GPT训练我的第三天,明白了应该咋说满分回答!😕😕😕
前端·javascript·面试
摸着石头过河的石头2 小时前
前端多仓库管理:从混乱到有序的进化之路
前端