souleyez 3.0.0__py3-none-any.whl → 3.0.9__py3-none-any.whl

souleyez/ui/attack_surface.py

@@ -6,9 +6,11 @@ Unified attack surface dashboard with host-centric analysis and exploitation tra
 Host-centric view: Clean main table showing hosts, with drill-down for details.
 Consolidates Intelligence View, Exploit Suggestions, and Attack Correlation.
 """
-import click
-from typing import Dict, List, Set, Any
+
 import shutil
+from typing import Any, Dict, List, Set
+
+import click
 
 
 def get_terminal_width() -> int:
@@ -154,11 +156,12 @@ def view_attack_surface(engagement_id: int):
     Args:
         engagement_id: The engagement ID
     """
+    from rich.console import Console
+    from rich.table import Table
+
     from souleyez.intelligence.surface_analyzer import AttackSurfaceAnalyzer
     from souleyez.storage.engagements import EngagementManager
     from souleyez.ui.design_system import DesignSystem
-    from rich.console import Console
-    from rich.table import Table
 
     em = EngagementManager()
     analyzer = AttackSurfaceAnalyzer()
@@ -540,14 +543,16 @@ def view_attack_surface(engagement_id: int):
 
 def _show_host_siem_alerts(engagement_id: int, host_ip: str, siem_type: str):
     """Show recent SIEM alerts for a specific host."""
-    from souleyez.ui.design_system import DesignSystem
-    from souleyez.ui.interactive import render_standard_header
-    from souleyez.integrations.wazuh.config import WazuhConfig
-    from souleyez.integrations.siem.factory import SIEMFactory
     from datetime import datetime, timedelta
+
     from rich.console import Console
     from rich.table import Table
 
+    from souleyez.integrations.siem.factory import SIEMFactory
+    from souleyez.integrations.wazuh.config import WazuhConfig
+    from souleyez.ui.design_system import DesignSystem
+    from souleyez.ui.interactive import render_standard_header
+
     DesignSystem.clear_screen()
     click.echo()
 
