mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
39 lines
947 B
Python
39 lines
947 B
Python
from typing import Optional
|
|
from nonebot.log import logger
|
|
from playwright.async_api import Browser, async_playwright
|
|
import nonebot
|
|
from nonebot import Driver
|
|
from services.log import logger
|
|
|
|
|
|
driver: Driver = nonebot.get_driver()
|
|
|
|
|
|
_browser: Optional[Browser] = None
|
|
|
|
|
|
async def init(**kwargs) -> Optional[Browser]:
|
|
# try:
|
|
global _browser
|
|
browser = await async_playwright().start()
|
|
_browser = await browser.chromium.launch(**kwargs)
|
|
return _browser
|
|
# except NotImplementedError:
|
|
# logger.warning("win环境下 初始化playwright失败....请替换环境至linux")
|
|
# return None
|
|
|
|
|
|
async def get_browser(**kwargs) -> Browser:
|
|
return _browser or await init(**kwargs)
|
|
|
|
|
|
@driver.on_startup
|
|
def install():
|
|
"""自动安装、更新 Chromium"""
|
|
logger.info("正在检查 Chromium 更新")
|
|
import sys
|
|
from playwright.__main__ import main
|
|
|
|
sys.argv = ["", "install", "chromium"]
|
|
main()
|