zhenxun_bot/services/db_context.py

from gino import Gino
from .log import logger
from configs.config import bind, sql_name, user, password, address, port, database


# 全局数据库连接对象
db = Gino()


async def init():
    i_bind = bind
    if not i_bind:
        i_bind = f"{sql_name}://{user}:{password}@{address}:{port}/{database}"
    try:
        await db.set_bind(i_bind)
        await db.gino.create_all()
        logger.info(f'Database loaded successfully!')
    except Exception as e:
        raise Exception(f'数据库连接错误.... e: {e}')


async def disconnect():
    await db.pop_bind().close()