fix stupid bug

This commit is contained in:
felinae98 2021-06-01 14:22:33 +08:00
parent d447563a1f
commit ae934b7e2c
No known key found for this signature in database
GPG Key ID: 00C8B010587FF610

View File

@ -2,7 +2,6 @@ import asyncio
from html import escape from html import escape
import os import os
from tempfile import NamedTemporaryFile from tempfile import NamedTemporaryFile
from time import asctime
from typing import Awaitable, Callable, Optional from typing import Awaitable, Callable, Optional
from pyppeteer import launch from pyppeteer import launch
@ -41,9 +40,9 @@ class Render(metaclass=Singleton):
retry_times += 1 retry_times += 1
logger.warning("render error {}\n".format(retry_times) + self.interval_log) logger.warning("render error {}\n".format(retry_times) + self.interval_log)
self.interval_log = '' self.interval_log = ''
if self.browser: # if self.browser:
await self.browser.close() # await self.browser.close()
self.lock.release() # self.lock.release()
def _inter_log(self, message: str) -> None: def _inter_log(self, message: str) -> None:
# self.interval_log += asctime() + '' + message + '\n' # self.interval_log += asctime() + '' + message + '\n'
@ -65,7 +64,7 @@ class Render(metaclass=Singleton):
self._inter_log('open page') self._inter_log('open page')
if viewport: if viewport:
await page.setViewport(viewport) await page.setViewport(viewport)
logger._inter_log('set viewport') self._inter_log('set viewport')
if target: if target:
target_ele = await page.querySelector(target) target_ele = await page.querySelector(target)
data = await target_ele.screenshot(type='jpeg', encoding='base64') data = await target_ele.screenshot(type='jpeg', encoding='base64')