adbpg-mcp-server 1.0.4__py3-none-any.whl → 1.0.6__py3-none-any.whl

adbpg_mcp_server/__init__.py

@@ -1,4 +0,0 @@
-from .adbpg_mcp_server import run
-
-__version__ = "1.0.3"
-name = "adbpg_mcp_server"

adbpg_mcp_server/adbpg_mcp_server.py

@@ -1,422 +0,0 @@
-import asyncio
-import logging
-import os
-import sys
-import psycopg
-from psycopg import OperationalError as Error
-from mcp.server import Server
-from mcp.types import Resource, Tool, TextContent, ResourceTemplate
-from pydantic import AnyUrl
-from dotenv import load_dotenv
-from mcp.server.stdio import stdio_server
-
-# 配置日志，输出到标准错误
-logging.basicConfig(
-    level=logging.DEBUG,
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
-    stream=sys.stderr
-)
-logger = logging.getLogger("adbpg_mcp_server")
-
-# 加载环境变量
-try:
-    load_dotenv()
-    logger.info("Environment variables loaded")
-    
-    # 检查必要的环境变量
-    required_vars = ["ADBPG_HOST", "ADBPG_PORT", "ADBPG_USER", "ADBPG_PASSWORD", "ADBPG_DATABASE"]
-    missing_vars = [var for var in required_vars if not os.getenv(var)]
-    if missing_vars:
-        error_msg = f"Missing required environment variables: {', '.join(missing_vars)}"
-        logger.error(error_msg)
-        raise ValueError(error_msg)
-    
-    logger.info("All required environment variables are set")
-except Exception as e:
-    logger.error(f"Error loading environment variables: {e}")
-    sys.exit(1)
-
-SERVER_VERSION = "0.1.0"
-
-def get_db_config():
-    """从环境变量获取数据库配置信息"""
-    try:
-        config = {
-            "host": os.getenv("ADBPG_HOST", "localhost"),
-            "port": os.getenv("ADBPG_PORT"),
-            "user": os.getenv("ADBPG_USER"),
-            "password": os.getenv("ADBPG_PASSWORD"),
-            "dbname": os.getenv("ADBPG_DATABASE"),
-            "application_name": f"adbpg-mcp-server-{SERVER_VERSION}"
-        }
-        
-        # 记录配置信息（不包含密码）
-        logger.info(f"Database config: host={config['host']}, port={config['port']}, user={config['user']}, dbname={config['dbname']}")
-        
-        return config
-    except Exception as e:
-        logger.error(f"Error getting database config: {str(e)}")
-        raise
-
-# 初始化服务器
-try:
-    app = Server("adbpg-mcp-server")
-    logger.info("MCP server initialized")
-except Exception as e:
-    logger.error(f"Error initializing MCP server: {e}")
-    sys.exit(1)
-
-@app.list_resources()
-async def list_resources() -> list[Resource]:
-    """列出可用的基本资源"""
-    try:
-        return [
-            Resource(
-                uri="adbpg:///schemas",
-                name="All Schemas",
-                description="AnalyticDB PostgreSQL schemas. List all schemas in the database",
-                mimeType="text/plain"
-            )
-        ]
-    except Exception as e:
-        logger.error(f"Error listing resources: {str(e)}")
-        raise
-
-@app.list_resource_templates()
-async def list_resource_templates() -> list[ResourceTemplate]:
-    """
-    定义动态资源模板
-    
-    返回:
-        list[ResourceTemplate]: 资源模板列表
-        包含以下模板：
-        - 列出schema中的表
-        - 获取表DDL
-        - 获取表统计信息
-    """
-    return [
-        ResourceTemplate(
-            uriTemplate="adbpg:///{schema}/tables",  # 表列表模板
-            name="Schema Tables",
-            description="List all tables in a specific schema",
-            mimeType="text/plain"
-        ),
-        ResourceTemplate(
-            uriTemplate="adbpg:///{schema}/{table}/ddl",  # 表DDL模板
-            name="Table DDL",
-            description="Get the DDL script of a table in a specific schema",
-            mimeType="text/plain"
-        ),
-        ResourceTemplate(
-            uriTemplate="adbpg:///{schema}/{table}/statistics",  # 表统计信息模板
-            name="Table Statistics",
-            description="Get statistics information of a table",
-            mimeType="text/plain"
-        )
-    ]
-
-@app.read_resource()
-async def read_resource(uri: AnyUrl) -> str:
-    """
-    读取资源内容
-    
-    参数:
-        uri (AnyUrl): 资源URI
-        
-    返回:
-        str: 资源内容
-        
-    支持的URI格式：
-    - adbpg:///schemas: 列出所有schema
-    - adbpg:///{schema}/tables: 列出指定schema中的表
-    - adbpg:///{schema}/{table}/ddl: 获取表的DDL
-    - adbpg:///{schema}/{table}/statistics: 获取表的统计信息
-    """
-    config = get_db_config()
-    uri_str = str(uri)
-    
-    if not uri_str.startswith("adbpg:///"):
-        raise ValueError(f"Invalid URI scheme: {uri_str}")
-    
-    try:
-        with psycopg.connect(**config) as conn:  # 建立数据库连接
-            conn.autocommit = True  # 设置自动提交
-            with conn.cursor() as cursor:  # 创建游标
-                path_parts = uri_str[9:].split('/')  # 解析URI路径
-                
-                if path_parts[0] == "schemas":
-                    # 列出所有schema
-                    query = """
-                        SELECT schema_name 
-                        FROM information_schema.schemata 
-                        WHERE schema_name NOT IN ('pg_catalog', 'information_schema')
-                        ORDER BY schema_name;
-                    """
-                    cursor.execute(query)
-                    schemas = cursor.fetchall()
-                    return "\n".join([schema[0] for schema in schemas])
-                    
-                elif len(path_parts) == 2 and path_parts[1] == "tables":
-                    # 列出指定schema中的表
-                    schema = path_parts[0]
-                    query = f"""
-                        SELECT table_name, table_type
-                        FROM information_schema.tables
-                        WHERE table_schema = %s
-                        ORDER BY table_name;
-                    """
-                    cursor.execute(query, (schema,))
-                    tables = cursor.fetchall()
-                    return "\n".join([f"{table[0]} ({table[1]})" for table in tables])
-                    
-                elif len(path_parts) == 3 and path_parts[2] == "ddl":
-                    # 获取表的DDL
-                    schema = path_parts[0]
-                    table = path_parts[1]
-                    query = f"""
-                        SELECT pg_get_ddl('{schema}.{table}'::regclass);
-                    """
-                    cursor.execute(query)
-                    ddl = cursor.fetchone()
-                    return ddl[0] if ddl else f"No DDL found for {schema}.{table}"
-                    
-                elif len(path_parts) == 3 and path_parts[2] == "statistics":
-                    # 获取表的统计信息
-                    schema = path_parts[0]
-                    table = path_parts[1]
-                    query = """
-                        SELECT 
-                            schemaname,
-                            tablename,
-                            attname,
-                            null_frac,
-                            avg_width,
-                            n_distinct,
-                            most_common_vals,
-                            most_common_freqs
-                        FROM pg_stats
-                        WHERE schemaname = %s AND tablename = %s
-                        ORDER BY attname;
-                    """
-                    cursor.execute(query, (schema, table))
-                    rows = cursor.fetchall()
-                    if not rows:
-                        return f"No statistics found for {schema}.{table}"
-                    
-                    result = []
-                    for row in rows:
-                        result.append(f"Column: {row[2]}")
-                        result.append(f"  Null fraction: {row[3]}")
-                        result.append(f"  Average width: {row[4]}")
-                        result.append(f"  Distinct values: {row[5]}")
-                        if row[6]:
-                            result.append(f"  Most common values: {row[6]}")
-                            result.append(f"  Most common frequencies: {row[7]}")
-                        result.append("")
-                    return "\n".join(result)
-                
-                raise ValueError(f"Invalid resource URI format: {uri_str}")
-      
-    except Error as e:
-        raise RuntimeError(f"Database error: {str(e)}")
-
-@app.list_tools()
-async def list_tools() -> list[Tool]:
-    """
-    列出可用的工具
-    
-    返回:
-        list[Tool]: 工具列表
-        包含以下工具：
-        - execute_select_sql: 执行SELECT查询
-        - execute_dml_sql: 执行DML操作
-        - execute_ddl_sql: 执行DDL操作
-        - analyze_table: 分析表统计信息
-        - explain_query: 获取查询执行计划
-    """
-    return [
-        Tool(
-            name="execute_select_sql",
-            description="Execute SELECT SQL to query data from ADBPG database.",
-            inputSchema={
-                "type": "object",
-                "properties": {
-                    "query": {
-                        "type": "string",
-                        "description": "The (SELECT) SQL query to execute"
-                    }
-                },
-                "required": ["query"]
-            }
-        ),
-        Tool(
-            name="execute_dml_sql",
-            description="Execute (INSERT, UPDATE, DELETE) SQL to modify data in ADBPG database.",
-            inputSchema={
-                "type": "object",
-                "properties": {
-                    "query": {
-                        "type": "string",
-                        "description": "The DML SQL query to execute"
-                    }
-                },
-                "required": ["query"]
-            }
-        ),
-        Tool(
-            name="execute_ddl_sql",
-            description="Execute (CREATE, ALTER, DROP) SQL statements to manage database objects.",
-            inputSchema={
-                "type": "object",
-                "properties": {
-                    "query": {
-                        "type": "string",
-                        "description": "The DDL SQL query to execute"
-                    }
-                },
-                "required": ["query"]
-            }
-        ),
-        Tool(
-            name="analyze_table",
-            description="Execute ANALYZE command to collect table statistics.",
-            inputSchema={
-                "type": "object",
-                "properties": {
-                    "schema": {
-                        "type": "string",
-                        "description": "Schema name"
-                    },
-                    "table": {
-                        "type": "string",
-                        "description": "Table name"
-                    }
-                },
-                "required": ["schema", "table"]
-            }
-        ),
-        Tool(
-            name="explain_query",
-            description="Get query execution plan.",
-            inputSchema={
-                "type": "object",
-                "properties": {
-                    "query": {
-                        "type": "string",
-                        "description": "The SQL query to analyze"
-                    }
-                },
-                "required": ["query"]
-            }
-        )
-    ]
-
-@app.call_tool()
-async def call_tool(name: str, arguments: dict) -> list[TextContent]:
-    """
-    执行工具操作
-    
-    参数:
-        name (str): 工具名称
-        arguments (dict): 工具参数
-        
-    返回:
-        list[TextContent]: 执行结果
-        
-    支持的工具：
-    - execute_select_sql: 执行SELECT查询
-    - execute_dml_sql: 执行DML操作
-    - execute_ddl_sql: 执行DDL操作
-    - analyze_table: 分析表统计信息
-    - explain_query: 获取查询执行计划
-    """
-    config = get_db_config()
-    
-    # 根据工具名称处理不同的操作
-    if name == "execute_select_sql":
-        query = arguments.get("query")
-        if not query:
-            raise ValueError("Query is required")
-        if not query.strip().upper().startswith("SELECT"):
-            raise ValueError("Query must be a SELECT statement")
-    elif name == "execute_dml_sql":
-        query = arguments.get("query")
-        if not query:
-            raise ValueError("Query is required")
-        if not any(query.strip().upper().startswith(keyword) for keyword in ["INSERT", "UPDATE", "DELETE"]):
-            raise ValueError("Query must be a DML statement (INSERT, UPDATE, DELETE)")
-    elif name == "execute_ddl_sql":
-        query = arguments.get("query")
-        if not query:
-            raise ValueError("Query is required")
-        if not any(query.strip().upper().startswith(keyword) for keyword in ["CREATE", "ALTER", "DROP"]):
-            raise ValueError("Query must be a DDL statement (CREATE, ALTER, DROP)")
-    elif name == "analyze_table":
-        schema = arguments.get("schema")
-        table = arguments.get("table")
-        if not all([schema, table]):
-            raise ValueError("Schema and table are required")
-        query = f"ANALYZE {schema}.{table}"
-    elif name == "explain_query":
-        query = arguments.get("query")
-        if not query:
-            raise ValueError("Query is required")
-        query = f"EXPLAIN {query}"
-    else:
-        raise ValueError(f"Unknown tool: {name}")
-    
-    try:
-        with psycopg.connect(**config) as conn:
-            conn.autocommit = True
-            with conn.cursor() as cursor:
-                cursor.execute(query)
-                
-                if name == "analyze_table":
-                    return [TextContent(type="text", text=f"Successfully analyzed table {schema}.{table}")]
-                
-                if cursor.description:
-                    columns = [desc[0] for desc in cursor.description]
-                    rows = cursor.fetchall()
-                    result = [",".join(map(str, row)) for row in rows]
-                    return [TextContent(type="text", text="\n".join([",".join(columns)] + result))]
-                else:
-                    return [TextContent(type="text", text="Query executed successfully")]
-    except Exception as e:
-        return [TextContent(type="text", text=f"Error executing query: {str(e)}")]
-
-async def main():
-    """服务器主入口点"""
-    try:
-        config = get_db_config()
-        logger.info("Starting ADBPG MCP server...")
-        
-        # 测试数据库连接
-        try:
-            with psycopg.connect(**config) as conn:
-                logger.info("Successfully connected to database")
-        except Exception as e:
-            logger.error(f"Failed to connect to database: {e}")
-            sys.exit(1)
-        
-        # 使用 stdio 传输
-        async with stdio_server() as (read_stream, write_stream):
-            try:
-                logger.info("Running MCP server with stdio transport...")
-                await app.run(
-                    read_stream=read_stream,
-                    write_stream=write_stream,
-                    initialization_options=app.create_initialization_options()
-                )
-            except Exception as e:
-                logger.error(f"Error running server: {str(e)}")
-                raise
-    except Exception as e:
-        logger.error(f"Server initialization error: {str(e)}")
-        raise
-
-def run():
-    """Run the MCP server."""
-    asyncio.run(main())
-
-if __name__ == "__main__":
-    run() 

