mirror of
https://github.com/TriM-Organization/LiteyukiBot-TriM.git
synced 2024-11-11 09:37:29 +08:00
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)
|