From 6790a597c41c7ab30509456afe6ac0803a0d6578 Mon Sep 17 00:00:00 2001
From: HibiKier <775757368@qq.com>
Date: Tue, 15 Jul 2025 17:06:24 +0800
Subject: [PATCH] =?UTF-8?q?:zap:=20=E6=B7=BB=E5=8A=A0=E6=97=A7=E7=89=88?=
 =?UTF-8?q?=E6=9C=AC=E5=85=BC=E5=AE=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 zhenxun/services/db_context/__init__.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/zhenxun/services/db_context/__init__.py b/zhenxun/services/db_context/__init__.py
index 6b01ea50..26fd9bcd 100644
--- a/zhenxun/services/db_context/__init__.py
+++ b/zhenxun/services/db_context/__init__.py
@@ -23,8 +23,13 @@ from .config import (
 from .exceptions import DbConnectError, DbUrlIsNode
 from .utils import with_db_timeout
 
+MODELS = db_model.models
+SCRIPT_METHOD = db_model.script_method
+
 __all__ = [
     "DB_TIMEOUT_SECONDS",
+    "MODELS",
+    "SCRIPT_METHOD",
     "SLOW_QUERY_THRESHOLD",
     "DbConnectError",
     "DbUrlIsNode",
@@ -93,6 +98,10 @@ def get_config() -> dict:
 
 @PriorityLifecycle.on_startup(priority=1)
 async def init():
+    global MODELS, SCRIPT_METHOD
+
+    MODELS = db_model.models
+    SCRIPT_METHOD = db_model.script_method
     if not BotConfig.db_url:
         error = prompt.format(host=driver.config.host, port=driver.config.port)
         raise DbUrlIsNode("\n" + error.strip())