mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-12-03 10:16:13 +00:00
8f424bf0d4
♻️ 重构插件系统 ⚙️ 重写插件 🎨 改进代码结构 📝 完善文档 Co-authored-by: zhxy-CN <admin@owo.cab> Co-authored-by: 洛水居室 <luoshuijs@outlook.com> Co-authored-by: xtaodada <xtao@xtaolink.cn> Co-authored-by: Li Chuangbo <im@chuangbo.li>
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
from typing import Optional
|
|
|
|
from playwright.async_api import Browser, Playwright, async_playwright
|
|
|
|
from core.service import Service
|
|
from utils.log import logger
|
|
|
|
|
|
class AioBrowser(Service):
|
|
|
|
def __init__(self, loop=None):
|
|
self.browser: Optional[Browser] = None
|
|
self._playwright: Optional[Playwright] = None
|
|
self._loop = loop
|
|
|
|
async def start(self):
|
|
if self._playwright is None:
|
|
logger.info("正在尝试启动 [blue]Playwright[/]")
|
|
self._playwright = await async_playwright().start()
|
|
logger.success("[blue]Playwright[/] 启动成功")
|
|
if self.browser is None:
|
|
logger.info("正在尝试启动 [blue]Browser[/]")
|
|
try:
|
|
self.browser = await self._playwright.chromium.launch(timeout=5000)
|
|
logger.success("[blue]Browser[/] 启动成功")
|
|
except TimeoutError as err:
|
|
logger.warning("[blue]Browser[/] 启动失败")
|
|
raise err
|
|
|
|
return self.browser
|
|
|
|
async def stop(self):
|
|
if self.browser is not None:
|
|
await self.browser.close()
|
|
if self._playwright is not None:
|
|
await self._playwright.stop()
|
|
|
|
async def get_browser(self) -> Browser:
|
|
if self.browser is None:
|
|
await self.start()
|
|
return self.browser
|