mcp-dbutils 0.14.0__py3-none-any.whl → 0.15.0__py3-none-any.whl

mcp_dbutils/__init__.py

@@ -1,15 +1,16 @@
 """MCP Connection Utilities Service"""
 
-import asyncio
 import argparse
+import asyncio
 import os
 import sys
+from importlib.metadata import metadata
 from pathlib import Path
+
 import yaml
-from importlib.metadata import metadata
 
+from .base import LOG_NAME, ConnectionServer
 from .log import create_logger
-from .base import ConnectionServer, LOG_NAME
 
 # 获取包信息
 pkg_meta = metadata("mcp-dbutils")

mcp_dbutils/base.py

@@ -12,18 +12,17 @@ class ConnectionError(ConnectionHandlerError):
     """Connection related errors"""
     pass
 
+import json
 from abc import ABC, abstractmethod
-from typing import Any, List, Optional, AsyncContextManager
 from contextlib import asynccontextmanager
-import json
-import yaml
-import time
 from datetime import datetime
 from importlib.metadata import metadata
-from mcp.server import Server, NotificationOptions
+from typing import AsyncContextManager
+
 import mcp.server.stdio
 import mcp.types as types
-from mcp.shared.session import RequestResponder
+import yaml
+from mcp.server import Server
 
 from .log import create_logger
 from .stats import ResourceStats

mcp_dbutils/config.py

@@ -1,11 +1,10 @@
 """Common configuration utilities"""
 
 import os
-import yaml
 from abc import ABC, abstractmethod
-from dataclasses import dataclass, field
-from typing import Optional, Dict, Any, Literal
-from pathlib import Path
+from typing import Any, Dict, Literal
+
+import yaml
 
 # Supported connection types
 ConnectionType = Literal['sqlite', 'postgres', 'mysql']

mcp_dbutils/log.py

@@ -1,8 +1,9 @@
 """日志处理模块"""
 
 import sys
+from collections.abc import Callable
 from datetime import datetime
-from typing import Callable
+
 
 def create_logger(name: str, is_debug: bool = False) -> Callable:
     """创建stderr日志函数，用于本地调试

mcp_dbutils/mysql/__init__.py

@@ -1,6 +1,6 @@
 """MySQL module"""
 
-from .handler import MySQLHandler
 from .config import MySQLConfig
+from .handler import MySQLHandler
 
 __all__ = ['MySQLHandler', 'MySQLConfig']

mcp_dbutils/mysql/config.py

@@ -1,9 +1,11 @@
 """MySQL configuration module"""
 from dataclasses import dataclass
-from typing import Optional, Dict, Any, Literal
-from urllib.parse import urlparse, parse_qs
+from typing import Any, Dict, Literal, Optional
+from urllib.parse import parse_qs, urlparse
+
 from ..config import ConnectionConfig
 
+
 @dataclass
 class SSLConfig:
     """SSL configuration for MySQL connection"""

mcp_dbutils/mysql/handler.py

@@ -1,12 +1,12 @@
 """MySQL connection handler implementation"""
 
-import mysql.connector
-from mysql.connector.pooling import MySQLConnectionPool
 import mcp.types as types
+import mysql.connector
 
 from ..base import ConnectionHandler, ConnectionHandlerError
 from .config import MySQLConfig
 
+
 class MySQLHandler(ConnectionHandler):
     @property
     def db_type(self) -> str:

mcp_dbutils/mysql/server.py

@@ -1,15 +1,15 @@
 """MySQL MCP server implementation"""
+from typing import Optional
+
+import mcp.types as types
 import mysql.connector
 from mysql.connector.pooling import MySQLConnectionPool, PooledMySQLConnection
-from typing import Optional, List
-import mcp.types as types
-from importlib.metadata import metadata
-from ..base import ConnectionServer
+
+# 获取包信息用于日志命名
+from ..base import LOG_NAME, ConnectionServer
 from ..log import create_logger
 from .config import MySQLConfig
 
-# 获取包信息用于日志命名
-from ..base import LOG_NAME
 
 class MySQLServer(ConnectionServer):
     def __init__(self, config: MySQLConfig, config_path: Optional[str] = None):

mcp_dbutils/postgres/__init__.py

@@ -1,6 +1,6 @@
 """PostgreSQL module"""
 
-from .handler import PostgreSQLHandler
 from .config import PostgreSQLConfig
+from .handler import PostgreSQLHandler
 
 __all__ = ['PostgreSQLHandler', 'PostgreSQLConfig']

mcp_dbutils/postgres/config.py

@@ -1,9 +1,11 @@
 """PostgreSQL configuration module"""
 from dataclasses import dataclass
-from typing import Optional, Dict, Any, Literal
-from urllib.parse import urlparse, parse_qs
+from typing import Any, Dict, Literal, Optional
+from urllib.parse import parse_qs, urlparse
+
 from ..config import ConnectionConfig
 
+
 @dataclass
 class SSLConfig:
     """SSL configuration for PostgreSQL connection"""

mcp_dbutils/postgres/handler.py

@@ -1,12 +1,12 @@
 """PostgreSQL connection handler implementation"""
 
-import psycopg2
-from psycopg2.pool import SimpleConnectionPool
 import mcp.types as types
+import psycopg2
 
 from ..base import ConnectionHandler, ConnectionHandlerError
 from .config import PostgreSQLConfig
 
+
 class PostgreSQLHandler(ConnectionHandler):
     @property
     def db_type(self) -> str:

mcp_dbutils/postgres/server.py

@@ -1,15 +1,15 @@
 """PostgreSQL MCP server implementation"""
+from typing import Optional
+
+import mcp.types as types
 import psycopg2
 from psycopg2.pool import SimpleConnectionPool
-from typing import Optional, List
-import mcp.types as types
-from importlib.metadata import metadata
-from ..base import ConnectionServer
+
+# 获取包信息用于日志命名
+from ..base import LOG_NAME, ConnectionServer
 from ..log import create_logger
 from .config import PostgreSQLConfig
 
-# 获取包信息用于日志命名
-from ..base import LOG_NAME
 
 class PostgreSQLServer(ConnectionServer):
     def __init__(self, config: PostgreSQLConfig, config_path: Optional[str] = None):

mcp_dbutils/sqlite/__init__.py

@@ -1,6 +1,6 @@
 """SQLite module"""
 
-from .handler import SQLiteHandler
 from .config import SQLiteConfig
+from .handler import SQLiteHandler
 
 __all__ = ['SQLiteHandler', 'SQLiteConfig']

mcp_dbutils/sqlite/config.py

@@ -2,10 +2,12 @@
 
 from dataclasses import dataclass
 from pathlib import Path
-from typing import Dict, Any, Optional, Literal
-from urllib.parse import urlparse, parse_qs
+from typing import Any, Dict, Literal, Optional
+from urllib.parse import parse_qs, urlparse
+
 from ..config import ConnectionConfig
 
+
 def parse_jdbc_url(jdbc_url: str) -> Dict[str, str]:
     """Parse JDBC URL into connection parameters
 