@@ -850,6 +855,7 @@ def _display_hosts_table(
     Returns: (current_page, total_pages)
     """
     from rich.table import Table
+
     from souleyez.ui.design_system import DesignSystem
 
     # Show active filter
@@ -1195,9 +1201,10 @@ def _host_bulk_action_menu(
 
 def _view_gap_detail(engagement_id: int, gap: Dict, analysis: Dict):
     """View detailed information for a single exploitation gap."""
-    from souleyez.ui.design_system import DesignSystem
     from rich.console import Console
 
+    from souleyez.ui.design_system import DesignSystem
+
     console = Console()
 
     while True:
@@ -1405,10 +1412,11 @@ def _view_gap_detail(engagement_id: int, gap: Dict, analysis: Dict):
 
 def _view_host_detail(engagement_id: int, host_data: Dict, analysis: Dict):
     """View detailed information for a single host with all sections."""
-    from souleyez.ui.design_system import DesignSystem
     from rich.console import Console
     from rich.table import Table
 
+    from souleyez.ui.design_system import DesignSystem
+
     console = Console()
 
     services = host_data.get("services", [])
@@ -1965,12 +1973,13 @@ def _view_service_detail(
     engagement_id: int, host_data: Dict, service: Dict, analysis: Dict
 ):
     """View detailed information for a single service (Level 3 drill-down)."""
-    from souleyez.ui.design_system import DesignSystem
+    from rich.console import Console
+    from rich.table import Table
+
     from souleyez.intelligence.exploit_suggestions import ExploitSuggestionEngine
     from souleyez.storage.findings import FindingsManager
     from souleyez.storage.hosts import HostManager
-    from rich.console import Console
-    from rich.table import Table
+    from souleyez.ui.design_system import DesignSystem
 
     console = Console()
     host_ip = host_data["host_ip"]
@@ -2140,11 +2149,12 @@ def _run_service_exploit(
     engagement_id: int, host_ip: str, port: int, exploits: List[Dict]
 ):
     """Run an exploit against a service."""
+    import os
+    import tempfile
+
     from souleyez.core.msf_integration import MSFConsoleManager
     from souleyez.storage.exploit_attempts import record_attempt
     from souleyez.storage.hosts import HostManager
-    import tempfile
-    import os
 
     click.echo()
     click.echo("Select exploit to run:")
@@ -2397,9 +2407,9 @@ def _get_critical_findings_for_host(engagement_id: int, host_ip: str) -> List[Di
 def _get_wazuh_vulns_for_host(engagement_id: int, host_ip: str) -> List[Dict]:
     """Get Wazuh vulnerabilities for a specific host."""
     try:
-        from souleyez.storage.wazuh_vulns import WazuhVulnsManager
-        from souleyez.storage.hosts import HostManager
         from souleyez.integrations.wazuh import WazuhConfig
+        from souleyez.storage.hosts import HostManager
+        from souleyez.storage.wazuh_vulns import WazuhVulnsManager
 
         # Check if Wazuh is configured
         config = WazuhConfig.get_config(engagement_id)
@@ -2885,10 +2895,11 @@ def _exploit_host(engagement_id: int, host_ip: str, exploits: List[Dict]):
 
 def _view_host_findings(engagement_id: int, host_ip: str):
     """View all findings for a host with interactive selection."""
+    from rich.console import Console
+
     from souleyez.storage.findings import FindingsManager
     from souleyez.storage.hosts import HostManager
     from souleyez.ui.interactive_selector import interactive_select
-    from rich.console import Console
 
     hm = HostManager()
     fm = FindingsManager()
@@ -2974,8 +2985,8 @@ def _view_host_findings(engagement_id: int, host_ip: str):
 
 def _view_host_wazuh_vulns(engagement_id: int, host_ip: str, vulns: List[Dict]):
     """View all Wazuh vulnerabilities for a host with interactive selection."""
-    from souleyez.ui.interactive_selector import interactive_select
     from souleyez.ui.design_system import DesignSystem
+    from souleyez.ui.interactive_selector import interactive_select
 
     if not vulns:
         click.echo("  No Wazuh vulnerabilities for this host")
@@ -3116,9 +3127,10 @@ def _view_wazuh_vuln_detail(vuln: Dict):
 
 def _view_finding_detail(engagement_id: int, host_ip: str, finding: Dict):
     """View detailed information for a single finding (full-page view)."""
-    from souleyez.ui.design_system import DesignSystem
     from rich.console import Console
 
+    from souleyez.ui.design_system import DesignSystem
+
     console = Console()
 
     while True:
@@ -3401,9 +3413,10 @@ def _get_exploit_job_status(host_ip: str, msf_module: str, engagement_id: int) -
 
 def _view_all_exploits(engagement_id: int, host_ip: str, exploits: List[Dict]):
     """View all exploit suggestions for a host with interactive selection."""
-    from souleyez.ui.interactive_selector import interactive_select
     from rich.console import Console
 
+    from souleyez.ui.interactive_selector import interactive_select
+
     if not exploits:
         click.echo("  No exploit suggestions available")
         click.pause()
@@ -3494,9 +3507,10 @@ def _view_all_exploits(engagement_id: int, host_ip: str, exploits: List[Dict]):
 
 def _view_exploit_detail(engagement_id: int, host_ip: str, exploit: Dict):
     """View detailed information for a single exploit."""
-    from souleyez.ui.design_system import DesignSystem
     from rich.console import Console
 
+    from souleyez.ui.design_system import DesignSystem
+
     console = Console()
 
     while True:
@@ -3841,9 +3855,10 @@ def _exploit_bulk_action_menu(
 
 def _view_gaps_centric(engagement_id: int, analysis: Dict, engagement: Dict):
     """Switch to gap-centric view (the old view)."""
-    from souleyez.ui.design_system import DesignSystem
     from rich.console import Console
 
+    from souleyez.ui.design_system import DesignSystem
+
     # State
     selected_gap_ids = set()
     gap_filter_severity = None
@@ -4253,6 +4268,7 @@ def display_exploitation_gaps_table(
     Returns: (current_page, total_pages)
     """
     from rich.table import Table
+
     from souleyez.ui.design_system import DesignSystem
 
     # Count by severity
@@ -4265,7 +4281,9 @@ def display_exploitation_gaps_table(
 
     click.echo("═" * width)
     click.echo(
-        click.style(f"⚠️  EXPLOITATION GAPS ({len(gaps)} total)", bold=True, fg="yellow")
+        click.style(
+            f"⚠️  EXPLOITATION GAPS ({len(gaps)} total)", bold=True, fg="yellow"
+        )
     )
 
     # Severity breakdown line
@@ -4881,9 +4899,10 @@ def display_exploit_suggestions(
     engagement_id: int, top_hosts: List[Dict], width: int, show_all: bool = False
 ):
     """Display exploit suggestions for top hosts."""
-    from souleyez.intelligence.exploit_suggestions import ExploitSuggestionEngine
     from rich.console import Console
 
+    from souleyez.intelligence.exploit_suggestions import ExploitSuggestionEngine
+
     console = Console()
     # Disable SearchSploit in dashboard to prevent UI hangs - use manual Exploit Suggestions menu instead
     engine = ExploitSuggestionEngine(use_searchsploit=False)
@@ -5277,6 +5296,7 @@ def view_host_details(engagement_id: int, hosts: List[Dict]):
             # Show all services in a table
             from rich.console import Console
             from rich.table import Table
+
             from souleyez.ui.design_system import DesignSystem
 
             console = Console(width=width - 4)
@@ -5485,10 +5505,11 @@ def _view_quick_wins(engagement_id: int, analysis: Dict):
     Shows services with high-severity CVEs or known MSF modules that are
     untried and have high success probability.
     """
-    from souleyez.ui.design_system import DesignSystem
+    from rich.console import Console
+
     from souleyez.intelligence.exploit_suggestions import ExploitSuggestionEngine
     from souleyez.storage.hosts import HostManager
-    from rich.console import Console
+    from souleyez.ui.design_system import DesignSystem
 
     console = Console()
     engine = ExploitSuggestionEngine(use_searchsploit=False)  # Fast mode
@@ -5780,10 +5801,11 @@ def _view_quick_wins(engagement_id: int, analysis: Dict):
 
 def _show_msf_commands_for_quick_wins(selected_wins: list):
     """Show MSF commands for selected quick wins."""
-    from souleyez.ui.design_system import DesignSystem
     from rich.console import Console
     from rich.panel import Panel
 
+    from souleyez.ui.design_system import DesignSystem
+
     console = Console()
     DesignSystem.clear_screen()
     width = get_terminal_width()

souleyez/ui/chain_rules_view.py

@@ -4,12 +4,15 @@ souleyez.ui.chain_rules_view
 
 Chain rule management interface for auto-chaining configuration.
 """
+
 import math
-import click
 from typing import List
-from souleyez.core.tool_chaining import ToolChaining, ChainRule, CATEGORY_ICONS
-from souleyez.ui.menu_components import StandardMenu
+
+import click
+
+from souleyez.core.tool_chaining import CATEGORY_ICONS, ChainRule, ToolChaining
 from souleyez.ui.design_system import DesignSystem
+from souleyez.ui.menu_components import StandardMenu
 
 
 def manage_chain_rules():

souleyez/ui/correlation_view.py

@@ -2,15 +2,16 @@
 """
 Correlation view UI for displaying attack correlation and gaps.
 """
+
 import click
 from rich.console import Console
-from rich.table import Table
 from rich.panel import Panel
+from rich.table import Table
 
 from souleyez.intelligence.correlation_analyzer import CorrelationAnalyzer
 from souleyez.intelligence.gap_detector import GapDetector
-from souleyez.ui.design_system import DesignSystem
 from souleyez.storage.engagements import EngagementManager
+from souleyez.ui.design_system import DesignSystem
 
 
 def show_correlation_view(engagement_id: int):