Как решить такую проблему с pyppeteer, asyncio и pyrogram?

Question

[Вячеслав Грачунов]

Есть событие появления нового сообщения, из него надо вызвать скриншотер на pyppeteer в цикле
Выглядит так:

....
async def screenshot(stock, loop, date_msg):
    await launch(headless=True, defaultViewport=None, args=['--no-sandbox'])
    .....
....

@client.on_message()
def msg_handler(client, message):
    .......
   for row in result:
       loop = asyncio.get_event_loop()
       loop.run_until_complete(screenshot(stock.upper(), loop, date_msg))

    screens.append(InputMediaPhoto(os.path.join(BASEDIR, stock.upper()+".png")))

В таком виде получаю такую ошибку:

RuntimeError: There is no current event loop in thread 'Handler_0'

Т.к. пытаюсь запускать не из главного потока

loop1 = asyncio.new_event_loop()
asyncio.set_event_loop(loop1)
loop = asyncio.get_event_loop()
loop.run_until_complete(screenshot(stock.upper(), loop, date_msg))

Вот в таком получаю следующее:
signal only works in main thread
Это уже от pyppeteer
Но если отключаю сигналы:

await launch(headless=True, defaultViewport=None, args=['--no-sandbox'], handleSIGINT=False, handleSIGTERM=False, handleSIGHUP=False)

То loop.run_until_complete просто не завершается и висит.
Что с этим делать?

Answer 1

[Вячеслав Грачунов]

Короче решил через multiprocessing - обернул в отдельный процесс и соответственно внутри этого процесса поток является главным и все работает

Comments

[Вячеслав Грачунов]

bogdanov1ch, вечером пришлю

[Вячеслав Грачунов]

bogdanov1ch,

....
from multiprocessing import Process, Queue
....




....
async def screenshot(stock, loop, date_msg):
    await launch(headless=True, defaultViewport=None, args=['--no-sandbox'])
     .....
    await page.screenshot({'path': os.path.join(BASEDIR, stock+".png")})
    .....

    if (....):
    	return data
    else:
    	return 0
....



loop = asyncio.get_event_loop()

@client.on_message()
def msg_handler(client, message):

	global loop
	....

	if message.caption is not None:

			result = re.findall(r'\w{1,6}\s+\w{1}\s+\d+\,*\d+\s+\d+\,*\d+\s+\d+\,*\d+', message.caption)
			if len(result):

				def make_screens(queue, result):
					....
					screens = []
					for row in result:
						date_msg = .......
						stock = .......
						....

						loop1 = asyncio.new_event_loop()
						asyncio.set_event_loop(loop1)
						loop = asyncio.get_event_loop()
						data = loop.run_until_complete(screenshot(stock.upper(), loop, date_msg))

						if data:
							screens.append(InputMediaPhoto(os.path.join(BASEDIR, stock.upper()+".png")))

					.......

					queue.put(screens)

				queue = Queue()
				p = Process(target=make_screens, args=(queue, result))
				p.start()
				p.join()
				screens = queue.get()