mcp-proxy-adapter 6.3.3__py3-none-any.whl → 6.3.5__py3-none-any.whl

mcp_proxy_adapter/commands/transport_management_command.py

@@ -13,11 +13,15 @@ from mcp_proxy_adapter.core.logging import logger
 
 class TransportManagementResult(SuccessResult):
     """Result class for transport management operations."""
-    
-    def __init__(self, transport_info: Dict[str, Any], message: str = "Transport management operation completed"):
+
+    def __init__(
+        self,
+        transport_info: Dict[str, Any],
+        message: str = "Transport management operation completed",
+    ):
         """
         Initialize transport management result.
-        
+
         Args:
             transport_info: Transport information
             message: Success message
@@ -28,18 +32,18 @@ class TransportManagementResult(SuccessResult):
 class TransportManagementCommand(Command):
     """
     Transport management command.
-    
+
     This command provides functionality to manage and query transport configurations.
     """
-    
+
     name = "transport_management"
     descr = "Manage and query transport configurations (HTTP, HTTPS, MTLS)"
-    
+
     @classmethod
     def get_schema(cls) -> Dict[str, Any]:
         """
         Get command schema.
-        
+
         Returns:
             Command schema dictionary
         """
@@ -49,25 +53,25 @@ class TransportManagementCommand(Command):
                 "action": {
                     "type": "string",
                     "enum": ["get_info", "validate", "reload"],
-                    "description": "Action to perform"
+                    "description": "Action to perform",
                 }
             },
-            "required": ["action"]
+            "required": ["action"],
         }
-    
+
     async def execute(self, **params) -> TransportManagementResult:
         """
         Execute transport management command.
-        
+
         Args:
             params: Command parameters
-            
+
         Returns:
             Transport management result
         """
         try:
             action = params.get("action", "get_info")
-            
+
             if action == "get_info":
                 return await self._get_transport_info()
             elif action == "validate":
@@ -77,56 +81,57 @@ class TransportManagementCommand(Command):
             else:
                 return TransportManagementResult(
                     transport_info={"error": f"Unknown action: {action}"},
-                    message=f"Unknown action: {action}"
+                    message=f"Unknown action: {action}",
                 )
-                
+
         except Exception as e:
             logger.error(f"Transport management command error: {e}")
             return TransportManagementResult(
                 transport_info={"error": str(e)},
-                message=f"Transport management failed: {e}"
+                message=f"Transport management failed: {e}",
             )
-    
+
     async def _get_transport_info(self) -> TransportManagementResult:
         """
         Get transport information.
-        
+
         Returns:
             Transport information result
         """
         transport_info = transport_manager.get_transport_info()
-        
+
         return TransportManagementResult(
             transport_info=transport_info,
-            message="Transport information retrieved successfully"
+            message="Transport information retrieved successfully",
         )
-    
+
     async def _validate_transport(self) -> TransportManagementResult:
         """
         Validate transport configuration.
-        
+
         Returns:
             Validation result
         """
         is_valid = transport_manager.validate_config()
-        
+
         transport_info = transport_manager.get_transport_info()
         transport_info["validation"] = {
             "is_valid": is_valid,
-            "timestamp": "2025-08-15T12:00:00Z"
+            "timestamp": "2025-08-15T12:00:00Z",
         }
-        
-        message = "Transport configuration validated successfully" if is_valid else "Transport configuration validation failed"
-        
-        return TransportManagementResult(
-            transport_info=transport_info,
-            message=message
+
+        message = (
+            "Transport configuration validated successfully"
+            if is_valid
+            else "Transport configuration validation failed"
         )
-    
+
+        return TransportManagementResult(transport_info=transport_info, message=message)
+
     async def _reload_transport(self) -> TransportManagementResult:
         """
         Reload transport configuration.
-        
+
         Returns:
             Reload result
         """
@@ -135,10 +140,10 @@ class TransportManagementCommand(Command):
         transport_info = transport_manager.get_transport_info()
         transport_info["reload"] = {
             "status": "completed",
-            "timestamp": "2025-08-15T12:00:00Z"
+            "timestamp": "2025-08-15T12:00:00Z",
         }
-        
+
         return TransportManagementResult(
             transport_info=transport_info,
-            message="Transport configuration reload completed"
-        ) 
+            message="Transport configuration reload completed",
+        )

mcp_proxy_adapter/commands/unload_command.py

@@ -13,31 +13,34 @@ class UnloadResult(SuccessResult):
     """
     Result of the unload command execution.
     """
-    
-    def __init__(self, success: bool, command_name: str, message: str, error: Optional[str] = None):
+
+    def __init__(
+        self,
+        success: bool,
+        command_name: str,
+        message: str,
+        error: Optional[str] = None,
+    ):
         """
         Initialize unload command result.
