25 lines
591 B
Python
25 lines
591 B
Python
|
# database/init_db.py
|
||
|
|
import asyncio
|
||
|
|
from database.database import db_manager
|
||
|
|
from utils.logger import get_logger
|
||
|
|
|
||
|
|
logger = get_logger(__name__)
|
||
|
|
|
||
|
|
async def init_database():
|
||
|
|
"""初始化数据库"""
|
||
|
|
try:
|
||
|
|
# 连接数据库
|
||
|
|
await db_manager.connect()
|
||
|
|
|
||
|
|
# 创建表
|
||
|
|
await db_manager.create_tables()
|
||
|
|
|
||
|
|
logger.info("数据库初始化完成")
|
||
|
|
return True
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"数据库初始化失败: {e}")
|
||
|
|
return False
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
asyncio.run(init_database())
|