claude-mpm 4.1.1__py3-none-any.whl → 4.1.2__py3-none-any.whl

claude_mpm/services/diagnostics/doctor_reporter.py

@@ -7,26 +7,25 @@ formatting for terminal display and JSON export.
 
 import json
 import sys
-from typing import Dict, List, Optional
 
 from .models import DiagnosticResult, DiagnosticStatus, DiagnosticSummary
 
 
 class DoctorReporter:
     """Format and display diagnostic results.
-    
+
     WHY: Consistent, user-friendly output that clearly shows system health
     status and provides actionable fixes for any issues.
     """
-    
+
     # Status symbols and colors
     STATUS_SYMBOLS = {
         DiagnosticStatus.OK: "✅",
         DiagnosticStatus.WARNING: "⚠️ ",
         DiagnosticStatus.ERROR: "❌",
-        DiagnosticStatus.SKIPPED: "⏭️ "
+        DiagnosticStatus.SKIPPED: "⏭️ ",
     }
-    
+
     # ANSI color codes
     COLORS = {
         "reset": "\033[0m",
@@ -35,22 +34,22 @@ class DoctorReporter:
         "green": "\033[92m",
         "yellow": "\033[93m",
         "blue": "\033[94m",
-        "gray": "\033[90m"
+        "gray": "\033[90m",
     }
-    
+
     def __init__(self, use_color: bool = True, verbose: bool = False):
         """Initialize reporter.
-        
+
         Args:
             use_color: Whether to use ANSI color codes
             verbose: Whether to include detailed information
         """
         self.use_color = use_color and sys.stdout.isatty()
         self.verbose = verbose
-    
+
     def report(self, summary: DiagnosticSummary, format: str = "terminal"):
         """Generate and output diagnostic report.
-        
+
         Args:
             summary: DiagnosticSummary with all results
             format: Output format ("terminal", "json", "markdown")
@@ -61,40 +60,40 @@ class DoctorReporter:
             self._report_markdown(summary)
         else:
             self._report_terminal(summary)
-    
+
     def _report_terminal(self, summary: DiagnosticSummary):
         """Generate terminal-formatted report."""
         # Header
         self._print_header()
-        
+
         # Results by category
         for result in summary.results:
             self._print_result(result)
-        
+
         # Summary
         self._print_summary(summary)
-        
+
         # Fix suggestions
         self._print_fixes(summary)
-    
+
     def _print_header(self):
         """Print report header."""
         print()
         print(self._color("Claude MPM Doctor Report", "bold"))
         print("=" * 40)
         print()
-    
+
     def _print_result(self, result: DiagnosticResult, indent: int = 0):
         """Print a single diagnostic result."""
         indent_str = "  " * indent
-        
+
         # Status symbol and category
         symbol = self.STATUS_SYMBOLS.get(result.status, "?")
         color = self._get_status_color(result.status)
-        
+
         # Main result line
         line = f"{indent_str}{symbol} {result.category}: "
-        
+
         if result.status == DiagnosticStatus.OK:
             line += self._color("OK", color)
         elif result.status == DiagnosticStatus.WARNING:
@@ -103,115 +102,149 @@ class DoctorReporter:
             line += self._color("Error", color)
         else:
             line += self._color("Skipped", color)
-        
+
         print(line)
-        
+
         # Message
         message_indent = "   " + indent_str
         print(f"{message_indent}{result.message}")
-        
+
         # Details (in verbose mode)
         if self.verbose and result.details:
             for key, value in result.details.items():
                 if key not in ["error", "issues"]:  # Skip complex fields
                     print(f"{message_indent}{self._color(key, 'gray')}: {value}")
-        
+
         # Fix suggestion
         if result.fix_command:
             fix_indent = "   " + indent_str
-            print(f"{fix_indent}{self._color('→ Fix:', 'blue')} Run '{result.fix_command}'")
+            print(
+                f"{fix_indent}{self._color('→ Fix:', 'blue')} Run '{result.fix_command}'"
+            )
             if result.fix_description:
                 print(f"{fix_indent}  {self._color(result.fix_description, 'gray')}")
-        
+
         # Sub-results (in verbose mode)
         if self.verbose and result.sub_results:
             for sub_result in result.sub_results:
                 self._print_result(sub_result, indent + 1)
-        
+
         if indent == 0:
             print()  # Extra line between top-level results
-    
+
     def _print_summary(self, summary: DiagnosticSummary):
         """Print summary statistics."""
         print(self._color("─" * 40, "gray"))
-        
-        status_line = f"Summary: "
+
+        status_line = "Summary: "
         parts = []
-        
+
         if summary.ok_count > 0:
             parts.append(self._color(f"{summary.ok_count} OK", "green"))
         if summary.warning_count > 0:
-            parts.append(self._color(f"{summary.warning_count} Warning{'s' if summary.warning_count != 1 else ''}", "yellow"))
+            parts.append(
+                self._color(
+                    f"{summary.warning_count} Warning{'s' if summary.warning_count != 1 else ''}",
+                    "yellow",
+                )
+            )
         if summary.error_count > 0:
-            parts.append(self._color(f"{summary.error_count} Error{'s' if summary.error_count != 1 else ''}", "red"))
+            parts.append(
+                self._color(
+                    f"{summary.error_count} Error{'s' if summary.error_count != 1 else ''}",
+                    "red",
+                )
+            )
         if summary.skipped_count > 0:
             parts.append(self._color(f"{summary.skipped_count} Skipped", "gray"))
-        
+
         status_line += " | ".join(parts)
         print(status_line)
-        
+
         # Overall health
         overall = summary.overall_status
         if overall == DiagnosticStatus.OK:
             print(self._color("\n✅ System is healthy!", "green"))
         elif overall == DiagnosticStatus.WARNING:
-            print(self._color("\n⚠️  System has minor issues that should be addressed.", "yellow"))
+            print(
+                self._color(
+                    "\n⚠️  System has minor issues that should be addressed.", "yellow"
+                )
+            )
         else:
-            print(self._color("\n❌ System has critical issues that need immediate attention!", "red"))
-    
+            print(
+                self._color(
+                    "\n❌ System has critical issues that need immediate attention!",
+                    "red",
+                )
+            )
+
     def _print_fixes(self, summary: DiagnosticSummary):
         """Print consolidated fix suggestions."""
         fixes = []
-        
+
         for result in summary.results:
             if result.fix_command and result.has_issues:
-                fixes.append((result.category, result.fix_command, result.fix_description))
-        
+                fixes.append(
+                    (result.category, result.fix_command, result.fix_description)
+                )
+
         if fixes:
             print()
             print(self._color("Suggested Fixes:", "bold"))
             print(self._color("─" * 40, "gray"))
-            
+
             for i, (category, command, description) in enumerate(fixes, 1):
                 print(f"{i}. {category}:")
                 print(f"   {self._color(command, 'blue')}")
                 if description:
                     print(f"   {self._color(description, 'gray')}")
                 print()
-            
+
             if self.verbose:
-                print(self._color("Run 'claude-mpm doctor --fix' to attempt automatic fixes", "gray"))
+                print(
+                    self._color(
+                        "Run 'claude-mpm doctor --fix' to attempt automatic fixes",
+                        "gray",
+                    )
+                )
             else:
-                print(self._color("Run 'claude-mpm doctor --verbose' for more details", "gray"))
-    
+                print(
+                    self._color(
+                        "Run 'claude-mpm doctor --verbose' for more details", "gray"
+                    )
+                )
+
     def _report_json(self, summary: DiagnosticSummary):
         """Generate JSON-formatted report."""
         output = summary.to_dict()
-        
+
         # Add metadata
         output["metadata"] = {
             "tool": "claude-mpm doctor",
             "version": self._get_version(),
-            "verbose": self.verbose
+            "verbose": self.verbose,
         }
-        
+
         # Add fix suggestions
         fixes = []
         for result in summary.results:
             if result.fix_command and result.has_issues:
-                fixes.append({
-                    "category": result.category,
-                    "command": result.fix_command,
-                    "description": result.fix_description
-                })
+                fixes.append(
+                    {
+                        "category": result.category,
+                        "command": result.fix_command,
+                        "description": result.fix_description,
+                    }
+                )
         output["fixes"] = fixes
-        
+
         print(json.dumps(output, indent=2))
-    
+
     def _report_markdown(self, summary: DiagnosticSummary):
         """Generate Markdown-formatted report."""
         print("# Claude MPM Doctor Report\n")
-        
+
         # Summary table
         print("## Summary\n")
         print("| Status | Count |")
@@ -221,31 +254,34 @@ class DoctorReporter:
         print(f"| ❌ Error | {summary.error_count} |")
         print(f"| ⏭️  Skipped | {summary.skipped_count} |")
         print()
-        
+
         # Detailed results
         print("## Diagnostic Results\n")
-        
+
         for result in summary.results:
             symbol = self.STATUS_SYMBOLS.get(result.status, "?")
             print(f"### {symbol} {result.category}\n")
             print(f"**Status:** {result.status.value}")
             print(f"**Message:** {result.message}\n")
-            
+
             if result.fix_command:
                 print(f"**Fix:** `{result.fix_command}`")
                 if result.fix_description:
                     print(f"_{result.fix_description}_\n")
-            
+
             if self.verbose and result.details:
                 print("**Details:**")
                 for key, value in result.details.items():
                     print(f"- {key}: {value}")
                 print()
-        
+
         # Fixes section
-        fixes = [(r.category, r.fix_command, r.fix_description) 
-                 for r in summary.results if r.fix_command and r.has_issues]
-        
+        fixes = [
+            (r.category, r.fix_command, r.fix_description)
+            for r in summary.results
+            if r.fix_command and r.has_issues
+        ]
+
         if fixes:
             print("## Suggested Fixes\n")
             for category, command, description in fixes:
@@ -253,31 +289,32 @@ class DoctorReporter:
                 if description:
                     print(f"  - {description}")
             print()
-    
+
     def _color(self, text: str, color: str) -> str:
         """Apply color to text if colors are enabled."""
         if not self.use_color:
             return text
-        
+
         color_code = self.COLORS.get(color, "")
         reset_code = self.COLORS["reset"]
         return f"{color_code}{text}{reset_code}"
-    
+
     def _get_status_color(self, status: DiagnosticStatus) -> str:
         """Get color for a status."""
         color_map = {
             DiagnosticStatus.OK: "green",
             DiagnosticStatus.WARNING: "yellow",
             DiagnosticStatus.ERROR: "red",
-            DiagnosticStatus.SKIPPED: "gray"
+            DiagnosticStatus.SKIPPED: "gray",
         }
         return color_map.get(status, "reset")
-    
+
     def _get_version(self) -> str:
         """Get claude-mpm version."""
         try:
             from ..version_service import VersionService
+
             service = VersionService()
             return service.get_version()
         except Exception:
-            return "unknown"
+            return "unknown"

claude_mpm/services/diagnostics/models.py

@@ -12,6 +12,7 @@ from typing import Any, Dict, List, Optional
 
 class DiagnosticStatus(Enum):
     """Status levels for diagnostic results."""
+
     OK = "ok"
     WARNING = "warning"
     ERROR = "error"
@@ -21,18 +22,19 @@ class DiagnosticStatus(Enum):
 @dataclass
 class DiagnosticResult:
     """Result from a diagnostic check.