-        
+
         Args:
             success: Whether unloading was successful
             command_name: Name of the command that was unloaded
             message: Result message
             error: Error message if unloading failed
         """
-        data = {
-            "success": success,
-            "command_name": command_name
-        }
+        data = {"success": success, "command_name": command_name}
         if error:
             data["error"] = error
-            
+
         super().__init__(data=data, message=message)
-    
+
     @classmethod
     def get_schema(cls) -> Dict[str, Any]:
         """
         Get JSON schema for result validation.
-        
+
         Returns:
             Dict[str, Any]: JSON schema
         """
@@ -49,69 +52,69 @@ class UnloadResult(SuccessResult):
                     "properties": {
                         "success": {"type": "boolean"},
                         "command_name": {"type": "string"},
-                        "error": {"type": "string"}
+                        "error": {"type": "string"},
                     },
-                    "required": ["success", "command_name"]
+                    "required": ["success", "command_name"],
                 }
             },
-            "required": ["data"]
+            "required": ["data"],
         }
 
 
 class UnloadCommand(Command):
     """
     Command that unloads loaded commands from registry.
-    
+
     This command allows removal of dynamically loaded commands from the command registry.
     Only commands that were loaded via the 'load' command or from the commands directory
     can be unloaded. Built-in commands and custom commands registered with higher priority
     cannot be unloaded using this command.
-    
+
     When a command is unloaded:
     - The command class is removed from the loaded commands registry
     - Any command instances are also removed
     - The command becomes unavailable for execution
     - Built-in and custom commands with the same name remain unaffected
-    
+
     This is useful for:
     - Removing outdated or problematic commands
     - Managing memory usage by unloading unused commands
     - Testing different versions of commands
     - Cleaning up temporary commands loaded for testing
-    
+
     Note: Unloading a command does not affect other commands and does not require
     a system restart. The command can be reloaded later if needed.
     """
-    
+
     name = "unload"
     result_class = UnloadResult
-    
+
     async def execute(self, command_name: str, **kwargs) -> UnloadResult:
         """
         Execute unload command.
-        
+
         Args:
             command_name: Name of the command to unload
             **kwargs: Additional parameters
-            
+
         Returns:
             UnloadResult: Unload command result
         """
         # Unload command from registry
         result = registry.unload_command(command_name)
-        
+
         return UnloadResult(
             success=result.get("success", False),
             command_name=result.get("command_name", command_name),
             message=result.get("message", "Unknown result"),
-            error=result.get("error")
+            error=result.get("error"),
         )
-    
+
     @classmethod
     def get_schema(cls) -> Dict[str, Any]:
         """
         Get JSON schema for command parameters.
-        
+
         Returns:
             Dict[str, Any]: JSON schema
         """
@@ -120,20 +123,22 @@ class UnloadCommand(Command):
             "properties": {
                 "command_name": {
                     "type": "string",
-                    "description": "Name of the command to unload (must be a loaded command)"
+                    "description": "Name of the command to unload (must be a loaded command)",
                 }
             },
-            "required": ["command_name"]
+            "required": ["command_name"],
         }
-    
+
     @classmethod
-    def _generate_examples(cls, params: Dict[str, Dict[str, Any]]) -> List[Dict[str, Any]]:
+    def _generate_examples(
+        cls, params: Dict[str, Dict[str, Any]]
+    ) -> List[Dict[str, Any]]:
         """
         Generate custom examples for unload command.
-        
+
         Args:
             params: Information about command parameters
