mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 00:32:06 +08:00
refactor code
This commit is contained in:
+3
-18
@@ -1,11 +1,8 @@
|
||||
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
|
||||
|
||||
|
||||
@@ -22,7 +19,7 @@ async def init(**kwargs) -> Optional[Browser]:
|
||||
_browser = await browser.chromium.launch(**kwargs)
|
||||
return _browser
|
||||
except NotImplementedError:
|
||||
logger.warning('win环境下 初始化playwright失败....请替换环境至linux')
|
||||
logger.warning("win环境下 初始化playwright失败....请替换环境至linux")
|
||||
return None
|
||||
|
||||
|
||||
@@ -36,18 +33,6 @@ def install():
|
||||
logger.info("正在检查 Chromium 更新")
|
||||
import sys
|
||||
from playwright.__main__ import main
|
||||
sys.argv = ['', 'install', 'chromium']
|
||||
|
||||
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
|
||||
|
||||
shutil.rmtree(dir)
|
||||
logger.info("已清理 Pyppeteer 依赖残留")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user