yuerenge-database-mcp 0.1.6

A database management tool based on the MCP protocol

g# Yuerenge Database MCP

Package Structure

The package follows a modular architecture organized as:

yuerenge_database_mcp/
├── config/          # Configuration management
├── db_tools/        # Database operations and tools
│   ├── connections/ # Connection management
│   ├── core/        # Core database functionality
│   ├── formatting/  # Data formatting utilities
│   ├── operations/  # Database operations
│   └── utils/       # Utility functions
└── server_lifecycle.py # Server lifecycle management

A database management tool based on the Model Context Protocol (MCP).

This package follows a modular architecture with separate modules for configuration management, database operations, and server lifecycle management.

Features

• Support for multiple databases (MySQL, Oracle, PostgreSQL, SQLite, SQL Server)
• Connection management
• Table structure operations
• Data querying and manipulation
• Advanced configuration management with validation
• Database adapter pattern for easy extension

Installation

pip install yuerenge-database-mcp

Package Modules

The package is organized into several modules:

• config: Handles database configuration loading, validation, and management
• db_tools: Contains all database operation tools and utilities
  • connections: Manages database connections and adapters
  • core: Core database management functionality
  • formatting: Data formatting utilities for different output formats
  • operations: Data and table operation implementations
  • utils: Utility functions including logging and specialized utilities

Quick Start

1. Install the package:

   pip install yuerenge-database-mcp
   

2. Run the server directly:

   yuerenge-database-mcp
   

   Or run with Python module syntax:

   python -m yuerenge_database_mcp
   

3. Create a configuration file (optional, uses defaults if not provided):

   {
     "connections": [
       {
         "name": "my_mysql_db",
         "type": "mysql",
         "host": "localhost",
         "port": 3306,
         "username": "user",
         "password": "password",
         "database": "mydb",
         "enabled": true
       }
     ]
   }
   

4. Run the server:

   yuerenge-database-mcp
   

5. Or run with a specific configuration:

   DATABASE_CONFIG_PATH=/path/to/your/config.json yuerenge-database-mcp
   

Usage

After installation, you can run the database MCP server:

yuerenge-database-mcp

Or with a specific configuration file:

DATABASE_CONFIG_PATH=/path/to/your/config.json yuerenge-database-mcp

You can also run the server using the Python module syntax:

python -m yuerenge_database_mcp

Or using the run_server.py script:

python run_server.py

Configuration

The tool uses a JSON configuration file to store database connection information.

Configuration priority (highest to lowest):

1. Environment variable DATABASE_CONFIG_PATH
2. Default configuration file (config/database_config.json)

Configuration Validation

The configuration manager validates all connection configurations to ensure:

• All required fields are present
• Port numbers are valid (1-65535)
• Database types are supported
• Enabled flags are boolean values

Supported Database Types

• MySQL
• Oracle
• PostgreSQL
• SQLite
• SQL Server

Connection Pool Settings

You can configure connection pool settings for each database connection:

{
  "connections": [
    {
      "name": "my_mysql_db",
      "type": "mysql",
      "host": "localhost",
      "port": 3306,
      "username": "user",
      "password": "password",
      "database": "mydb",
      "enabled": true,
      "pool_size": 5,
      "max_overflow": 10,
      "pool_timeout": 30,
      "pool_recycle": 3600
    }
  ]
}

Available connection pool settings:

• pool_size: The number of connections to keep open inside the connection pool (default: 10)
• max_overflow: The number of connections to allow in connection pool "overflow" (default: 20)
• pool_timeout: The number of seconds to wait before giving up on getting a connection from the pool (default: 30)
• pool_recycle: Number of seconds after which to recreate idle connections (default: 3600)

Environment Variables

The server supports the following environment variables:

• DATABASE_CONFIG_PATH: Specifies the path to the database configuration file
• ERROR_LOG_PATH: Specifies the directory where error logs will be stored (default: ./error_logs)

