From 64309c53b7478188663ae155d5481f88389cea36 Mon Sep 17 00:00:00 2001 From: overcuriousity Date: Wed, 24 Sep 2025 12:01:33 +0200 Subject: [PATCH] fixes to export manager --- utils/export_manager.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) 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."""