快捷導航

Python使用pyppeteer進行網(wǎng)頁截圖并發(fā)送機器人實例

更新時間：2024年04月12日 09:45:57 作者：Meepoljd

這篇文章主要介紹了Python使用pyppeteer進行網(wǎng)頁截圖并發(fā)送機器人實例,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教

腳本截圖

使用pyppeteer其實和我們自己開瀏覽器的邏輯差不多，同時要結(jié)合async一起使用，首先要通過lauch函數(shù)啟動一個瀏覽器，啟動瀏覽器的時候可以設(shè)置啟動參數(shù)，其中包括使用的代理服務(wù)器：

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

啟動瀏覽器后我們要開一個新的標簽頁來訪問網(wǎng)頁，goto就是用來訪問對應的網(wǎng)頁的：

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

現(xiàn)在我們就開始訪問對應的網(wǎng)頁了，根據(jù)經(jīng)驗我們都知道有的網(wǎng)頁剛開啟的時候不是完全渲染完的，所以可以添加一個asyncio.sleep(10)來等待十秒鐘:

await asyncio.sleep(10)

如果需要運行一些js腳本語句，可以通過evaluate執(zhí)行：

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

然后可以使用字典訪問對應的數(shù)據(jù)：

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()

發(fā)送到機器人

截圖和pdf都生成了，現(xiàn)在都可以發(fā)送給企業(yè)微信機器人了

發(fā)送圖片

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)

發(fā)送PDF

發(fā)送PDF按照教程需要先調(diào)用upload_media接口，然后在發(fā)送圖片的時候把media_id作為參數(shù)傳入

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)

發(fā)送效果如下：