main.py

import pkgutil
from pathlib import Path

import kayaku
from creart import create

# from graia.amnesia.builtins.uvicorn import UvicornService
from graia.ariadne.app import Ariadne
from graia.ariadne.connection.config import (
    HttpClientConfig,
    WebsocketClientConfig,
    config,
)
from graia.ariadne.model import LogConfig
from graia.saya import Saya
from graiax.playwright import PlaywrightService

kayaku.initialize({"{**}": "./config/{**}"})

from libs import log_level_handler, replace_logger  # noqa: E402
from libs.config import basic_cfg, modules_cfg  # noqa: E402

# from libs.fastapi_service import FastAPIStarletteService  # noqa: E402
from libs.database_services import DatabaseInitService  # noqa: E402
from libs.send_action import Safe  # noqa: E402
from static.path import modules_path, root_path  # noqa: E402

kayaku.bootstrap()

# loop = create(AbstractEventLoop)  # 若不需要 loop.run_until_complete()，则不需要此行
# 若 create Saya 则可以省掉 bcc 和 scheduler 的
# sche = create(GraiaScheduler)
# bcc = create(Broadcast)
saya = create(Saya)

Ariadne.options['installed_log'] = True
app = Ariadne(
    connection=config(
        basic_cfg.miraiApiHttp.account,  # 你的机器人的 qq 号
        basic_cfg.miraiApiHttp.verifyKey,  # 填入 verifyKey
        HttpClientConfig(host=basic_cfg.miraiApiHttp.host),
        WebsocketClientConfig(host=basic_cfg.miraiApiHttp.host),
    ),
    log_config=LogConfig(log_level_handler),
)
app.default_send_action = Safe

replace_logger(level=0 if basic_cfg.debug else 20, richuru=False)

ignore = ('__init__.py', '__pycache__')

with saya.module_context():
    core_modules_path = Path(root_path, 'core_modules')
    for module in pkgutil.iter_modules([str(core_modules_path)]):
        if module.name in ignore or module.name[0] in ('#', '.', '_'):
            continue
        saya.require(f'core_modules.{module.name}')

if modules_cfg.enabled:
    with saya.module_context():
        for module in pkgutil.iter_modules([str(modules_path)]):
            if module.name in ignore or module.name[0] in ('#', '.', '_'):
                continue
            saya.require(f'modules.{module.name}')

if basic_cfg.miraiApiHttp.account == 123456789:
    raise ValueError('在?¿ 填一下配置文件？')

app.launch_manager.add_service(DatabaseInitService())
# app.launch_manager.add_service(FastAPIStarletteService())
# app.launch_manager.add_service(UvicornService())
app.launch_manager.add_service(PlaywrightService())

Ariadne.launch_blocking()
kayaku.save_all()