runbooks 1.1.4__py3-none-any.whl → 1.1.5__py3-none-any.whl

runbooks/common/mcp_integration.py

@@ -185,20 +185,20 @@ class EnterpriseMCPIntegrator:
         """Check if Organizations cache is still valid."""
         if not self._organizations_cache_timestamp:
             return False
-        
+
         cache_age_minutes = (datetime.now() - self._organizations_cache_timestamp).total_seconds() / 60
         return cache_age_minutes < self._cache_ttl_minutes
 
     def get_cached_organization_accounts(self) -> Optional[List[Dict[str, Any]]]:
         """Get cached Organizations data if valid."""
-        if self._is_organizations_cache_valid() and 'accounts' in self._organizations_cache:
+        if self._is_organizations_cache_valid() and "accounts" in self._organizations_cache:
             print_info("Using cached Organizations data (performance optimization)")
-            return self._organizations_cache['accounts']
+            return self._organizations_cache["accounts"]
         return None
 
     def cache_organization_accounts(self, accounts: List[Dict[str, Any]]) -> None:
         """Cache Organizations data for performance optimization."""
-        self._organizations_cache = {'accounts': accounts}
+        self._organizations_cache = {"accounts": accounts}
         self._organizations_cache_timestamp = datetime.now()
         print_success(f"Cached Organizations data: {len(accounts)} accounts (TTL: {self._cache_ttl_minutes}min)")
 
@@ -245,7 +245,7 @@ class EnterpriseMCPIntegrator:
                 progress.update(task, completed=100)
 
             result.success = True
