tapdb-maxcompute-mcp 0.1.9__py3-none-any.whl → 0.1.10__py3-none-any.whl

tapdb_maxcompute_mcp/server.py

@@ -9,6 +9,8 @@ from mcp.server.models import InitializationOptions
 from odps import ODPS
 import sqlglot
 from sqlglot import exp, parse_one,Expression
+from datetime import datetime,date,timedelta
+from decimal import Decimal
 
 logger = logging.getLogger('tapdb-maxcompute-mcp')
 # logger.setLevel(logging.DEBUG)
@@ -16,6 +18,18 @@ logger = logging.getLogger('tapdb-maxcompute-mcp')
 
 logger.info("Starting MaxCompute MCP Server(tapdb revised)")
 
+def dd_json_serializer(obj):
+    if isinstance(obj, datetime):
+        if obj.microsecond == 0:
+            return obj.strftime('%Y-%m-%d %H:%M:%S')
+        else:
+            return obj.strftime('%Y-%m-%d %H:%M:%S.%f')
+    elif isinstance(obj, date):
+        return obj.strftime('%Y-%m-%d')
+    elif isinstance(obj, Decimal):
+        return float(obj)
+    else:
+        raise TypeError("%r is not JSON serializable" % obj)
 
 
 def validate_readonly_sql(sql_text:str):
@@ -69,7 +83,7 @@ class MaxCompute:
                 logger.debug(f"Read query returned {len(results)} rows")
 
                 if save_file is not None:
-                    save_file.write(json.dumps(results))
+                    save_file.write(json.dumps(results,default=dd_json_serializer))
                     return []
                 else:
                     return results

{tapdb_maxcompute_mcp-0.1.9.dist-info → tapdb_maxcompute_mcp-0.1.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tapdb-maxcompute-mcp
-Version: 0.1.9
+Version: 0.1.10
 Summary: A MCP server for MaxCompute(tapdb revised)
 License-File: LICENSE
 Requires-Python: >=3.10

tapdb_maxcompute_mcp-0.1.10.dist-info/RECORD

@@ -0,0 +1,7 @@
+tapdb_maxcompute_mcp/__init__.py,sha256=j0-l3GKDkk17_-h9EfvAsxZ4eskj7qEusam9RPq3hGM,215
+tapdb_maxcompute_mcp/server.py,sha256=quwH9M5IsGobivp7vKQ6G5GuCq6SWQC2zg8HE-2QyK0,7149
+tapdb_maxcompute_mcp-0.1.10.dist-info/METADATA,sha256=GsnDPBKneKuw2hVtKovB3iXyG7yUxG6Xfr8MRMlr9TU,2526
+tapdb_maxcompute_mcp-0.1.10.dist-info/WHEEL,sha256=KGYbc1zXlYddvwxnNty23BeaKzh7YuoSIvIMO4jEhvw,87
+tapdb_maxcompute_mcp-0.1.10.dist-info/entry_points.txt,sha256=uooKPBbMis_h5b1_XYKUwF4yzpnJl8UGO0-22lPj-z8,67
+tapdb_maxcompute_mcp-0.1.10.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+tapdb_maxcompute_mcp-0.1.10.dist-info/RECORD,,

tapdb_maxcompute_mcp-0.1.9.dist-info/RECORD

@@ -1,7 +0,0 @@
-tapdb_maxcompute_mcp/__init__.py,sha256=j0-l3GKDkk17_-h9EfvAsxZ4eskj7qEusam9RPq3hGM,215
-tapdb_maxcompute_mcp/server.py,sha256=EUbs2LwyY-Z2dgZKp-7f_pohInf_g4-8K876CFZskic,6626
-tapdb_maxcompute_mcp-0.1.9.dist-info/METADATA,sha256=DOqiaABJRmW4QECYc6FsF0mg5xkyvlDJ-bAAhrhR7SI,2525
-tapdb_maxcompute_mcp-0.1.9.dist-info/WHEEL,sha256=KGYbc1zXlYddvwxnNty23BeaKzh7YuoSIvIMO4jEhvw,87
-tapdb_maxcompute_mcp-0.1.9.dist-info/entry_points.txt,sha256=uooKPBbMis_h5b1_XYKUwF4yzpnJl8UGO0-22lPj-z8,67
-tapdb_maxcompute_mcp-0.1.9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-tapdb_maxcompute_mcp-0.1.9.dist-info/RECORD,,