From 1161be91f8b6e98c39f35c3d50aa78fee8decda8 Mon Sep 17 00:00:00 2001
From: fit2cloud-chenyw <yawen.chen@fit2cloud.com>
Date: Thu, 28 Aug 2025 18:15:29 +0800
Subject: [PATCH] perf: Embedded dynamic data source acquisition version

---
 backend/apps/db/db.py | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/backend/apps/db/db.py b/backend/apps/db/db.py
index 47147d77..f6c1b181 100644
--- a/backend/apps/db/db.py
+++ b/backend/apps/db/db.py
@@ -155,8 +155,19 @@ def check_connection(trans: Trans, ds: CoreDatasource, is_raise: bool = False):
                     return False
 
 
-def get_version(ds: CoreDatasource):
-    conf = DatasourceConf(**json.loads(aes_decrypt(ds.configuration))) if ds.type != "excel" else get_engine_config()
+def get_version(ds: CoreDatasource | AssistantOutDsSchema):
+    conf = None
+    if isinstance(ds, CoreDatasource):
+        conf = DatasourceConf(**json.loads(aes_decrypt(ds.configuration))) if ds.type != "excel" else get_engine_config()
+    if isinstance(ds, AssistantOutDsSchema):
+        conf = DatasourceConf()
+        conf.host = ds.host
+        conf.port = ds.port
+        conf.username = ds.user
+        conf.password = ds.password
+        conf.database = ds.dataBase
+        conf.dbSchema = ds.db_schema
+        conf.timeout = 10
     db = DB.get_db(ds.type)
     sql = get_version_sql(ds, conf)
     try: