refactor: enhance JSON serialization with custom serializer for datetime and Decimal types
This commit is contained in:
parent
0ef5ccc4c2
commit
f3e5f26caf
@ -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")
|
||||
|
||||
# 数据库连接信息
|
||||
|
||||
Loading…
Reference in New Issue
Block a user