-            
+
         Returns:
             List of examples
         """
@@ -141,18 +146,18 @@ class UnloadCommand(Command):
             {
                 "command": cls.name,
                 "params": {"command_name": "test_command"},
-                "description": "Unload a previously loaded test command"
+                "description": "Unload a previously loaded test command",
             },
             {
                 "command": cls.name,
                 "params": {"command_name": "remote_command"},
-                "description": "Unload a command that was loaded from URL"
+                "description": "Unload a command that was loaded from URL",
             },
             {
                 "command": cls.name,
                 "params": {"command_name": "custom_command"},
-                "description": "Unload a custom command loaded from local file"
-            }
+                "description": "Unload a custom command loaded from local file",
+            },
         ]
-        
-        return examples 
+
+        return examples

mcp_proxy_adapter/config.py

@@ -39,7 +39,7 @@ class Config:
                 "host": "0.0.0.0",
                 "port": 8000,
                 "debug": False,
-                "log_level": "INFO"
+                "log_level": "INFO",
             },
             "logging": {
                 "level": "INFO",
@@ -53,7 +53,7 @@ class Config:
                 "format": "%(asctime)s - %(name)s - %(levelname)s - %(message)s",
                 "date_format": "%Y-%m-%d %H:%M:%S",
                 "console_output": True,
-                "file_output": True
+                "file_output": True,
             },
             "commands": {
                 "auto_discovery": True,
@@ -63,7 +63,7 @@ class Config:
                 "auto_install_dependencies": True,
                 "enabled_commands": ["health", "echo", "list", "help"],
                 "disabled_commands": [],
-                "custom_commands_path": "./commands"
+                "custom_commands_path": "./commands",
             },
             "ssl": {
                 "enabled": False,
@@ -75,7 +75,7 @@ class Config:
                 "client_cert_required": False,
                 "cipher_suites": [
                     "TLS_AES_256_GCM_SHA384",
-                    "TLS_CHACHA20_POLY1305_SHA256"
+                    "TLS_CHACHA20_POLY1305_SHA256",
                 ],
                 "min_tls_version": "TLSv1.2",
                 "max_tls_version": "1.3",
@@ -86,8 +86,8 @@ class Config:
                     "tokens_file": "tokens.json",
                     "token_expiry": 3600,
                     "jwt_secret": "",
-                    "jwt_algorithm": "HS256"
-                }
+                    "jwt_algorithm": "HS256",
+                },
             },
             "roles": {
                 "enabled": False,
@@ -96,10 +96,10 @@ class Config:
                     "deny_by_default": False,
                     "require_role_match": False,
                     "case_sensitive": False,
-                    "allow_wildcard": False
+                    "allow_wildcard": False,
                 },
                 "auto_load": False,
-                "validation_enabled": False
+                "validation_enabled": False,
             },
             "transport": {
                 "type": "http",
@@ -110,8 +110,8 @@ class Config:
                     "key_file": None,
                     "ca_cert": None,
                     "verify_client": False,
-                    "client_cert_required": False
-                }
+                    "client_cert_required": False,
+                },
             },
             "proxy_registration": {
                 "enabled": False,
@@ -124,12 +124,9 @@ class Config:
                 "retry_attempts": 3,
                 "retry_delay": 5,
                 "auto_register_on_startup": True,
-                "auto_unregister_on_shutdown": True
-            },
-            "debug": {
-                "enabled": False,
-                "level": "WARNING"
+                "auto_unregister_on_shutdown": True,
             },
+            "debug": {"enabled": False, "level": "WARNING"},
             "security": {
                 "framework": "mcp_security_framework",
                 "enabled": False,
@@ -150,7 +147,7 @@ class Config:
                     "basic_auth": False,
                     "oauth2_config": None,
                     "public_paths": ["/health", "/docs", "/openapi.json"],
-                    "security_headers": None
+                    "security_headers": None,
                 },
                 "ssl": {
                     "enabled": False,
@@ -165,7 +162,7 @@ class Config:
                     "cipher_suite": None,
                     "check_hostname": True,
                     "check_expiry": True,
-                    "expiry_warning_days": 30
+                    "expiry_warning_days": 30,
                 },
                 "certificates": {
                     "enabled": False,
@@ -181,7 +178,7 @@ class Config:
                     "crl_url": None,
                     "crl_validity_days": 30,
                     "auto_renewal": False,
-                    "renewal_threshold_days": 30
+                    "renewal_threshold_days": 30,
                 },
                 "permissions": {
                     "enabled": False,
@@ -193,7 +190,7 @@ class Config:
                     "permission_cache_ttl": 300,
                     "wildcard_permissions": False,
                     "strict_mode": False,
-                    "roles": None
+                    "roles": None,
                 },
                 "rate_limit": {
                     "enabled": False,
@@ -205,7 +202,7 @@ class Config:
                     "redis_config": None,
                     "cleanup_interval": 300,
                     "exempt_paths": ["/health", "/docs", "/openapi.json"],
-                    "exempt_roles": ["admin"]
+                    "exempt_roles": ["admin"],
                 },
                 "logging": {
                     "enabled": True,
@@ -219,21 +216,21 @@ class Config:
                     "json_format": False,
                     "include_timestamp": True,
                     "include_level": True,
-                    "include_module": True
-                }
+                    "include_module": True,
+                },
             },
             "protocols": {
                 "enabled": True,
                 "allowed_protocols": ["http", "jsonrpc"],
                 "default_protocol": "http",
-                "auto_discovery": True
-            }
+                "auto_discovery": True,
+            },
         }
 
         # Try to load configuration from file
         if os.path.exists(self.config_path):
             try:
-                with open(self.config_path, 'r', encoding='utf-8') as f:
+                with open(self.config_path, "r", encoding="utf-8") as f:
                     file_config = json.load(f)
                     self._update_nested_dict(self.config_data, file_config)
             except Exception as e:
@@ -261,7 +258,7 @@ class Config:
         prefix = "SERVICE_"
         for key, value in os.environ.items():
             if key.startswith(prefix):
-                parts = key[len(prefix):].lower().split("_", 1)
+                parts = key[len(prefix) :].lower().split("_", 1)
                 if len(parts) == 2:
                     section, param = parts
                     if section not in self.config_data:
@@ -357,7 +354,7 @@ class Config:
                   self.config_path is used.
         """
         save_path = path or self.config_path