mcp_dbutils/sqlite/handler.py

@@ -1,13 +1,13 @@
 """SQLite connection handler implementation"""
 
 import sqlite3
-import json
-from typing import Any
+
 import mcp.types as types
 
 from ..base import ConnectionHandler, ConnectionHandlerError
 from .config import SQLiteConfig
 
+
 class SQLiteHandler(ConnectionHandler):
     @property
     def db_type(self) -> str:

mcp_dbutils/sqlite/server.py

@@ -1,18 +1,17 @@
 """SQLite MCP server implementation"""
 
 import sqlite3
-from pathlib import Path
 from contextlib import closing
-from typing import Optional, List
+from pathlib import Path
+from typing import Optional
+
 import mcp.types as types
-from importlib.metadata import metadata
 
-from ..base import ConnectionServer
+# 获取包信息用于日志命名
+from ..base import LOG_NAME, ConnectionServer
 from ..log import create_logger
 from .config import SQLiteConfig
 
-# 获取包信息用于日志命名
-from ..base import LOG_NAME
 
 class SQLiteServer(ConnectionServer):
     def __init__(self, config: SQLiteConfig, config_path: Optional[str] = None):

mcp_dbutils/stats.py

@@ -1,10 +1,11 @@
 """Resource monitoring statistics module"""
 
+import statistics
+import sys
 from dataclasses import dataclass
 from datetime import datetime
-from typing import Optional, List, Tuple
-import sys
-import statistics
+from typing import List, Optional, Tuple
+
 
 @dataclass
 class ResourceStats:

