bot/app
3
6.9k
Fork 2
mirror of https://github.com/LiteyukiStudio/LiteyukiBot.git synced 2025-07-27 19:10:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

📝 使用 driver_manager 自动管理启用的驱动器，无需手动配置环境变量 (当配置了环境变量时环境变量优先)

Browse Source
This commit is contained in:
Expliyh
2024-05-17 16:33:48 +08:00
parent a72eeb4c3f
commit 6509b293db
4 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
liteyuki/utils/driver_manager/auto_set_env.py Normal file
View File

@ -0,0 +1,20 @@
import os
import dotenv
import nonebot
from .defines import *
def auto_set_env(config: dict):
dotenv.load_dotenv()
if os.getenv("DRIVER", None) is not None:
nonebot.logger.info("Driver already set in environment variable, skip auto configure.")
return
if config.get("enable_satori", False):
os.environ["DRIVER"] = get_driver_string(ASGI_DRIVER, HTTPX_DRIVER, WEBSOCKETS_DRIVER)
nonebot.logger.info("Enable Satori, set driver to ASGI+HTTPX+WEBSOCKETS")
else:
os.environ["DRIVER"] = get_driver_string(ASGI_DRIVER)
nonebot.logger.info("Disable Satori, set driver to ASGI")
return