zhenxun_bot/bot.py

import contextlib
import platform

import nonebot

htmlrender_browser_channel = None
system = platform.system()

if system == "Windows":
    import winreg

    paths = {
        "chrome": r"SOFTWARE\Clients\StartMenuInternet\Google Chrome\DefaultIcon",
        "msedge": r"SOFTWARE\Clients\StartMenuInternet\Microsoft Edge\DefaultIcon",
    }
    for name, path in paths.items():
        with contextlib.suppress(FileNotFoundError):
            winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, path)
            htmlrender_browser_channel = name
            break

elif system == "Darwin":  # macOS
    from pathlib import Path

    mac_paths = {
        "chrome": "/Applications/Google Chrome.app",
        "msedge": "/Applications/Microsoft Edge.app",
    }
    for name, path in mac_paths.items():
        if Path(path).exists():
            htmlrender_browser_channel = name
            break

if htmlrender_browser_channel:
    nonebot.logger.info(
        f"使用 {htmlrender_browser_channel} 作为 htmlrender 驱动启动..."
    )


# from nonebot.adapters.discord import Adapter as DiscordAdapter
# from nonebot.adapters.dodo import Adapter as DoDoAdapter
# from nonebot.adapters.kaiheila import Adapter as KaiheilaAdapter
from nonebot.adapters.onebot.v11 import Adapter as OneBotV11Adapter

nonebot.init(htmlrender_browser_channel=htmlrender_browser_channel)


driver = nonebot.get_driver()
driver.register_adapter(OneBotV11Adapter)
# driver.register_adapter(KaiheilaAdapter)
# driver.register_adapter(DoDoAdapter)
# driver.register_adapter(DiscordAdapter)

from zhenxun.services.db_context import disconnect

# driver.on_startup(init)
driver.on_shutdown(disconnect)

# nonebot.load_builtin_plugins("echo")
nonebot.load_plugins("zhenxun/builtin_plugins")
nonebot.load_plugins("zhenxun/plugins")


if __name__ == "__main__":
    nonebot.run()