diff --git a/bot.py b/bot.py index 5342be7ce..65e420c7f 100644 --- a/bot.py +++ b/bot.py @@ -200,6 +200,35 @@ def raw_main(): logger.info("检查EULA和隐私条款完成") easter_egg() + + # 在此处初始化数据库 + from src.config.config import global_config + from src.common.database.database import initialize_sql_database + from src.common.database.sqlalchemy_models import initialize_database as init_db + from src.common.database.db_migration import check_and_migrate_database + + logger.info("正在初始化数据库连接...") + try: + initialize_sql_database(global_config.database) + logger.info(f"数据库连接初始化成功,使用 {global_config.database.database_type} 数据库") + except Exception as e: + logger.error(f"数据库连接初始化失败: {e}") + raise e + + logger.info("正在初始化数据库表结构...") + try: + init_db() + logger.info("数据库表结构初始化完成") + except Exception as e: + logger.error(f"数据库表结构初始化失败: {e}") + raise e + + # 执行数据库自动迁移检查 + try: + check_and_migrate_database() + except Exception as e: + logger.error(f"数据库自动迁移失败: {e}") + raise e # 返回MainSystem实例 return MainSystem() diff --git a/src/common/database/sqlalchemy_models.py b/src/common/database/sqlalchemy_models.py index 85ad6c281..91b6fa837 100644 --- a/src/common/database/sqlalchemy_models.py +++ b/src/common/database/sqlalchemy_models.py @@ -9,7 +9,6 @@ from sqlalchemy.orm import sessionmaker from sqlalchemy.pool import QueuePool import os import datetime -from src.config.config import global_config from src.common.logger import get_logger import threading from contextlib import contextmanager @@ -25,6 +24,7 @@ def get_string_field(max_length=255, **kwargs): 根据数据库类型返回合适的字符串字段 MySQL需要指定长度的VARCHAR用于索引,SQLite可以使用Text """ + from src.config.config import global_config if global_config.database.database_type == "mysql": return String(max_length, **kwargs) else: @@ -436,6 +436,7 @@ _SessionLocal = None def get_database_url(): """获取数据库连接URL""" + from src.config.config import global_config config = global_config.database if config.database_type == "mysql": @@ -482,6 +483,7 @@ def initialize_database(): return _engine, _SessionLocal database_url = get_database_url() + from src.config.config import global_config config = global_config.database # 配置引擎参数 diff --git a/src/config/config.py b/src/config/config.py index 98d71a665..e3a6b3f07 100644 --- a/src/config/config.py +++ b/src/config/config.py @@ -481,24 +481,4 @@ logger.info("正在品鉴配置文件...") global_config = load_config(config_path=os.path.join(CONFIG_DIR, "bot_config.toml")) model_config = api_ada_load_config(config_path=os.path.join(CONFIG_DIR, "model_config.toml")) -# 初始化数据库连接 -logger.info("正在初始化数据库连接...") -from src.common.database.database import initialize_sql_database -try: - initialize_sql_database(global_config.database) - logger.info(f"数据库连接初始化成功,使用 {global_config.database.database_type} 数据库") -except Exception as e: - logger.error(f"数据库连接初始化失败: {e}") - raise e - -# 初始化数据库表结构 -logger.info("正在初始化数据库表结构...") -from src.common.database.sqlalchemy_models import initialize_database as init_db -try: - init_db() - logger.info("数据库表结构初始化完成") -except Exception as e: - logger.error(f"数据库表结构初始化失败: {e}") - raise e - logger.info("非常的新鲜,非常的美味!")