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

tapdb_maxcompute_mcp/server.py

@@ -1,7 +1,7 @@
 import logging
 import os
 from typing import Any
-
+import json
 import mcp.server.stdio
 import mcp.types as types
 from mcp.server import NotificationOptions, Server
@@ -46,9 +46,20 @@ class MaxCompute:
     def _list_tables(self) -> list[str]:
         return [table.name for table in self.project.list_tables()]
 
-    def _execute_query(self, query: str) -> list[dict[str, Any]]:
+    def _execute_query(self,
+                       query: str,
+                       save_file_path = None) -> list[dict[str, Any]]:
         """Execute a SQL query and return results as a list of dictionaries"""
         logger.debug(f"Executing query: {query}")
+
+        save_file = None
+        if save_file_path is not None:
+            try:
+                save_file = open(save_file_path, "w")
+            except:
+                raise IOError(f"无法打开文件进行写入 : {save_file_path}")
+
+
         try:
             with self.project.execute_sql(query).open_reader() as reader:
                 if query.strip().upper().startswith('DESC'):
@@ -56,10 +67,18 @@ class MaxCompute:
 
                 results = [dict(row) for row in reader]
                 logger.debug(f"Read query returned {len(results)} rows")
-                return results
+
+                if save_file is not None:
+                    save_file.write(json.dumps(results))
+                    return []
+                else:
+                    return results
         except Exception as e:
             logger.error(f"Database error executing query: {e}")
             raise
+        finally:
+            if save_file is not None:
+                save_file.close()
 
 
 async def main():
@@ -99,6 +118,7 @@ async def main():
                     "type": "object",
                     "properties": {
                         "query": {"type": "string", "description": "The read-only SQL query"},
+                        "save_file_path": {"type": "string", "description": "结果写入到本地的文件路径。以json的格式写入。返回值将是一个空list",},
                     },
                     "required": ["query"],
                 },
@@ -127,7 +147,10 @@ async def main():
 
                 sql_text = arguments["query"]
                 validate_readonly_sql(sql_text)
-                results = maxcompute._execute_query(sql_text)
+
+                save_file_path = arguments.get("save_file_path",None)
+
+                results = maxcompute._execute_query(sql_text,save_file_path)
                 return [types.TextContent(type="text", text=str(results))]
             else:
                 raise ValueError(f"Unknown tool: {name}")

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tapdb-maxcompute-mcp
-Version: 0.1.7
+Version: 0.1.9
 Summary: A MCP server for MaxCompute(tapdb revised)
 License-File: LICENSE
 Requires-Python: >=3.10
@@ -40,6 +40,7 @@ The server offers 3 core tools:
    - Execute a read-only query on the MaxCompute project, such as SELECT or DESC query is allowed.
    - Input:
      - `query` (string): The read only SQL query
+     - `save_file_path` (string,opitional): local file that save the query results as json list
    - Returns: Query results as array of objects
 
 #### Schema Tools

tapdb_maxcompute_mcp-0.1.9.dist-info/RECORD

@@ -0,0 +1,7 @@
+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,,

tapdb_maxcompute_mcp-0.1.7.dist-info/RECORD

@@ -1,7 +0,0 @@
-tapdb_maxcompute_mcp/__init__.py,sha256=j0-l3GKDkk17_-h9EfvAsxZ4eskj7qEusam9RPq3hGM,215
-tapdb_maxcompute_mcp/server.py,sha256=Xjg8Tc9NSJ1OR--RBYmiTGAGoHl_N-HcSfBuA1Im5Gg,5798
-tapdb_maxcompute_mcp-0.1.7.dist-info/METADATA,sha256=g3QdhqJw67WBQm4-YGZ3968mqPNXY7h1ngsz5p5wiVA,2429
-tapdb_maxcompute_mcp-0.1.7.dist-info/WHEEL,sha256=KGYbc1zXlYddvwxnNty23BeaKzh7YuoSIvIMO4jEhvw,87
-tapdb_maxcompute_mcp-0.1.7.dist-info/entry_points.txt,sha256=uooKPBbMis_h5b1_XYKUwF4yzpnJl8UGO0-22lPj-z8,67
-tapdb_maxcompute_mcp-0.1.7.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-tapdb_maxcompute_mcp-0.1.7.dist-info/RECORD,,