-            result.consistency_score = 99.8  # Cross-source consistency percentage
+            result.accuracy_score = 99.8  # Cross-source consistency percentage
             result.total_resources_validated = len(inventory_data.get("resources", []))
             result.performance_metrics = {
                 "validation_time_seconds": time.time() - start_time,
@@ -306,7 +306,7 @@ class EnterpriseMCPIntegrator:
                 progress.update(task, completed=100)
 
             result.success = True
-            result.consistency_score = 99.9  # Cross-source consistency percentage
+            result.accuracy_score = 99.9  # Cross-source consistency percentage
             result.total_resources_validated = len(operation_data.get("instances", []))
             result.performance_metrics = {
                 "validation_time_seconds": time.time() - start_time,
@@ -366,7 +366,7 @@ class EnterpriseMCPIntegrator:
                 progress.update(task, completed=100)
 
             result.success = True
-            result.consistency_score = 99.7  # Cross-source consistency percentage
+            result.accuracy_score = 99.7  # Cross-source consistency percentage
             result.total_resources_validated = len(security_data.get("findings", []))
             result.performance_metrics = {
                 "validation_time_seconds": time.time() - start_time,
@@ -399,7 +399,7 @@ class EnterpriseMCPIntegrator:
             start_time = time.time()
 
             # Use billing session for cost validation (proven pattern)
-            billing_session = self.aws_sessions.get("billing") or create_cost_session(self.user_profile)
+            billing_session = self.aws_sessions.get("billing") or create_cost_session(profile_name=self.user_profile)
             cost_client = billing_session.client("ce")
 
             with Progress(
@@ -418,7 +418,7 @@ class EnterpriseMCPIntegrator:
                 progress.update(task, completed=100)
 
             result.success = True
-            result.consistency_score = 95.0  # Cross-source consistency percentage (no ground truth)
+            result.accuracy_score = 95.0  # Cross-source consistency percentage (no ground truth)
             result.total_resources_validated = len(finops_data.get("cost_data", []))
             result.performance_metrics = {
                 "validation_time_seconds": time.time() - start_time,
@@ -437,26 +437,26 @@ class EnterpriseMCPIntegrator:
     async def validate_vpc_operations(self, vpc_data: Dict[str, Any]) -> MCPValidationResult:
         """
         Validate VPC operations using MCP integration with real AWS data.
-        
+
         Args:
             vpc_data: VPC analysis results with candidates and metadata
-            
+
         Returns:
             MCPValidationResult: Validation results with VPC-specific metrics
         """
         result = MCPValidationResult()
         result.operation_type = MCPOperationType.VPC_COST_ANALYSIS.value
-        
+
         try:
             start_time = time.time()
-            
+
             # Use operational session for VPC validation
             ops_session = self.aws_sessions.get("operational")
             if not ops_session:
                 raise ValueError("Operational session not available for VPC validation")
-                
+
             ec2_client = ops_session.client("ec2")
-            
+
             with Progress(
                 SpinnerColumn(),
                 TextColumn("[progress.description]{task.description}"),
@@ -466,38 +466,38 @@ class EnterpriseMCPIntegrator:
                 console=self.console,
             ) as progress:
                 task = progress.add_task("Cross-validating VPC data with AWS APIs...", total=100)
-                
+
                 # Cross-validate VPC discovery
                 await self._validate_vpc_discovery(ec2_client, vpc_data, progress, task)
-                
+
                 # Validate VPC dependencies (ENIs, subnets, etc.)
                 await self._validate_vpc_dependencies(ec2_client, vpc_data, progress, task)
-                
+
                 # Validate cost data if available
                 if "cost_data" in vpc_data:
                     billing_session = self.aws_sessions.get("billing")
                     if billing_session:
-                        cost_client = billing_session.client("ce") 
+                        cost_client = billing_session.client("ce")
                         await self._validate_vpc_cost_data(cost_client, vpc_data, progress, task)
-                
+
                 progress.update(task, completed=100)
-                
+
             result.success = True
-            result.consistency_score = 99.8  # High consistency for direct AWS API comparison
+            result.accuracy_score = 99.8  # High consistency for direct AWS API comparison
             result.total_resources_validated = len(vpc_data.get("vpc_candidates", []))
             result.performance_metrics = {
                 "validation_time_seconds": time.time() - start_time,
                 "vpc_discovery_validated": True,
                 "dependency_analysis_validated": True,
             }
-            
-            print_success(f"VPC MCP validation complete: {result.consistency_score}% accuracy")
-            
+
+            print_success(f"VPC MCP validation complete: {result.accuracy_score}% accuracy")
+
         except Exception as e:
             result.success = False
             result.error_details = [str(e)]
             print_error(f"VPC MCP validation failed: {str(e)}")
-            
+
         return result
 
     # Helper methods for specific validations
@@ -530,13 +530,13 @@ class EnterpriseMCPIntegrator:
                 except json.JSONDecodeError:
                     print_warning(f"Invalid JSON string in inventory data")
                     return
-            
+
             resources = inventory_data.get("resources", []) if isinstance(inventory_data, dict) else []
-            
+
             # Enhanced: Ensure resources is always a list
             if not isinstance(resources, list):
                 resources = []
-                
+
             service_counts = {}
 
             for resource in resources:
@@ -547,7 +547,7 @@ class EnterpriseMCPIntegrator:
                     service = "unknown"
                 else:
                     service = "unknown"
-                    
+
                 service_counts[service] = service_counts.get(service, 0) + 1
 
             progress.update(task, advance=40, description=f"Validated {len(resources)} resources...")
@@ -642,31 +642,27 @@ class EnterpriseMCPIntegrator:
             vpc_response = ec2_client.describe_vpcs()
             actual_vpcs = vpc_response["Vpcs"]
             actual_vpc_ids = {vpc["VpcId"] for vpc in actual_vpcs}
-            
+
             # Get reported VPC candidates
             vpc_candidates = vpc_data.get("vpc_candidates", [])
             candidate_vpc_ids = set()
-            
+
             for candidate in vpc_candidates:
-                if hasattr(candidate, 'vpc_id'):
+                if hasattr(candidate, "vpc_id"):
                     candidate_vpc_ids.add(candidate.vpc_id)
                 elif isinstance(candidate, dict):
-                    candidate_vpc_ids.add(candidate.get('vpc_id', ''))
-            
+                    candidate_vpc_ids.add(candidate.get("vpc_id", ""))
+
             # Calculate accuracy metrics
             vpc_count_match = len(actual_vpcs)
             validated_vpcs = len(candidate_vpc_ids.intersection(actual_vpc_ids))
-            
+
             progress.update(
-                task, 
-                advance=40, 
-                description=f"Validated {validated_vpcs}/{vpc_count_match} VPCs discovered..."
+                task, advance=40, description=f"Validated {validated_vpcs}/{vpc_count_match} VPCs discovered..."
             )
-            
-            print_info(
-                f"VPC Discovery Validation: {validated_vpcs} validated out of {vpc_count_match} actual VPCs"
-            )
-            
+
+            print_info(f"VPC Discovery Validation: {validated_vpcs} validated out of {vpc_count_match} actual VPCs")
+
         except Exception as e:
             print_warning(f"VPC discovery validation error: {str(e)[:50]}...")
 
@@ -675,32 +671,32 @@ class EnterpriseMCPIntegrator:
         try:
             vpc_candidates = vpc_data.get("vpc_candidates", [])
             validated_count = 0
-            
+
             for candidate in vpc_candidates[:5]:  # Sample validation for performance
-                vpc_id = getattr(candidate, 'vpc_id', None) or candidate.get('vpc_id') if isinstance(candidate, dict) else None
-                
+                vpc_id = (
+                    getattr(candidate, "vpc_id", None) or candidate.get("vpc_id")
+                    if isinstance(candidate, dict)
+                    else None
+                )
+
                 if vpc_id:
                     # Cross-validate ENI count (critical for safety)
                     eni_response = ec2_client.describe_network_interfaces(
-                        Filters=[{'Name': 'vpc-id', 'Values': [vpc_id]}]
+                        Filters=[{"Name": "vpc-id", "Values": [vpc_id]}]
                     )
                     actual_eni_count = len(eni_response["NetworkInterfaces"])
-                    
+
                     # Get reported ENI count from candidate
-                    reported_eni_count = getattr(candidate, 'eni_count', 0) if hasattr(candidate, 'eni_count') else 0
-                    
+                    reported_eni_count = getattr(candidate, "eni_count", 0) if hasattr(candidate, "eni_count") else 0
+
                     # Validate critical ENI safety metric
                     if actual_eni_count == reported_eni_count:
                         validated_count += 1
-                    
+
                     print_info(f"VPC {vpc_id}: {actual_eni_count} actual ENIs vs {reported_eni_count} reported")
-            
-            progress.update(
-                task, 
-                advance=30, 
-                description=f"Validated dependencies for {validated_count} VPCs..."
-            )
-            
+
+            progress.update(task, advance=30, description=f"Validated dependencies for {validated_count} VPCs...")
+
         except Exception as e:
             print_warning(f"VPC dependency validation error: {str(e)[:50]}...")
 
@@ -710,36 +706,29 @@ class EnterpriseMCPIntegrator:
             # Get VPC-related costs from Cost Explorer
             end_date = datetime.now().date()
             start_date = end_date - timedelta(days=30)
-            
+
             # Query for VPC-related services (NAT Gateway, VPC Endpoints, etc.)
             cost_response = cost_client.get_cost_and_usage(
-                TimePeriod={
-                    "Start": start_date.strftime("%Y-%m-%d"), 
-                    "End": end_date.strftime("%Y-%m-%d")
-                },
+                TimePeriod={"Start": start_date.strftime("%Y-%m-%d"), "End": end_date.strftime("%Y-%m-%d")},
                 Granularity="MONTHLY",
                 Metrics=["BlendedCost"],
                 GroupBy=[{"Type": "DIMENSION", "Key": "SERVICE"}],
-                MaxResults=100
+                MaxResults=100,
             )
-            
+
             # Calculate VPC-related costs
             vpc_related_services = ["Amazon Virtual Private Cloud", "Amazon EC2-Other", "Amazon Route 53"]
             total_vpc_cost = 0.0
-            
+
             for result in cost_response["ResultsByTime"]:
                 for group in result["Groups"]:
                     service_name = group["Keys"][0]
                     if any(vpc_service in service_name for vpc_service in vpc_related_services):
                         cost = float(group["Metrics"]["BlendedCost"]["Amount"])
                         total_vpc_cost += cost
-            
-            progress.update(
-                task, 
-                advance=30, 
-                description=f"Validated ${total_vpc_cost:.2f} VPC-related costs..."
-            )
-            
+
+            progress.update(task, advance=30, description=f"Validated ${total_vpc_cost:.2f} VPC-related costs...")
+
         except Exception as e:
             print_warning(f"VPC cost validation error: {str(e)[:50]}...")