使用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

相关推荐
会豪25 分钟前
Electron-Vite (一)快速构建桌面应用
前端
中微子28 分钟前
React 执行阶段与渲染机制详解(基于 React 18+ 官方文档)
前端
唐某人丶29 分钟前
教你如何用 JS 实现 Agent 系统(2)—— 开发 ReAct 版本的“深度搜索”
前端·人工智能·aigc
中微子30 分钟前
深入剖析 useState产生的 setState的完整执行流程
前端
遂心_1 小时前
JavaScript 函数参数传递机制:一道经典面试题解析
前端·javascript
小徐_23331 小时前
uni-app vue3 也能使用 Echarts?Wot Starter 是这样做的!
前端·uni-app·echarts
RoyLin1 小时前
TypeScript设计模式:适配器模式
前端·后端·node.js
遂心_2 小时前
深入理解 React Hook:useEffect 完全指南
前端·javascript·react.js
Moonbit2 小时前
MoonBit 正式加入 WebAssembly Component Model 官方文档 !
前端·后端·编程语言
龙在天2 小时前
ts中的函数重载
前端