adbpg_mcp_server-1.0.4.dist-info/METADATA

@@ -1,128 +0,0 @@
-Metadata-Version: 2.4
-Name: adbpg_mcp_server
-Version: 1.0.4
-Summary: ADBPG MCP Server
-Home-page: https://github.com/aliyun/alibabacloud-adbpg-mcp-server
-Author: Yutian Qiu
-Author-email: qiuytian@gmail.com
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: psycopg>=3.1.0
-Requires-Dist: psycopg-binary>=3.1.0
-Requires-Dist: mcp>=1.4.0
-Requires-Dist: pydantic>=2.0.0
-Requires-Dist: python-dotenv>=1.0.0
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: license-file
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
-
-# AnalyticDB PostgreSQL MCP Server
-
-AnalyticDB PostgreSQL MCP Server serves as a universal interface between AI Agents and AnalyticDB PostgreSQL databases. It enables seamless communication between AI Agents and AnalyticDB PostgreSQL, helping AI Agents retrieve database metadata and execute SQL operations.
-
-## Configuration
-
-
-
-#### Download
-
-Download from Github
-
-```shell
-git clone https://github.com/aliyun/alibabacloud-adbpg-mcp-server.git
-```
-
-#### MCP Integration
-
-Add the following configuration to the MCP client configuration file:
-
-```json
-"mcpServers": {
-  "adbpg-mcp-server": {
-    "command": "uv",
-    "args": [
-      "--directory",
-      "/path/to/adbpg-mcp-server",
-      "run",
-      "adbpg-mcp-server"
-    ],
-    "env": {
-      "ADBPG_HOST": "host",
-      "ADBPG_PORT": "port",
-      "ADBPG_USER": "username",
-      "ADBPG_PASSWORD": "password",
-      "ADBPG_DATABASE": "database"
-    }
-  }
-}
-```
-
-## Components
-
-### Tools
-
-* `execute_select_sql`: Execute SELECT SQL queries on the AnalyticDB PostgreSQL server
-* `execute_dml_sql`: Execute DML (INSERT, UPDATE, DELETE) SQL queries on the AnalyticDB PostgreSQL server
-* `execute_ddl_sql`: Execute DDL (CREATE, ALTER, DROP) SQL queries on the AnalyticDB PostgreSQL server
-* `analyze_table`: Collect table statistics
-* `explain_query`: Get query execution plan
-
-### Resources
-
-#### Built-in Resources
-
-* `adbpg:///schemas`: Get all schemas in the database
-
-#### Resource Templates
-
-* `adbpg:///{schema}/tables`: List all tables in a specific schema
-* `adbpg:///{schema}/{table}/ddl`: Get table DDL
-* `adbpg:///{schema}/{table}/statistics`: Show table statistics
-
-## Environment Variables
-
-MCP Server requires the following environment variables to connect to AnalyticDB PostgreSQL instance:
-
-- `ADBPG_HOST`: Database host address
-- `ADBPG_PORT`: Database port
-- `ADBPG_USER`: Database username
-- `ADBPG_PASSWORD`: Database password
-- `ADBPG_DATABASE`: Database name
-
-## Dependencies
-
-- Python 3.10 or higher
-- Required packages:
-  - mcp >= 1.4.0
-  - psycopg >= 3.1.0
-  - python-dotenv >= 1.0.0
-  - pydantic >= 2.0.0
-
-## Running
-
-```bash
-# Create and activate virtual environment
-uv venv .venv
-source .venv/bin/activate  # Linux/Mac
-# or
-.venv\Scripts\activate     # Windows
-
-# Install dependencies
-uv pip install -e .
-
-# Run server
-uv run adbpg-mcp-server
-```
-
-