MCP Server Configuration

To use this as an MCP server, add the following configuration to your MCP client configuration:

{
  "yuerenge-database-mcp": {
    "command": "uvx",
    "args": [
      "yuerenge-database-mcp"
    ],
    "env": {
      "DATABASE_CONFIG_PATH": "path/to/config.json",
      "ERROR_LOG_PATH": "path/to/log/directory"
    }
  }
}

Tools Provided

This MCP server provides the following tools:

1. Connection Management Tools

   • add_database_connection: Add a database connection
   • remove_database_connection: Remove a database connection
   • list_database_connections: List all active connections

2. Configuration Management Tools

   • list_configured_connections: List all connections from the configuration file
   • enable_configured_connection: Enable a connection in the configuration
   • disable_configured_connection: Disable a connection in the configuration
   • reload_configurations: Reload connections from the configuration file

3. Table Structure Tools

   • list_tables: List all tables in the database
   • get_table_structure: Get table structure information
   • create_table: Create a new table
   • drop_table: Drop a table
   • alter_table: Alter table structure

4. Data Query Tools

   • execute_query: Execute raw SQL queries
   • select_data: Select data from a table and format it intelligently

5. Data Manipulation Tools

   • insert_data: Insert data into a table
   • batch_insert_data: Batch insert data
   • update_data: Update data in a table
   • batch_update_data: Batch update data
   • delete_data: Delete data from a table
   • batch_delete_data: Batch delete data

6. Advanced Query Tools

   • select_data_smart: Format table data intelligently
   • select_data_paged: Display table data in pages
   • select_data_summary: Display summary of table data
   • select_data_html: Display table data as HTML and open in browser

These tools are implemented across multiple modules within the db_tools package:

• Core database operations are in the db_tools.core module
• Connection management is in the db_tools.connections module
• Data operations are in the db_tools.operations module
• Data formatting is in the db_tools.formatting module
• Utility functions are in the db_tools.utils module

Usage Examples

Adding a Database Connection

# Using the MCP client to call the tool
add_database_connection(
    name="mysql_test",
    db_type="mysql",
    host="localhost",
    port=3306,
    username="root",
    password="password",
    database="testdb",
    save_to_config=True
)

Querying Data

# Select data from a table
select_data(connection_name="mysql_test", table_name="users", conditions={"age": 25}, limit=10)

Inserting Data

# Insert a single record
insert_data(connection_name="mysql_test", table_name="users", data={
    "name": "John Doe",
    "age": 30,
    "email": "[e-mail]"
})

Batch Operations

# Batch insert multiple records
batch_insert_data(connection_name="mysql_test", table_name="users", data_list=[
    {"name": "Jane Smith", "age": 28, "email": "[e-mail]"},
    {"name": "Bob Johnson", "age": 32, "email": "[e-mail]"}
])

Advanced Queries

# Smart formatting of query results
select_data_smart(connection_name="mysql_test", table_name="users", max_columns=8)

# Paged display of large datasets
select_data_paged(connection_name="mysql_test", table_name="large_table", 
                  columns_per_page=5, rows_per_page=15)

# Display as HTML and open in browser
select_data_html(connection_name="mysql_test", table_name="users")

Best Practices

1. Connection Management

   • Use configuration files to manage database connections, avoiding hardcoded sensitive information
   • Set appropriate connection pool parameters for performance optimization
   • Regularly use the reload_configurations tool to update connection settings

2. Data Operations

   • For bulk data insertion, prefer the batch_insert_data tool
   • Use the conditions parameter for precise queries to avoid full table scans
   • Before executing update or delete operations, first use select_data to verify conditions

3. Data Presentation

   • For tables with many columns, use select_data_paged or select_data_summary tools
   • For data analysis scenarios, use select_data_html for better visualization
   • Choose appropriate limit parameters based on data volume to prevent memory overflow