asyncio python 多线程 并发

330 篇文章 3 订阅
import asyncio
import time
from threading import Thread
from datetime import datetime

from pyppeteer import launch
from fine_bi.bi_config import bu_list

width, height = 1366, 768

url = 'http://fbcbi.efoxconn.com/decision/view/form?viewlet=%252FInventory_PRD%252FBU_Home_Page.frm&__parameters__=%257B%2522__pi__%2522%253Atrue%252C%2522bucode%2522%253A%2522CPB%2522%252C%2522rmbrate%2522%253A%2522M%2520RMB%2522%257D&_=1660202745718'

async def main(url,name):
    browser = await launch({'headless': False, 'args': ['--start-maximized']})
    page = await browser.newPage()
    await page.setViewport({'width': width, 'height': height})
    await page.goto(url)
    await page.waitForSelector('div[widgetname=TITLE_REPORT3]')
    await page.evaluate('''() =>{ document.querySelector('div[widgetname=FORM]').style='';
    document.querySelector('div[widgetname=BODY]').style=''; 
     }''')
    page_height = await page.evaluate('document.body.scrollHeight')
    print(name, 'page_height:', page_height)
    await page.setViewport({'width': width, 'height':page_height})
    await page.goto(url)
    await page.waitForSelector('div[widgetname=TITLE_REPORT3]')
    await page.evaluate('''() =>{ document.querySelector('div[widgetname=FORM]').style='';
        document.querySelector('div[widgetname=BODY]').style=''; 
         }''')
    time.sleep(3)
    # time.sleep(1000)
    image_root = r'E:\code\dock_project\fine_bi\images'
    image_path = image_root + '\\' + name + '.png'
    await page.screenshot({'path': image_path})
    await browser.close()

if __name__ == '__main__':
    start = datetime.now()
    tasks = []
    for bu in bu_list:
        print(bu['name'], bu['url'])
        name = bu['name'],
        print(name)
        # time.sleep(1000)
        url = bu['url']
        tasks.append(asyncio.ensure_future(main(bu['url'], bu['name'])))
    # tasks = [asyncio.ensure_future(main(url, '222')), asyncio.ensure_future(main(url, '333'))]

    loop = asyncio.get_event_loop()
    loop.run_until_complete(asyncio.wait(tasks))
    # asyncio.get_event_loop().run_until_complete(main(url, '222'))
    print('done')
    end = datetime.now()
    print('time:', end - start)

参考
https://baijiahao.baidu.com/s?id=1714015900283907482&wfr=spider&for=pc

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值