{mcp_dbutils-0.14.0.dist-info → mcp_dbutils-0.15.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-dbutils
-Version: 0.14.0
+Version: 0.15.0
 Summary: MCP Database Utilities Service
 Author: Dong Hao
 License-Expression: MIT
@@ -14,10 +14,12 @@ Requires-Dist: pyyaml>=6.0.2
 Provides-Extra: test
 Requires-Dist: aiosqlite>=0.19.0; extra == 'test'
 Requires-Dist: docker>=7.0.0; extra == 'test'
+Requires-Dist: pre-commit>=3.6.0; extra == 'test'
 Requires-Dist: pytest-asyncio>=0.23.0; extra == 'test'
 Requires-Dist: pytest-cov>=4.1.0; extra == 'test'
 Requires-Dist: pytest-docker>=2.0.0; extra == 'test'
 Requires-Dist: pytest>=7.0.0; extra == 'test'
+Requires-Dist: ruff>=0.3.0; extra == 'test'
 Requires-Dist: testcontainers>=3.7.0; extra == 'test'
 Description-Content-Type: text/markdown
 
@@ -483,6 +485,19 @@ Our CI/CD pipeline automatically performs:
 
 Pull requests that don't meet these standards will be automatically blocked from merging.
 
+### Code Style
+We use Ruff for code style checking and formatting:
+
+[![Code Style](https://img.shields.io/badge/code%20style-ruff-000000.svg)](https://github.com/astral-sh/ruff)
+
+All code must follow our style guide:
+- Line length: 88 characters
+- Indentation: 4 spaces
+- Quotes: Double quotes
+- Naming: PEP8 conventions
+
+For detailed guidelines, see [STYLE_GUIDE.md](docs/STYLE_GUIDE.md).
+
 ### Local Development
 To check code quality locally:
 1. Run tests with coverage:
@@ -491,6 +506,26 @@ To check code quality locally:
    ```
 2. Use SonarLint in your IDE to catch issues early
 3. Review SonarCloud analysis results in PR comments
+4. Run Ruff for code style checking:
+   ```bash
+   # Install Ruff
+   uv pip install ruff
+   
+   # Check code style
+   ruff check .
+   
+   # Format code
+   ruff format .
+   ```
+5. Use pre-commit hooks for automatic checks:
+   ```bash
+   # Install pre-commit
+   uv pip install pre-commit
+   pre-commit install
+   
+   # Run all checks
+   pre-commit run --all-files
+   ```
 
 ## Contributing
 Contributions are welcome! Here's how you can help:

mcp_dbutils-0.15.0.dist-info/RECORD

@@ -0,0 +1,22 @@
+mcp_dbutils/__init__.py,sha256=6LLccQv7je2L4IpY_I3OzSJZcK32VUDJv2IY31y6eYg,1900
+mcp_dbutils/base.py,sha256=A9BUV3YmFpjkW88nfnJv38Fbk4gu3tHIs2D8Wpitg9E,27571
+mcp_dbutils/config.py,sha256=bmXpOd1fyYfoyUS75I035ChT6t3wP5AyEnJ06e2ZS2o,1848
+mcp_dbutils/log.py,sha256=pwnY1Y8R0wWrAFleFJlOM6m1ey1itgwijsXZmWFd2Z0,819
+mcp_dbutils/stats.py,sha256=dOEvkEgZwgbbQn10diS-VNUyuhCCNGlQ990ZoCqTKHM,7102
+mcp_dbutils/mysql/__init__.py,sha256=gNhoHaxK1qhvMAH5AVl1vfV1rUpcbV9KZWUQb41aaQk,129
+mcp_dbutils/mysql/config.py,sha256=Yvdd02ZwPMM7RCjEvjNJphGiUfImI7Q2gcEH6Zi3Vjo,8071
+mcp_dbutils/mysql/handler.py,sha256=J-vitJ1DDZIGB3OTDdAai3rzvR2DEpFrvZ9egcKbJi4,19682
+mcp_dbutils/mysql/server.py,sha256=6Tcs5pAb_YZPjP5EjtGMT-bHa4kICXehbLqbS22nLOM,8526
+mcp_dbutils/postgres/__init__.py,sha256=-2zYuEJEQ2AMvmGhH5Z_umerSvt7S4xOa_XV4wgvGfI,154
+mcp_dbutils/postgres/config.py,sha256=NyQOVhkXJ1S-JD0w-ePNjTKI1Ja-aZQkDUdHi6U7Vl4,7752
+mcp_dbutils/postgres/handler.py,sha256=3uiXK0-Qgn5Rbyr-C2FX9Khk7H2O7-jYQ06kfM1RAJg,24351
+mcp_dbutils/postgres/server.py,sha256=_CiJC9PitpI1NB99Q1Bcs5TYADNgDpYMwv88fRHQunE,8640
+mcp_dbutils/sqlite/__init__.py,sha256=fK_3-WylCBYpBAzwuopi8hlwoIGJm2TPAlwcPWG46I0,134
+mcp_dbutils/sqlite/config.py,sha256=B_LlB8I1Akh70brcTSuIESEVTz27wayN-NLoVXk5ykE,4535
+mcp_dbutils/sqlite/handler.py,sha256=y1ICrwhoJNczUiACkadhS4FAAb2c5RvMZwEk4J_qwq0,17698
+mcp_dbutils/sqlite/server.py,sha256=OMKOMCDJLlodKQMsqtsRbHAteKYDNaLV6tvz_LhZWdo,7631
+mcp_dbutils-0.15.0.dist-info/METADATA,sha256=7oY5Z-fYgAo7GqTyv4eUaiL6gVoqqEqdEdlVL732J8Q,16095
+mcp_dbutils-0.15.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_dbutils-0.15.0.dist-info/entry_points.txt,sha256=XTjt0QmYRgKOJQT6skR9bp1EMUfIrgpHeZJPZ3CJffs,49
+mcp_dbutils-0.15.0.dist-info/licenses/LICENSE,sha256=1A_CwpWVlbjrKdVEYO77vYfnXlW7oxcilZ8FpA_BzCI,1065
+mcp_dbutils-0.15.0.dist-info/RECORD,,

mcp_dbutils-0.14.0.dist-info/RECORD

@@ -1,22 +0,0 @@
-mcp_dbutils/__init__.py,sha256=t3Ds0b_WhFSSw2widqJ2IDQdIGNQuM5215UU-hENPXw,1899
-mcp_dbutils/base.py,sha256=TDE5y7e2-Q5WOKFCesMnoNfLBzEb4UNd0-DvAj_IUy0,27672
-mcp_dbutils/config.py,sha256=K0rlNio8sjuTp-CAuY1CnG8zljp2dhPsfZY7wUhwZX4,1923
-mcp_dbutils/log.py,sha256=wKyMzB8IpKI7wsDUsN5SRfjf-bh_se6U9QhmzoNuxhw,809
-mcp_dbutils/stats.py,sha256=WYD9NAKHH2bFKmUSTOg18-SuIXergpznPq8AtIjtxdI,7101
-mcp_dbutils/mysql/__init__.py,sha256=OgwFO0gdymkvcEw-d1yqJB55XhcucmYQMEZkQ__CG_M,129
-mcp_dbutils/mysql/config.py,sha256=C977PtaRk7gJm47fV7NaPeEDGInDZNBIQKfcgvjdBbU,8069
-mcp_dbutils/mysql/handler.py,sha256=AJ1C-aVsBD3NXD31UjFC44cl4yuNeOO-HOY2mFBym0U,19737
-mcp_dbutils/mysql/server.py,sha256=lL_Q9D5AFKTKdm_sl3Eo8LC3_vKOffYpqokw4kc4IrY,8588
-mcp_dbutils/postgres/__init__.py,sha256=XanCXw-kVE6ayqFqjuLJ9swWPcCVlcKZXB2Et2Wjl9w,154
-mcp_dbutils/postgres/config.py,sha256=FG0YUV4TtP1IGZT87ov_PK-ouOEjAOHdSaK42klk5OE,7750
-mcp_dbutils/postgres/handler.py,sha256=O45K0MOosNQ68SJ8piXXH3aZ74RVQ981hudk6faj28Y,24397
-mcp_dbutils/postgres/server.py,sha256=VKiIQhEoup2ye1XX14z12lRtR2-mYXMRe4Lfso0crOg,8702
-mcp_dbutils/sqlite/__init__.py,sha256=lTUOkSfSWNw_BohJ_KNxYKMCANdtpGa3JK_ZyJsg_Ls,134
-mcp_dbutils/sqlite/config.py,sha256=2ekTp89rcCu4qQN1O3sZjIcOwjhzWi4tisD20T2WAik,4533
-mcp_dbutils/sqlite/handler.py,sha256=VxBVbwpNSQHyiudQXYFYpcdoScvbK2IGzYMbkcpvFcI,17731
-mcp_dbutils/sqlite/server.py,sha256=PBjQLvxuxMq7FhTuu5ZX1y1AR1casCbio4dp1eeX6y0,7694
-mcp_dbutils-0.14.0.dist-info/METADATA,sha256=luBQ5_SiC5LDs5aO-TBDNHqqJt7eb_Q1fC_g6opDe3o,15241
-mcp_dbutils-0.14.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_dbutils-0.14.0.dist-info/entry_points.txt,sha256=XTjt0QmYRgKOJQT6skR9bp1EMUfIrgpHeZJPZ3CJffs,49
-mcp_dbutils-0.14.0.dist-info/licenses/LICENSE,sha256=1A_CwpWVlbjrKdVEYO77vYfnXlW7oxcilZ8FpA_BzCI,1065
-mcp_dbutils-0.14.0.dist-info/RECORD,,