diff --git a/zhenxun/models/bot_console.py b/zhenxun/models/bot_console.py index f16e39f5..8d21a763 100644 --- a/zhenxun/models/bot_console.py +++ b/zhenxun/models/bot_console.py @@ -2,6 +2,7 @@ from typing import Literal, overload from tortoise import fields +from zhenxun.configs.config import BotConfig from zhenxun.services.cache.runtime_cache import BotMemoryCache from zhenxun.services.db_context import Model from zhenxun.utils.enum import CacheType @@ -469,9 +470,32 @@ class BotConsole(Model): @classmethod async def _run_script(cls): - return [ + db_type = (BotConfig.get_sql_type() or "").lower() + + scripts = [ "ALTER TABLE bot_console RENAME COLUMN block_plugin TO block_plugins;", "ALTER TABLE bot_console RENAME COLUMN block_task TO block_tasks;", "ALTER TABLE bot_console ADD available_plugins text default '';", "ALTER TABLE bot_console ADD available_tasks text default '';", ] + + if "postgres" in db_type: + scripts.extend( + [ + "ALTER TABLE bot_console ALTER COLUMN block_plugins TYPE TEXT;", + "ALTER TABLE bot_console ALTER COLUMN block_tasks TYPE TEXT;", + "ALTER TABLE bot_console ALTER COLUMN available_plugins TYPE TEXT;", + "ALTER TABLE bot_console ALTER COLUMN available_tasks TYPE TEXT;", + ] + ) + elif "mysql" in db_type: + scripts.extend( + [ + "ALTER TABLE bot_console MODIFY COLUMN block_plugins TEXT;", + "ALTER TABLE bot_console MODIFY COLUMN block_tasks TEXT;", + "ALTER TABLE bot_console MODIFY COLUMN available_plugins TEXT;", + "ALTER TABLE bot_console MODIFY COLUMN available_tasks TEXT;", + ] + ) + + return scripts diff --git a/zhenxun/models/group_console.py b/zhenxun/models/group_console.py index 5d596e7d..f1f317db 100644 --- a/zhenxun/models/group_console.py +++ b/zhenxun/models/group_console.py @@ -4,6 +4,7 @@ from typing_extensions import Self from tortoise import fields from tortoise.backends.base.client import BaseDBAsyncClient +from zhenxun.configs.config import BotConfig from zhenxun.models.plugin_info import PluginInfo from zhenxun.models.task_info import TaskInfo from zhenxun.services.cache import CacheRoot @@ -514,11 +515,52 @@ class GroupConsole(Model): @classmethod def _run_script(cls): - return [ + db_type = (BotConfig.get_sql_type() or "").lower() + + scripts = [ "ALTER TABLE group_console ADD superuser_block_plugin" - " character varying(255) NOT NULL DEFAULT '';", + " Text NOT NULL DEFAULT '';", "ALTER TABLE group_console ADD superuser_block_task" - " character varying(255) NOT NULL DEFAULT '';", + " Text NOT NULL DEFAULT '';", "CREATE INDEX idx_group_console_group_id ON group_console(group_id);", - "CREATE INDEX idx_group_console_group_null_channel ON group_console(group_id) WHERE channel_id IS NULL;", # 单独创建channel为空的索引 # noqa: E501 + ( + "CREATE INDEX idx_group_console_group_null_channel ON " + "group_console(group_id) WHERE channel_id IS NULL;" + ), ] + + if "postgres" in db_type: + scripts.extend( + [ + ( + "ALTER TABLE group_console ALTER COLUMN " + "block_plugin TYPE TEXT;" + ), + ( + "ALTER TABLE group_console ALTER COLUMN " + "superuser_block_plugin TYPE TEXT;" + ), + ("ALTER TABLE group_console ALTER COLUMN " "block_task TYPE TEXT;"), + ( + "ALTER TABLE group_console ALTER COLUMN " + "superuser_block_task TYPE TEXT;" + ), + ] + ) + elif "mysql" in db_type: + scripts.extend( + [ + ("ALTER TABLE group_console MODIFY COLUMN " "block_plugin TEXT;"), + ( + "ALTER TABLE group_console MODIFY COLUMN " + "superuser_block_plugin TEXT;" + ), + ("ALTER TABLE group_console MODIFY COLUMN " "block_task TEXT;"), + ( + "ALTER TABLE group_console MODIFY COLUMN " + "superuser_block_task TEXT;" + ), + ] + ) + + return scripts diff --git a/zhenxun/models/group_info.py b/zhenxun/models/group_info.py index b60120a1..23d32711 100644 --- a/zhenxun/models/group_info.py +++ b/zhenxun/models/group_info.py @@ -1,5 +1,6 @@ from tortoise import fields +from zhenxun.configs.config import BotConfig from zhenxun.services.db_context import Model @@ -109,7 +110,9 @@ class GroupInfo(Model): @classmethod def _run_script(cls): - return [ + db_type = (BotConfig.get_sql_type() or "").lower() + + scripts = [ "ALTER TABLE group_info ADD group_flag Integer NOT NULL DEFAULT 0;", # group_info表添加一个group_flag "ALTER TABLE group_info ALTER COLUMN group_id TYPE character varying(255);", @@ -118,3 +121,20 @@ class GroupInfo(Model): "ALTER TABLE group_info ADD platform character varying(255) NOT NULL" " DEFAULT 'qq';", ] + + if "postgres" in db_type: + scripts.extend( + [ + "ALTER TABLE group_info ALTER COLUMN block_plugin TYPE TEXT;", + "ALTER TABLE group_info ALTER COLUMN block_task TYPE TEXT;", + ] + ) + elif "mysql" in db_type: + scripts.extend( + [ + "ALTER TABLE group_info MODIFY COLUMN block_plugin TEXT;", + "ALTER TABLE group_info MODIFY COLUMN block_task TEXT;", + ] + ) + + return scripts diff --git a/zhenxun/services/db_context/__init__.py b/zhenxun/services/db_context/__init__.py index b5ec0be6..f42e24e4 100644 --- a/zhenxun/services/db_context/__init__.py +++ b/zhenxun/services/db_context/__init__.py @@ -7,6 +7,7 @@ import nonebot from nonebot.utils import is_coroutine_callable from tortoise import Tortoise from tortoise.connection import connections +from tortoise.exceptions import OperationalError from zhenxun.configs.config import BotConfig from zhenxun.services.log import logger @@ -50,11 +51,8 @@ def get_config() -> dict: raise DbUrlIsNode("数据库Url连接字符串为空，请检查配置文件（.env.dev）") parsed = urlparse(BotConfig.db_url) - # 基础配置 config = { - "connections": { - "default": BotConfig.db_url # 默认直接使用连接字符串 - }, + "connections": {"default": BotConfig.db_url}, "apps": { "models": { "models": db_model.models, @@ -64,7 +62,6 @@ def get_config() -> dict: "timezone": "Asia/Shanghai", } - # 根据数据库类型应用高级配置 if parsed.scheme.startswith("postgres"): config["connections"]["default"] = { "engine": "tortoise.backends.asyncpg", @@ -143,7 +140,24 @@ async def init(): await asyncio.wait_for( db.execute_query_dict(sql), timeout=DB_TIMEOUT_SECONDS ) - # await TestSQL.raw(sql) + except OperationalError as e: + err_str = str(e).lower() + if any( + x in err_str + for x in [ + "already exists", + "duplicate column", + "已经存在", + "已存在", + ] + ): + pass + elif any( + x in err_str for x in ["does not exist", "check that", "不存在"] + ) and ("drop" in sql.lower() or "rename" in sql.lower()): + pass + else: + logger.warning(f"执行SQL警告: {sql} || {e}") except Exception as e: logger.debug(f"执行SQL: {sql} 错误...", e=e) if sql_list: