claude-mpm 4.1.26__py3-none-any.whl → 5.0.9__py3-none-any.whl

claude_mpm/cli/commands/aggregate.py

@@ -16,6 +16,7 @@ import json
 import sys
 from typing import Optional
 
+from ...core.enums import OutputFormat
 from ...core.logger import get_logger
 from ...services.event_aggregator import (
     aggregator_status,
@@ -110,7 +111,10 @@ def aggregate_command(args):
     result = command.execute(args)
 
     # Print result if structured output format is requested
-    if hasattr(args, "format") and args.format in ["json", "yaml"]:
+    if hasattr(args, "format") and str(args.format).lower() in (
+        OutputFormat.JSON,
+        OutputFormat.YAML,
+    ):
         command.print_result(result, args)
 
     return result.exit_code
@@ -361,7 +365,7 @@ def view_command_legacy(args):
             print(f"   Tools: {len(delegation.tool_operations)}")
             print(f"   Files: {len(delegation.file_changes)}")
             if delegation.duration_ms:
-                print(f"   Duration: {delegation.duration_ms/1000:.1f}s")
+                print(f"   Duration: {delegation.duration_ms / 1000:.1f}s")
             print(f"   Status: {'✅ Success' if delegation.success else '❌ Failed'}")
             if delegation.error:
                 print(f"   Error: {delegation.error}")
@@ -407,9 +411,9 @@ def export_command_legacy(args):
         output_path = Path(f"session_{session.session_id[:8]}_export.json")
 
     # Export based on format
-    if args.format == "json":
+    if str(args.format).lower() == OutputFormat.JSON:
         # Full JSON export
-        with open(output_path, "w") as f:
+        with output_path.open("w") as f:
             json.dump(session.to_dict(), f, indent=2)
         print(f"✅ Exported session to {output_path}")
 
@@ -436,7 +440,7 @@ def export_command_legacy(args):
             ],
         }
 
-        with open(output_path, "w") as f:
+        with output_path.open("w") as f:
             json.dump(summary, f, indent=2)
         print(f"✅ Exported session summary to {output_path}")
 
@@ -444,7 +448,7 @@ def export_command_legacy(args):
         # Events-only export
         events_data = [e.to_dict() for e in session.events]
 
-        with open(output_path, "w") as f:
+        with output_path.open("w") as f:
             json.dump(events_data, f, indent=2)
         print(f"✅ Exported {len(events_data)} events to {output_path}")
 
@@ -529,7 +533,7 @@ def add_aggregate_parser(subparsers):
         "--format",
         "-f",
         choices=["json", "summary", "events"],
-        default="json",
+        default=OutputFormat.JSON,
         help="Export format (default: json)",
     )
 

claude_mpm/cli/commands/analyze.py

@@ -19,10 +19,11 @@ import os
 import re
 import subprocess
 import sys
-from datetime import datetime
+from datetime import datetime, timezone
 from pathlib import Path
 from typing import Dict, List, Optional
 
+from ...core.enums import OutputFormat
 from ...core.logging_config import get_logger
 from ...services.cli.session_manager import SessionManager
 from ..shared import BaseCommand, CommandResult
@@ -132,7 +133,11 @@ class AnalyzeCommand(BaseCommand):
             self._save_output(output, args.output)
 
         return CommandResult.success_result(
-            message=output if args.format == "text" else "Analysis completed",
+            message=(
+                output
+                if str(args.format).lower() == OutputFormat.TEXT
+                else "Analysis completed"
+            ),
             data=result_data,
         )
 
@@ -345,7 +350,7 @@ class AnalyzeCommand(BaseCommand):
 
         for i, match in enumerate(matches):
             # Try to extract title from preceding line
-            title = f"diagram_{i+1}"
+            title = f"diagram_{i + 1}"
 
             # Look for a title pattern before the diagram
             title_pattern = r"(?:#+\s*)?([^\n]+)\n+```mermaid"
@@ -372,14 +377,14 @@ class AnalyzeCommand(BaseCommand):
         diagram_dir = args.diagram_output or Path("./diagrams")
         diagram_dir.mkdir(parents=True, exist_ok=True)
 
-        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        timestamp = datetime.now(timezone.utc).strftime("%Y%m%d_%H%M%S")
 
         for diagram in diagrams:
             filename = f"{timestamp}_{diagram['title']}.mermaid"
             filepath = diagram_dir / filename
 
             try:
-                with open(filepath, "w") as f:
+                with filepath.open("w") as f:
                     # Write mermaid header comment
                     f.write("// Generated by Claude MPM Code Analyzer\n")
                     f.write(f"// Timestamp: {timestamp}\n")
@@ -422,14 +427,14 @@ class AnalyzeCommand(BaseCommand):
         Returns:
             Formatted output string
         """
-        if format_type == "json":
+        if str(format_type).lower() == OutputFormat.JSON:
             result_data["diagrams"] = diagrams
             return json.dumps(result_data, indent=2)
 
         if format_type == "markdown":
             output = "# Code Analysis Report\n\n"
             output += f"**Target:** `{result_data['target']}`\n"
-            output += f"**Timestamp:** {datetime.now().isoformat()}\n"
+            output += f"**Timestamp:** {datetime.now(timezone.utc).isoformat()}\n"
 
             if result_data.get("session_id"):
                 output += f"**Session ID:** {result_data['session_id']}\n"
@@ -451,7 +456,7 @@ class AnalyzeCommand(BaseCommand):
             return output
 
         # text format
-        output = f"Code Analysis Report\n{'='*50}\n\n"
+        output = f"Code Analysis Report\n{'=' * 50}\n\n"
         output += f"Target: {result_data['target']}\n"
 
         if diagrams:
@@ -464,7 +469,7 @@ class AnalyzeCommand(BaseCommand):
             for filepath in result_data["saved_diagrams"]:
                 output += f"  • {filepath}\n"
 
-        output += f"\n{'-'*50}\nAnalysis Results:\n{'-'*50}\n"
+        output += f"\n{'-' * 50}\nAnalysis Results:\n{'-' * 50}\n"
         output += result_data.get("analysis", "No analysis results")
 
         return output
@@ -478,7 +483,7 @@ class AnalyzeCommand(BaseCommand):
         """
         try:
             filepath.parent.mkdir(parents=True, exist_ok=True)
-            with open(filepath, "w") as f:
+            with filepath.open("w") as f:
                 f.write(content)
             self.logger.info(f"Saved output to {filepath}")
         except Exception as e:
@@ -492,7 +497,7 @@ class AnalyzeCommand(BaseCommand):
         """
         session_data = {
             "context": "code_analysis",
-            "created_at": datetime.now().isoformat(),
+            "created_at": datetime.now(timezone.utc).isoformat(),
             "type": "analysis",
         }
         session_id = self.session_manager.create_session(session_data)
@@ -516,7 +521,7 @@ def analyze_command(args):
     result = command.run(args)
 
     if result.success:
-        if args.format == "json":
+        if str(args.format).lower() == OutputFormat.JSON:
             print(json.dumps(result.data, indent=2))
         else:
             print(result.message)
@@ -540,7 +545,7 @@ if __name__ == "__main__":
     parser.add_argument("--focus", nargs="+")
     parser.add_argument("--session-id", type=str)
     parser.add_argument("--no-session", action="store_true")
-    parser.add_argument("--format", default="text")
+    parser.add_argument("--format", default=OutputFormat.TEXT)
     parser.add_argument("--output", type=Path)
     parser.add_argument("--verbose", action="store_true")
 

claude_mpm/cli/commands/analyze_code.py

@@ -20,7 +20,6 @@ from pathlib import Path
 from typing import Any, List, Optional
 
 try:
-    from rich import print as rprint
     from rich.console import Console
     from rich.progress import (
         BarColumn,
@@ -38,6 +37,7 @@ except ImportError:
     RICH_AVAILABLE = False
     Console = None
 
+from ...core.enums import OutputFormat
 from ...core.logging_config import get_logger
 from ...tools.code_tree_analyzer import CodeTreeAnalyzer
 from ...tools.code_tree_builder import CodeTreeBuilder
@@ -74,7 +74,11 @@ class AnalyzeCodeCommand(BaseCommand):
             return f"Path is not a directory: {path}"
 
         # Validate output format
-        if args.output and args.output not in ("json", "tree", "stats"):
+        if args.output and str(args.output).lower() not in (
+            OutputFormat.JSON,
+            "tree",
+            "stats",
+        ):
             return f"Invalid output format: {args.output}"
 
         return None
@@ -124,7 +128,7 @@ class AnalyzeCodeCommand(BaseCommand):
             )
 
             # Handle output format
-            if args.output == "json":
+            if str(args.output).lower() == OutputFormat.JSON:
                 # Output JSON to stdout
                 json.dump(analysis_result, sys.stdout, indent=2, default=str)
                 sys.stdout.write("\n")
@@ -137,7 +141,7 @@ class AnalyzeCodeCommand(BaseCommand):
             # Save to file if specified
             if args.save:
                 save_path = Path(args.save)
-                with open(save_path, "w") as f:
+                with save_path.open("w") as f:
                     json.dump(analysis_result, f, indent=2, default=str)
                 self.logger.info(f"Analysis saved to {save_path}")