-        with open(save_path, 'w', encoding='utf-8') as f:
+        with open(save_path, "w", encoding="utf-8") as f:
             json.dump(self.config_data, f, indent=2)
 
     def _update_nested_dict(self, d: Dict, u: Dict) -> Dict:
@@ -444,13 +441,15 @@ class Config:
             True if feature is enabled, False otherwise
         """
         if feature == "ssl":
-            return (self.get("ssl.enabled", False) or
-                    self.get("security.ssl.enabled", False))
+            return self.get("ssl.enabled", False) or self.get(
+                "security.ssl.enabled", False
+            )
         elif feature == "auth":
             return self.get("security.auth.enabled", False)
         elif feature == "roles":
-            return (self.get("security.permissions.enabled", False) or
-                    self.get("roles.enabled", False))
+            return self.get("security.permissions.enabled", False) or self.get(
+                "roles.enabled", False
+            )
         elif feature == "proxy_registration":
             return self.get("proxy_registration.enabled", False)
         elif feature == "security":
@@ -535,11 +534,13 @@ class Config:
             elif mode == "certificate":
                 self.set("proxy_registration.auth_method", "certificate")
                 if "cert_file" in kwargs:
-                    self.set("proxy_registration.certificate.cert_file",
-                             kwargs["cert_file"])
+                    self.set(
+                        "proxy_registration.certificate.cert_file", kwargs["cert_file"]
+                    )
                 if "key_file" in kwargs:
-                    self.set("proxy_registration.certificate.key_file",
-                             kwargs["key_file"])
+                    self.set(
+                        "proxy_registration.certificate.key_file", kwargs["key_file"]
+                    )
             elif mode == "api_key":
                 self.set("proxy_registration.auth_method", "api_key")
                 if "key" in kwargs:

mcp_proxy_adapter/core/__init__.py

@@ -12,7 +12,6 @@ __all__ = [
     "InvalidParamsError",
     "CommandExecutionError",
     "ConfigurationError",
-    
     # Logging
     "setup_logging",
     "get_logger",
@@ -20,25 +19,24 @@ __all__ = [
     "RequestLogger",
     "CustomFormatter",
     "RequestContextFilter",
-    
     # Settings
-"Settings",
-"ServerSettings",
-"LoggingSettings",
-"CommandsSettings",
-"get_server_host",
-"get_server_port",
-"get_server_debug",
-"get_logging_level",
-"get_logging_dir",
-"get_auto_discovery",
-"get_discovery_path",
-"get_setting",
-"set_setting",
-"reload_settings",
-"add_custom_settings",
-"get_custom_settings",
-"get_custom_setting_value",
-"set_custom_setting_value",
-"clear_custom_settings"
+    "Settings",
+    "ServerSettings",
+    "LoggingSettings",
+    "CommandsSettings",
+    "get_server_host",
+    "get_server_port",
+    "get_server_debug",
+    "get_logging_level",
+    "get_logging_dir",
+    "get_auto_discovery",
+    "get_discovery_path",
+    "get_setting",
+    "set_setting",
+    "reload_settings",
+    "add_custom_settings",
+    "get_custom_settings",
+    "get_custom_setting_value",
+    "set_custom_setting_value",
+    "clear_custom_settings",
 ]