From f3e5f26caf068748e16847465053e13d931d4f47 Mon Sep 17 00:00:00 2001
From: CaptainB <bin@fit2cloud.com>
Date: Fri, 4 Jul 2025 13:19:09 +0800
Subject: [PATCH] refactor: enhance JSON serialization with custom serializer
 for datetime and Decimal types

---
 apps/tools/migrations/internal_tool.sql | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/apps/tools/migrations/internal_tool.sql b/apps/tools/migrations/internal_tool.sql
index 3cb66d614..5e93dfa00 100644
--- a/apps/tools/migrations/internal_tool.sql
+++ b/apps/tools/migrations/internal_tool.sql
@@ -66,6 +66,15 @@ def query_mysql(host,port, user, password, database, sql):
     import pymysql
     import json
     from pymysql.cursors import DictCursor
+    from datetime import datetime, date
+
+    def default_serializer(obj):
+        from decimal import Decimal
+        if isinstance(obj, (datetime, date)):
+            return obj.isoformat()  # 将 datetime/date 转换为 ISO 格式字符串
+        elif isinstance(obj, Decimal):
+            return float(obj)  # 将 Decimal 转换为 float
+        raise TypeError(f"Type {type(obj)} not serializable")
 
     try:
         # 创建连接
@@ -94,7 +103,7 @@ def query_mysql(host,port, user, password, database, sql):
                     row[key] = value.decode("utf-8")  # 转换为字符串
 
         # 将数据序列化为 JSON
-        json_data = json.dumps(data, ensure_ascii=False)
+        json_data = json.dumps(data, default=default_serializer, ensure_ascii=False)
         return json_data
 
         # 关闭数据库连接
@@ -111,8 +120,11 @@ def queryPgSQL(database, user, password, host, port, query):
 
     # 自定义 JSON 序列化函数
     def default_serializer(obj):
+        from decimal import Decimal
         if isinstance(obj, datetime):
             return obj.isoformat()  # 将 datetime 转换为 ISO 格式字符串
+        elif isinstance(obj, Decimal):
+            return float(obj)  # 将 Decimal 转换为 float
         raise TypeError(f"Type {type(obj)} not serializable")
 
     # 数据库连接信息