adbpg_mcp_server-1.0.4.dist-info/RECORD

@@ -1,8 +0,0 @@
-adbpg_mcp_server/__init__.py,sha256=7emeTRXWecoiR32c5Jr_KFnO94NAlTP5FOJOHRfgMjo,82
-adbpg_mcp_server/adbpg_mcp_server.py,sha256=PP42ZblxSY68dUUpElywnoJ-2DHfoVrxChH9GeTnGnU,15246
-adbpg_mcp_server-1.0.4.dist-info/licenses/LICENSE,sha256=KfhlNIQJu0to_CCmMo4whKiv23yAI_yKJSufqkJN1nw,1073
-adbpg_mcp_server-1.0.4.dist-info/METADATA,sha256=ROtUqARP6-LLfMy0lv7MTkxnI_O5EEAvZohdy18TWD0,3144
-adbpg_mcp_server-1.0.4.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
-adbpg_mcp_server-1.0.4.dist-info/entry_points.txt,sha256=hAsJBdSxCnAe8Y1D8O_vf1PclV_HDbVXAn3TZ6fSNmc,75
-adbpg_mcp_server-1.0.4.dist-info/top_level.txt,sha256=krAI593mc5XIlEZEMGe8qBvn-uqkJtMAr2tBld6ea3M,17
-adbpg_mcp_server-1.0.4.dist-info/RECORD,,

adbpg_mcp_server-1.0.4.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[console_scripts]
-adbpg_mcp_server = adbpg_mcp_server.adbpg_mcp_server:run

adbpg_mcp_server-1.0.4.dist-info/licenses/LICENSE

@@ -1,19 +0,0 @@
-Copyright (c) 2025 The Python Packaging Authority
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.

adbpg_mcp_server-1.0.4.dist-info/top_level.txt

@@ -1 +0,0 @@
-adbpg_mcp_server