-    
+
     WHY: Standardized result format ensures consistent reporting
     and makes it easy to aggregate and display results.
     """
+
     category: str  # e.g., "Installation", "Agents", "MCP Server"
     status: DiagnosticStatus
     message: str
     details: Dict[str, Any] = field(default_factory=dict)
     fix_command: Optional[str] = None
     fix_description: Optional[str] = None
-    sub_results: List['DiagnosticResult'] = field(default_factory=list)
-    
+    sub_results: List["DiagnosticResult"] = field(default_factory=list)
+
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dictionary for JSON serialization."""
         return {
@@ -42,14 +44,14 @@ class DiagnosticResult:
             "details": self.details,
             "fix_command": self.fix_command,
             "fix_description": self.fix_description,
-            "sub_results": [r.to_dict() for r in self.sub_results]
+            "sub_results": [r.to_dict() for r in self.sub_results],
         }
-    
+
     @property
     def has_issues(self) -> bool:
         """Check if this result indicates any issues."""
         return self.status in (DiagnosticStatus.WARNING, DiagnosticStatus.ERROR)
-    
+
     @property
     def severity_level(self) -> int:
         """Get numeric severity level for sorting."""
@@ -57,7 +59,7 @@ class DiagnosticResult:
             DiagnosticStatus.OK: 0,
             DiagnosticStatus.SKIPPED: 1,
             DiagnosticStatus.WARNING: 2,
-            DiagnosticStatus.ERROR: 3
+            DiagnosticStatus.ERROR: 3,
         }
         return severity_map.get(self.status, 0)
 
@@ -65,22 +67,23 @@ class DiagnosticResult:
 @dataclass
 class DiagnosticSummary:
     """Summary of all diagnostic results.
-    
+
     WHY: Provides a high-level overview of system health
     and quick access to issues that need attention.
     """
+
     total_checks: int = 0
     ok_count: int = 0
     warning_count: int = 0
     error_count: int = 0
     skipped_count: int = 0
     results: List[DiagnosticResult] = field(default_factory=list)
-    
+
     def add_result(self, result: DiagnosticResult):
         """Add a result to the summary."""
         self.results.append(result)
         self.total_checks += 1
