diff --git a/utils/export_manager.py b/utils/export_manager.py index 322d3ff..9d8d795 100644 --- a/utils/export_manager.py +++ b/utils/export_manager.py @@ -290,10 +290,13 @@ class ExportManager: f"• Data Providers Used: {len(audit_trail.get('session_metadata', {}).get('providers_used', []))}", ]) + correlation_provider = next((p for p in scanner.providers if p.get_name() == 'correlation'), None) + correlation_count = len(correlation_provider.correlation_index) if correlation_provider else 0 + report.extend([ "", "Correlation Analysis:", - f"• Entity Correlations Identified: {len(scanner.graph.correlation_index)}", + f"• Entity Correlations Identified: {correlation_count}", f"• Cross-Reference Validation: {self._count_cross_validated_relationships(edges)} relationships verified by multiple sources", "" ]) @@ -476,20 +479,19 @@ class ExportManager: 'dns_ns_record': 0.7 } - scored_edges = [] + edges = [] for edge in edges: type_weight = relationship_importance.get(edge.get('label', ''), 0.5) - scored_edges.append({ + edges.append({ 'source': edge['from'], 'target': edge['to'], 'type': edge.get('label', ''), 'provider': edge.get('source_provider', ''), - 'score': combined_score }) # Return top relationships by score - return sorted(scored_edges, key=lambda x: x['score'], reverse=True) + return sorted(edges, key=lambda x: x['score'], reverse=True) def _analyze_certificate_infrastructure(self, nodes: List[Dict]) -> Dict[str, Any]: """Analyze certificate infrastructure across all domains."""