mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
29 lines
834 B
Python
Executable File
29 lines
834 B
Python
Executable File
|
|
from gino import Gino
|
|
from .log import logger
|
|
from utils.text_utils import prompt2cn
|
|
from configs.config import bind, sql_name, user, password, address, port, database
|
|
|
|
|
|
# 全局数据库连接对象
|
|
db = Gino()
|
|
|
|
|
|
async def init():
|
|
if not bind and (not user and not password and not address and not port and not database):
|
|
raise ValueError("\n" + prompt2cn("数据库配置未填写", 28))
|
|
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'数据库连接错误.... {type(e)}: {e}')
|
|
|
|
|
|
async def disconnect():
|
|
await db.pop_bind().close()
|
|
|