Python使用pyppeteer进行网页截图并发送机器人实例

 更新时间:2024年04月12日 09:45:57   作者:Meepoljd  
这篇文章主要介绍了Python使用pyppeteer进行网页截图并发送机器人实例,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

Pyppeteer是对Puppeteer的一个Python封装,常用在爬虫方面,最近使用它做网页巡检报告的截图,记录一下

脚本截图

使用pyppeteer其实和我们自己开浏览器的逻辑差不多,同时要结合async一起使用,首先要通过lauch函数启动一个浏览器,启动浏览器的时候可以设置启动参数,其中包括使用的代理服务器:

browser = await launch(args=["--no-sandbox", "--proxy-server=http://10.1.1.1:8443"])

启动浏览器后我们要开一个新的标签页来访问网页,goto就是用来访问对应的网页的:

page = await browser.newPage()
await page.goto(url)

现在我们就开始访问对应的网页了,根据经验我们都知道有的网页刚开启的时候不是完全渲染完的,所以可以添加一个asyncio.sleep(10)来等待十秒钟:

await asyncio.sleep(10)

如果需要运行一些js脚本语句,可以通过evaluate执行:

dimensions = await page.evaluate('''() => {
    return {
        width: document.documentElement.scrollWidth,
        height: document.documentElement.scrollHeight,
    }
}''')

然后可以使用字典访问对应的数据:

dimensions['height']

接下来就是截图的操作了,甚至可以同时截成pdf,完整代码如下:

async def main():
    browser = await launch(args=["--no-sandbox", "--proxy-server=http://10.1.1.1:8443"])
    # 新建选项卡
    page = await browser.newPage()
    await page.goto(url)

    await asyncio.sleep(10)
    dimensions = await page.evaluate('''() => {
       return {
           width: document.documentElement.scrollWidth,
           height: document.documentElement.scrollHeight,
       }
    }''')
    await page.setViewport({'width': 1920, 'height': dimensions['height']})
    await page.screenshot({'path': "test.png", "clip": {"x": 300, "y": 10, "width": 1320, "height": dimensions['height']}})

    await page.pdf({'path': file,  'width': 1920, 'height': dimensions['height']})
    await browser.close()

发送到机器人

截图和pdf都生成了,现在都可以发送给企业微信机器人了

发送图片

def post_img_to_robot(page):
    with open(page, 'rb') as file:
        data = file.read()
        encodestr = base64.b64encode(data)
        image_data = str(encodestr, 'utf-8')
    with open(page, 'rb') as file:
        md = hashlib.md5()
        md.update(file.read())
        image_md5 = md.hexdigest()
    data = {
        "msgtype": "image",
        "image": {
            "base64": image_data,
            "md5": image_md5
        }
    }
    headers = {'content-type': 'application/json;charset=utf-8'}
    body = json.dumps(data)
    requests.post(qy_ex, data=body, headers=headers)

发送PDF

发送PDF按照教程需要先调用upload_media接口,然后在发送图片的时候把media_id作为参数传入

def post_file_to_robot(file):
    parsed_url = urlparse(qy_ex)
    param = parse_qs(parsed_url.query)
    webHookKey = param["key"][0]
    upload_url = f'https://qyapi.weixin.qq.com/cgi-bin/webhook/upload_media?key={webHookKey}&type=file'
    headers = {
        "Accept": "application/json, text/plain, */*", "Accept-Encoding": "gzip, deflate",
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/80.0.3987.100 Safari/537.36"
    }
    data = {'file': open(file, 'rb')}
    req = requests.post(url=upload_url, files=data, headers=headers).json()
    media_id = req["media_id"]

    data = {
        "msgtype": "file",
        "file": {"media_id": media_id}
    }

    requests.post(url=qy_ex, json=data)

发送效果如下:

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

相关文章

  • Python实现轻松比较两个文件夹的多种方法

    Python实现轻松比较两个文件夹的多种方法

    在处理数据集时,我们经常会遇到两个文件夹中文件数目不一致的情况,本文我们就来使用Python比较一下两个文件夹,轻松实现找出多出的文件,有需要的可以了解下
    2025-09-09
  • Python简单I/O操作示例

    Python简单I/O操作示例

    这篇文章主要介绍了Python简单I/O操作,结合实例形式分析了Python针对文件的I/O读写及cPickle模块相关使用操作技巧,需要的朋友可以参考下
    2019-03-03
  • Python面向对象的内置方法梳理讲解

    Python面向对象的内置方法梳理讲解

    面向对象编程是一种编程方式,此编程方式的落地需要使用“类”和 “对象”来实现,所以,面向对象编程其实就是对 “类”和“对象” 的使用,今天给大家介绍下python 面向对象开发及基本特征,感兴趣的朋友一起看看吧
    2022-10-10
  • python进程间数据交互的几种实现方式

    python进程间数据交互的几种实现方式

    本文主要介绍了python进程数据交互的几种实现方式,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2023-05-05
  • Numpy中vstack()和hstack()的使用方式

    Numpy中vstack()和hstack()的使用方式

    文章详细介绍了Numpy中vstack()和hstack()两种数组拼接方法,vstack()是垂直堆叠数组,需保证除第一轴(行)外,数组形状相同,hstack()是按顺序堆叠数组,需保证除第二轴(列)外,数组形状相同,两者都可通过堆叠给定数组最后形成至少二维的数组
    2024-10-10
  • python黑魔法之参数传递

    python黑魔法之参数传递

    这篇文章主要介绍了python黑魔法之参数传递,分析了python参数传递的方法,感兴趣的小伙伴们可以参考一下
    2016-02-02
  • Python中相见恨晚的技巧(记得收藏)

    Python中相见恨晚的技巧(记得收藏)

    这篇文章主要介绍了一些Python中相见恨晚的使用技巧,帮助大家更好的理解和学习使用python,感兴趣的朋友可以了解下
    2021-04-04
  • python一键升级所有pip package的方法

    python一键升级所有pip package的方法

    下面小编就为大家带来一篇python一键升级所有pip package的方法。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2017-01-01
  • 分享python机器学习中应用所产生的聚类数据集方法

    分享python机器学习中应用所产生的聚类数据集方法

    本文根据 机器学习中常用的聚类数据集生成方法 中的内容进行编辑实验和整理而得,有需要的朋友可以参考想,希望可以对大家在聚类数据方面有所帮助
    2021-08-08
  • python抓取网站的图片并下载到本地的方法

    python抓取网站的图片并下载到本地的方法

    今天小编就为大家分享一篇python抓取网站的图片并下载到本地的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-05-05

最新评论