-        
+
         if result.status == DiagnosticStatus.OK:
             self.ok_count += 1
         elif result.status == DiagnosticStatus.WARNING:
@@ -89,22 +92,21 @@ class DiagnosticSummary:
             self.error_count += 1
         elif result.status == DiagnosticStatus.SKIPPED:
             self.skipped_count += 1
-    
+
     @property
     def has_issues(self) -> bool:
         """Check if there are any warnings or errors."""
         return self.warning_count > 0 or self.error_count > 0
-    
+
     @property
     def overall_status(self) -> DiagnosticStatus:
         """Get overall system status."""
         if self.error_count > 0:
             return DiagnosticStatus.ERROR
-        elif self.warning_count > 0:
+        if self.warning_count > 0:
             return DiagnosticStatus.WARNING
-        else:
-            return DiagnosticStatus.OK
-    
+        return DiagnosticStatus.OK
+
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dictionary for JSON serialization."""
         return {
@@ -114,7 +116,7 @@ class DiagnosticSummary:
                 "warnings": self.warning_count,
                 "errors": self.error_count,
                 "skipped": self.skipped_count,
-                "overall_status": self.overall_status.value
+                "overall_status": self.overall_status.value,
             },
-            "results": [r.to_dict() for r in self.results]
-        }
+            "results": [r.to_dict() for r in self.results],
+        }

claude_mpm/services/event_aggregator.py

@@ -13,7 +13,6 @@ aggregator to run alongside the dashboard without any conflicts.
 
 import asyncio
 import json
-import os
 import signal
 import sys
 import threading
@@ -30,8 +29,10 @@ except ImportError:
     SOCKETIO_AVAILABLE = False
     socketio = None
 
+import contextlib
+
 from ..core.logger import get_logger
-from ..models.agent_session import AgentSession, EventCategory
+from ..models.agent_session import AgentSession
 
 
 class EventAggregator:
@@ -61,7 +62,6 @@ class EventAggregator:
         self.logger = get_logger("event_aggregator")
 
         # Load configuration using ConfigLoader
-        from claude_mpm.core.config import Config
         from claude_mpm.core.shared.config_loader import ConfigLoader
 
         config_loader = ConfigLoader()
@@ -136,10 +136,9 @@ class EventAggregator:
         if self.connected:
             self.logger.info("Event Aggregator started successfully")
             return True
-        else:
-            self.logger.error("Failed to connect to Socket.IO server")
-            self.running = False
-            return False
+        self.logger.error("Failed to connect to Socket.IO server")
+        self.running = False
+        return False
 
     def stop(self):
         """Stop the aggregator service."""
@@ -151,12 +150,10 @@ class EventAggregator:
 
         # Disconnect Socket.IO client
         if self.sio_client and self.connected:
-            try:
+            with contextlib.suppress(Exception):
                 asyncio.run_coroutine_threadsafe(
                     self.sio_client.disconnect(), self.client_loop
                 ).result(timeout=2)
-            except:
-                pass
 
         # Stop the client thread
         if self.client_thread and self.client_thread.is_alive():
@@ -207,10 +204,8 @@ class EventAggregator:
             # Cancel cleanup task
             if self.cleanup_task:
                 self.cleanup_task.cancel()
-                try:
+                with contextlib.suppress(asyncio.CancelledError):
                     await self.cleanup_task
-                except asyncio.CancelledError:
-                    pass
 
         except Exception as e:
             self.logger.error(f"Connection error: {e}")
@@ -470,9 +465,7 @@ class EventAggregator:
             "sessions_completed": self.sessions_completed,
             "total_events": self.total_events_captured,
             "events_by_type": dict(self.events_by_type),
-            "active_session_ids": [
-                sid[:8] + "..." for sid in self.active_sessions.keys()
-            ],
+            "active_session_ids": [sid[:8] + "..." for sid in self.active_sessions],
         }
 
     def list_sessions(self, limit: int = 10) -> List[Dict[str, Any]]:
@@ -496,7 +489,7 @@ class EventAggregator:
         for filepath in session_files:
             try:
                 # Load just the metadata, not the full session
-                with open(filepath, "r") as f:
+                with open(filepath) as f:
                     data = json.load(f)
 
                 sessions.append(

claude_mpm/services/event_bus/__init__.py

@@ -15,4 +15,4 @@ WHY event bus architecture:
 from .event_bus import EventBus
 from .relay import SocketIORelay
 
-__all__ = ["EventBus", "SocketIORelay"]
+__all__ = ["EventBus", "SocketIORelay"]