mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
from typing import Optional
|
||
import shutil
|
||
from nonebot.log import logger
|
||
from playwright.async_api import Browser, async_playwright
|
||
import nonebot
|
||
from nonebot import Driver
|
||
from appdirs import AppDirs
|
||
from pathlib import Path
|
||
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 NotImplemented:
|
||
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()
|
||
|
||
|
||
@driver.on_startup
|
||
def delete_pyppeteer():
|
||
"""删除 Pyppeteer 遗留的 Chromium"""
|
||
|
||
dir = Path(AppDirs('pyppeteer').user_data_dir)
|
||
if not dir.exists():
|
||
return
|
||
|
||
# if not config.haruka_delete_pyppeteer:
|
||
# logger.info("检测到 Pyppeteer 依赖（约 300 M)，"
|
||
# "新版 HarukaBot 已经不需要这些文件了。"
|
||
# "如果没有其他程序依赖 Pyppeteer，请在 '.env.*' 中设置"
|
||
# " 'HARUKA_DELETE_PYPPETEER=True' 并重启 Bot 后，将自动清除残留")
|
||
# else:
|
||
shutil.rmtree(dir)
|
||
logger.info("已清理 Pyppeteer 依赖残留")
|
||
|