mcp-dbutils 0.5.0__py3-none-any.whl → 0.7.0__py3-none-any.whl

mcp_dbutils/base.py

@@ -220,29 +220,60 @@ class DatabaseServer:
                         },
                         "required": ["database", "sql"]
                     }
+                ),
+                types.Tool(
+                    name="list_tables",
+                    description="List all available tables in the specified database",
+                    inputSchema={
+                        "type": "object",
+                        "properties": {
+                            "database": {
+                                "type": "string",
+                                "description": "Database configuration name"
+                            }
+                        },
+                        "required": ["database"]
+                    }
                 )
             ]
 
         @self.server.call_tool()
         async def handle_call_tool(name: str, arguments: dict) -> list[types.TextContent]:
-            if name != "query":
-                raise ConfigurationError(f"Unknown tool: {name}")
-
             if "database" not in arguments:
                 raise ConfigurationError("Database configuration name must be specified")
 
-            sql = arguments.get("sql", "").strip()
-            if not sql:
-                raise ConfigurationError("SQL query cannot be empty")
-
-            # Only allow SELECT statements
-            if not sql.lower().startswith("select"):
-                raise ConfigurationError("Only SELECT queries are supported for security reasons")
-
             database = arguments["database"]
-            async with self.get_handler(database) as handler:
-                result = await handler.execute_query(sql)
-                return [types.TextContent(type="text", text=result)]
+
+            if name == "list_tables":
+                async with self.get_handler(database) as handler:
+                    tables = await handler.get_tables()
+                    if not tables:
+                        # 空表列表的情况也返回数据库类型
+                        return [types.TextContent(type="text", text=f"[{handler.db_type}] No tables found")]
+                    
+                    formatted_tables = "\n".join([
+                        f"Table: {table.name}\n" +
+                        f"URI: {table.uri}\n" +
+                        (f"Description: {table.description}\n" if table.description else "") +
+                        "---"
+                        for table in tables
+                    ])
+                    # 添加数据库类型前缀
+                    return [types.TextContent(type="text", text=f"[{handler.db_type}]\n{formatted_tables}")]
+            elif name == "query":
+                sql = arguments.get("sql", "").strip()
+                if not sql:
+                    raise ConfigurationError("SQL query cannot be empty")
+
+                # Only allow SELECT statements
+                if not sql.lower().startswith("select"):
+                    raise ConfigurationError("Only SELECT queries are supported for security reasons")
+
+                async with self.get_handler(database) as handler:
+                    result = await handler.execute_query(sql)
+                    return [types.TextContent(type="text", text=result)]
+            else:
+                raise ConfigurationError(f"Unknown tool: {name}")
 
     async def run(self):
         """Run server"""

