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

相关推荐
kyriewen29 分钟前
别再乱用useEffect了——你写的10个里有8个不该存在
前端·javascript·react.js
Ivanqhz40 分钟前
Rust &‘static str浅析
java·前端·javascript·rust
IT_陈寒1 小时前
SpringBoot这个分页坑,我踩了三天才爬出来
前端·人工智能·后端
颜酱1 小时前
05 | 召回前置准备:根据业务数据库生成各数据库(读取配置阶段)
前端·人工智能·后端
zandy10112 小时前
衡石 Agentic BI的ReAct 推理框架在 Agentic BI 中的工程化实践
前端·javascript·react.js
罗超驿2 小时前
JavaEE进阶之路:从Web架构原理到HTML标签全解析
前端·html·web·javaee
西安小哥2 小时前
从前端到AI工程师:一场跨越鸿沟的真实蜕变之旅
前端
Prince4182 小时前
侧边栏收起缩放适配方案
前端
用户7783366132112 小时前
serpbase + Cloudflare R2 边缘持久化实战
前端·人工智能
এ慕ོ冬℘゜2 小时前
jQuery attr() 方法超详细讲解:属性获取、赋值、实战踩坑全解
前端·javascript·jquery