mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-07-04 17:20:19 +08:00
Alex Ott
40 lines
998 B
Python
40 lines
998 B
Python
# -*- coding: utf-8 -*-
|
|
"""Bot logic."""
|
|
import aiosqlite
|
|
import hikari
|
|
import lightbulb
|
|
import miru
|
|
from bot.api_client import OasstApiClient
|
|
from bot.config import Config
|
|
|
|
config = Config.from_env()
|
|
|
|
bot = lightbulb.BotApp(
|
|
token=config.token,
|
|
logs="DEBUG",
|
|
prefix=config.prefix,
|
|
default_enabled_guilds=config.declare_global_commands,
|
|
owner_ids=config.owner_ids,
|
|
intents=hikari.Intents.ALL,
|
|
)
|
|
|
|
|
|
@bot.listen()
|
|
async def on_starting(event: hikari.StartingEvent):
|
|
"""Setup."""
|
|
miru.install(bot) # component handler
|
|
bot.load_extensions_from("./bot/extensions") # load extensions
|
|
|
|
bot.d.db = await aiosqlite.connect("./bot/db/database.db")
|
|
await bot.d.db.executescript(open("./bot/db/schema.sql").read())
|
|
await bot.d.db.commit()
|
|
|
|
bot.d.oasst_api = OasstApiClient("http://localhost:8080", "any_key")
|
|
|
|
|
|
@bot.listen()
|
|
async def on_stopping(event: hikari.StoppingEvent):
|
|
"""Cleanup."""
|
|
await bot.d.db.close()
|
|
await bot.d.oasst_api.close()
|