From d45baaddbc911956f8352830ba80cff95c8fdb9a Mon Sep 17 00:00:00 2001
From: HibiKier <775757368@qq.com>
Date: Sat, 20 Jul 2024 00:45:26 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20:=20=E4=BB=A3=E7=A0=81=E4=BC=98?=
 =?UTF-8?q?=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 zhenxun/builtin_plugins/init/init_config.py | 3 +--
 zhenxun/services/db_context.py              | 4 +++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/zhenxun/builtin_plugins/init/init_config.py b/zhenxun/builtin_plugins/init/init_config.py
index a837d4f9..26534d4d 100644
--- a/zhenxun/builtin_plugins/init/init_config.py
+++ b/zhenxun/builtin_plugins/init/init_config.py
@@ -4,8 +4,7 @@ import nonebot
 from nonebot import get_loaded_plugins
 from nonebot.drivers import Driver
 from nonebot.plugin import Plugin
-from ruamel import yaml
-from ruamel.yaml import YAML, round_trip_dump, round_trip_load
+from ruamel.yaml import YAML
 from ruamel.yaml.comments import CommentedMap
 
 from zhenxun.configs.config import Config
diff --git a/zhenxun/services/db_context.py b/zhenxun/services/db_context.py
index dfd02925..4bd8db7b 100644
--- a/zhenxun/services/db_context.py
+++ b/zhenxun/services/db_context.py
@@ -80,10 +80,12 @@ async def init():
                     # await TestSQL.raw(sql)
                 except Exception as e:
                     logger.debug(f"执行SQL: {sql} 错误...", e=e)
+            if sql_list:
+                logger.debug("SCRIPT_METHOD方法执行完毕!")
         await Tortoise.generate_schemas()
         logger.info(f"Database loaded successfully!")
     except Exception as e:
-        raise Exception(f"数据库连接错误...", e=e)
+        raise Exception(f"数据库连接错误...")
 
 
 async def disconnect():