mcp-dbutils 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

mcp_dbutils/postgres/config.py

@@ -1,8 +1,40 @@
 """PostgreSQL configuration module"""
 from dataclasses import dataclass
 from typing import Optional, Dict, Any, Literal
+from urllib.parse import urlparse
 from ..config import DatabaseConfig
 
+def parse_jdbc_url(jdbc_url: str) -> Dict[str, str]:
+    """Parse JDBC URL into connection parameters
+    
+    Args:
+        jdbc_url: JDBC URL (e.g. jdbc:postgresql://host:port/dbname)
+        
+    Returns:
+        Dictionary of connection parameters
+    """
+    if not jdbc_url.startswith('jdbc:postgresql://'):
+        raise ValueError("Invalid PostgreSQL JDBC URL format")
+    
+    # Remove jdbc: prefix and ensure no credentials in URL
+    url = jdbc_url[5:]
+    if '@' in url:
+        raise ValueError("JDBC URL should not contain credentials. Please provide username and password separately.")
+    
+    # Parse URL
+    parsed = urlparse(url)
+    
+    params = {
+        'host': parsed.hostname or 'localhost',
+        'port': str(parsed.port or 5432),
+        'dbname': parsed.path.lstrip('/') if parsed.path else '',
+    }
+    
+    if not params['dbname']:
+        raise ValueError("Database name must be specified in URL")
+        
+    return params
+
 @dataclass
 class PostgresConfig(DatabaseConfig):
     dbname: str
@@ -35,12 +67,64 @@ class PostgresConfig(DatabaseConfig):
         if db_config['type'] != 'postgres':
             raise ValueError(f"Configuration is not PostgreSQL type: {db_config['type']}")
 
+        # Check required credentials
+        if not db_config.get('user'):
+            raise ValueError("User must be specified in database configuration")
+        if not db_config.get('password'):
+            raise ValueError("Password must be specified in database configuration")
+
+        # Get connection parameters
+        if 'jdbc_url' in db_config:
+            # Parse JDBC URL for connection parameters
+            params = parse_jdbc_url(db_config['jdbc_url'])
+            config = cls(
+                dbname=params['dbname'],
+                user=db_config['user'],
+                password=db_config['password'],
+                host=params['host'],
+                port=params['port'],
+                local_host=local_host,
+            )
+        else:
+            if not db_config.get('dbname'):
+                raise ValueError("Database name must be specified in configuration")
+            if not db_config.get('host'):
+                raise ValueError("Host must be specified in configuration")
+            if not db_config.get('port'):
+                raise ValueError("Port must be specified in configuration")
+            config = cls(
+                dbname=db_config['dbname'],
+                user=db_config['user'],
+                password=db_config['password'],
+                host=db_config['host'],
+                port=str(db_config['port']),
+                local_host=local_host,
+            )
+        config.debug = cls.get_debug_mode()
+        return config
+
+    @classmethod
+    def from_jdbc_url(cls, jdbc_url: str, user: str, password: str, 
+                     local_host: Optional[str] = None) -> 'PostgresConfig':
+        """Create configuration from JDBC URL and credentials
+        
+        Args:
+            jdbc_url: JDBC URL (jdbc:postgresql://host:port/dbname)
+            user: Username for database connection
+            password: Password for database connection
+            local_host: Optional local host address
+            
+        Raises:
+            ValueError: If URL format is invalid or required parameters are missing
+        """
+        params = parse_jdbc_url(jdbc_url)
+        
         config = cls(
-            dbname=db_config.get('dbname', ''),
-            user=db_config.get('user', ''),
-            password=db_config.get('password', ''),
-            host=db_config.get('host', 'localhost'),
-            port=str(db_config.get('port', 5432)),
+            dbname=params['dbname'],
+            user=user,
+            password=password,
+            host=params['host'],
+            port=params['port'],
             local_host=local_host,
         )
         config.debug = cls.get_debug_mode()

{mcp_dbutils-0.3.0.dist-info → mcp_dbutils-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp-dbutils
-Version: 0.3.0
+Version: 0.4.0
 Summary: MCP Database Utilities Service
 Author: Dong Hao
 License-Expression: MIT

{mcp_dbutils-0.3.0.dist-info → mcp_dbutils-0.4.0.dist-info}/RECORD

@@ -4,15 +4,15 @@ 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
 mcp_dbutils/postgres/__init__.py,sha256=Y6v_RsI79pqAfpKM3SrT1T1I9r5yWuKT0GUUNmHD3DE,146
-mcp_dbutils/postgres/config.py,sha256=ipjskdlv3u949R9rC2AKPCINRrtEAiKDjeogpFfM6aE,2401
+mcp_dbutils/postgres/config.py,sha256=Np0GS5iUaUuWpYI8QfnyjUDy7v-vResQEAexR4W92-o,5447
 mcp_dbutils/postgres/handler.py,sha256=JC_Qyw1s6qjVR0tdH7FLAEAs9EPl_-wgFQLryQyMq5c,6089
 mcp_dbutils/postgres/server.py,sha256=_S3HF1KooxPB9gX1FedoOOGn93tHlIevCab6vjCt2TU,8715
 mcp_dbutils/sqlite/__init__.py,sha256=QV4th2ywzUmCCa3GHCcBf8blJ_E8OYy0dJ2fSf1TfSU,134
 mcp_dbutils/sqlite/config.py,sha256=QK1JlY-ZBB5VyhydbiU-aAKNESTWMtyBfawbv2DAVCQ,2452
 mcp_dbutils/sqlite/handler.py,sha256=bf_k93rCcJn09zc7tsqrlbiTGUg3FspimfWKxK_JQTs,4970
 mcp_dbutils/sqlite/server.py,sha256=7Bbq9l7Ca_4dzkAbbdRcXxvHoO_NFLzZHwlhKB0HIJc,7724
-mcp_dbutils-0.3.0.dist-info/METADATA,sha256=hy1Xl8rcvFAxmLRq_FHP8Ow1LM_x6Ujz5NO_TkzUtXY,9478
-mcp_dbutils-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_dbutils-0.3.0.dist-info/entry_points.txt,sha256=XTjt0QmYRgKOJQT6skR9bp1EMUfIrgpHeZJPZ3CJffs,49
-mcp_dbutils-0.3.0.dist-info/licenses/LICENSE,sha256=1A_CwpWVlbjrKdVEYO77vYfnXlW7oxcilZ8FpA_BzCI,1065
-mcp_dbutils-0.3.0.dist-info/RECORD,,
+mcp_dbutils-0.4.0.dist-info/METADATA,sha256=1c8QEdqJmgHUCjsMxJEMh1C5Qdpq0EdXWAO07yIv77k,9478
+mcp_dbutils-0.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_dbutils-0.4.0.dist-info/entry_points.txt,sha256=XTjt0QmYRgKOJQT6skR9bp1EMUfIrgpHeZJPZ3CJffs,49
+mcp_dbutils-0.4.0.dist-info/licenses/LICENSE,sha256=1A_CwpWVlbjrKdVEYO77vYfnXlW7oxcilZ8FpA_BzCI,1065
+mcp_dbutils-0.4.0.dist-info/RECORD,,