forked from bot/app
27 lines
886 B
Python
27 lines
886 B
Python
import json
|
|
import os
|
|
|
|
import nonebot
|
|
from nonebot.adapters import satori
|
|
|
|
|
|
def init(config: dict):
|
|
if config.get("satori", None) is None:
|
|
nonebot.logger.info("Satori config not found, skip Satori init.")
|
|
return None
|
|
satori_config = config.get("satori")
|
|
if not satori_config.get("enable", False):
|
|
nonebot.logger.info("Satori not enabled, skip Satori init.")
|
|
return None
|
|
if os.getenv("SATORI_CLIENTS", None) is not None:
|
|
nonebot.logger.info("Satori clients already set in environment variable, skip.")
|
|
os.environ["SATORI_CLIENTS"] = json.dumps(satori_config.get("hosts", []), ensure_ascii=False)
|
|
config['satori_clients'] = satori_config.get("hosts", [])
|
|
return
|
|
|
|
|
|
def register():
|
|
if os.getenv("SATORI_CLIENTS", None) is not None:
|
|
driver = nonebot.get_driver()
|
|
driver.register_adapter(satori.Adapter)
|