{mcp_dbutils-0.5.0.dist-info → mcp_dbutils-0.7.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-dbutils
-Version: 0.5.0
+Version: 0.7.0
 Summary: MCP Database Utilities Service
 Author: Dong Hao
 License-Expression: MIT
@@ -39,6 +39,7 @@ MCP Database Utilities is a unified database access service that supports multip
 - Support for multiple database configurations
 - Secure read-only query execution
 - Table structure and schema information retrieval
+- Database tables listing via MCP tools
 - Intelligent connection management and resource cleanup
 - Debug mode support
 
@@ -165,16 +166,31 @@ databases:
     user: postgres            # Credentials must be provided separately
     password: secret          # Not included in JDBC URL for security
 
-  # SQLite example (when using Docker)
+  # SQLite standard configuration
   my_sqlite:
     type: sqlite
-    path: /app/sqlite.db       # Mapped path inside container
+    path: /app/sqlite.db       # Database file path
     password: optional_password # optional
+
+  # SQLite with JDBC URL configuration
+  my_sqlite_jdbc:
+    type: sqlite
+    jdbc_url: jdbc:sqlite:/app/data.db?mode=ro&cache=shared  # Supports query parameters
+    password: optional_password    # Provided separately for security
 ```
 
-The configuration supports two formats for PostgreSQL:
+The configuration supports JDBC URL format for both PostgreSQL and SQLite:
+
+PostgreSQL:
 1. Standard configuration with individual parameters
-2. JDBC URL configuration with separate credentials (recommended for better compatibility)
+2. JDBC URL configuration with separate credentials
+
+SQLite:
+1. Standard configuration with path parameter
+2. JDBC URL configuration with query parameters support:
+   - mode=ro: Read-only mode
+   - cache=shared: Shared cache mode
+   - Other SQLite URI parameters
 
 ### Debug Mode
 Set environment variable `MCP_DEBUG=1` to enable debug mode for detailed logging output.
@@ -277,9 +293,24 @@ except Exception as e:
 ### DatabaseServer
 Core server class providing:
 - Resource list retrieval
-- Tool call handling
+- Tool call handling (list_tables, query)
 - Database handler management
 
+### MCP Tools
+
+#### list_tables
+Lists all tables in the specified database.
+- Parameters:
+  * database: Database configuration name
+- Returns: Text content with a list of table names
+
+#### query
+Executes a SQL query on the specified database.
+- Parameters:
+  * database: Database configuration name
+  * sql: SQL query to execute (SELECT only)
+- Returns: Query results in a formatted text
+
 ### DatabaseHandler
 Abstract base class defining interfaces:
 - get_tables(): Get table resource list

{mcp_dbutils-0.5.0.dist-info → mcp_dbutils-0.7.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 mcp_dbutils/__init__.py,sha256=xcfE1spAaONAoxBYB1ZyDX8tw7nxV1PMqo_RwxLDp0A,1892
-mcp_dbutils/base.py,sha256=xdeLOKv9-dbMUzlIbD-BbRVbqAf1a3RjO25aw61vAzY,9691
+mcp_dbutils/base.py,sha256=aawU0mY78Ga8YhSYvzOaBi1V8qSYTWyAuZzmXTc37Rg,11232
 mcp_dbutils/config.py,sha256=EwnPNuQVCBKd5WOXQfROyDTM-YpM_Odp0GhCPRg8YwE,1863
 mcp_dbutils/log.py,sha256=fibVIwsb1HVU5zriGrDZTMEirKjgIuxuN_B_YTdAJ7I,996
 mcp_dbutils/stats.py,sha256=2hiKi_M8V4xhVHlH5FS-Df5GuMEpuzif12C8ik06Khs,2538
@@ -11,8 +11,8 @@ mcp_dbutils/sqlite/__init__.py,sha256=QV4th2ywzUmCCa3GHCcBf8blJ_E8OYy0dJ2fSf1TfS
 mcp_dbutils/sqlite/config.py,sha256=RTHT2Xx--g-osD73CpT8DrCk0VHpHfPil3D6YUzXD-g,4519
 mcp_dbutils/sqlite/handler.py,sha256=bf_k93rCcJn09zc7tsqrlbiTGUg3FspimfWKxK_JQTs,4970
 mcp_dbutils/sqlite/server.py,sha256=7Bbq9l7Ca_4dzkAbbdRcXxvHoO_NFLzZHwlhKB0HIJc,7724
-mcp_dbutils-0.5.0.dist-info/METADATA,sha256=g3WO-pqGFLsXxu1RcLQg9KjHEjVSG9ncrjqTzcgkm3A,9966
-mcp_dbutils-0.5.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_dbutils-0.5.0.dist-info/entry_points.txt,sha256=XTjt0QmYRgKOJQT6skR9bp1EMUfIrgpHeZJPZ3CJffs,49
-mcp_dbutils-0.5.0.dist-info/licenses/LICENSE,sha256=1A_CwpWVlbjrKdVEYO77vYfnXlW7oxcilZ8FpA_BzCI,1065
-mcp_dbutils-0.5.0.dist-info/RECORD,,
+mcp_dbutils-0.7.0.dist-info/METADATA,sha256=smgCaufFbIkvOFhDGqrajHYC9WxeQcDCWRerr_bxHp8,10840
+mcp_dbutils-0.7.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_dbutils-0.7.0.dist-info/entry_points.txt,sha256=XTjt0QmYRgKOJQT6skR9bp1EMUfIrgpHeZJPZ3CJffs,49
+mcp_dbutils-0.7.0.dist-info/licenses/LICENSE,sha256=1A_CwpWVlbjrKdVEYO77vYfnXlW7oxcilZ8FpA_BzCI,1065
+mcp_dbutils-0.7.0.dist-info/RECORD,,