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 依赖残留")
|
|||
|
|
|