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

souleyez/ui/dashboard.py

@@ -2,29 +2,30 @@
 """
 souleyez.ui.dashboard - Live dashboard with real-time updates
 """
-import click
-from souleyez.config import read_config, write_config
-import time
-import os
-import wcwidth
+
 import getpass
+import os
+import time
 from datetime import datetime
-from typing import Optional
 from enum import Enum
+from io import StringIO
+from typing import Optional
 
-from souleyez.engine.background import list_jobs, get_job
-from souleyez.storage.engagements import EngagementManager
-from souleyez.storage.hosts import HostManager
-from souleyez.storage.findings import FindingsManager
-from souleyez.storage.credentials import CredentialsManager
-from souleyez.storage.evidence import EvidenceManager
-from souleyez.ui.log_formatter import format_log_stream
-from souleyez.ui.design_system import DesignSystem
-from souleyez.ui.tutorial_state import get_tutorial_state, TutorialStep
+import click
+import wcwidth
 from rich.console import Console
 from rich.table import Table
-from io import StringIO
 
+from souleyez.config import read_config, write_config
+from souleyez.engine.background import get_job, list_jobs
+from souleyez.storage.credentials import CredentialsManager
+from souleyez.storage.engagements import EngagementManager
+from souleyez.storage.evidence import EvidenceManager
+from souleyez.storage.findings import FindingsManager
+from souleyez.storage.hosts import HostManager
+from souleyez.ui.design_system import DesignSystem
+from souleyez.ui.log_formatter import format_log_stream
+from souleyez.ui.tutorial_state import TutorialStep, get_tutorial_state
 
 # Header cache to avoid expensive recalculations on every refresh
 # Format: {key: (timestamp, value)}
@@ -502,11 +503,11 @@ def render_engagement_summary(engagement_id: int, width: int) -> list:
     Render compact engagement summary - single section with key metrics.
     6 lines total for maximum clarity.
     """
-    from souleyez.storage.hosts import HostManager
-    from souleyez.storage.findings import FindingsManager
+    from souleyez.engine.background import list_jobs
     from souleyez.storage.credentials import CredentialsManager
     from souleyez.storage.deliverables import DeliverableManager
-    from souleyez.engine.background import list_jobs
+    from souleyez.storage.findings import FindingsManager
+    from souleyez.storage.hosts import HostManager
 
     lines = []
     lines.extend(DesignSystem.section_header("📊", "ENGAGEMENT SUMMARY"))
@@ -1286,8 +1287,8 @@ def render_intelligence_summary(
         )
     else:
         try:
-            from souleyez.integrations.wazuh.config import WazuhConfig
             from souleyez.integrations.siem import SIEMFactory
+            from souleyez.integrations.wazuh.config import WazuhConfig
 
             config = WazuhConfig.get_config(engagement_id)
 
@@ -1396,8 +1397,8 @@ def render_intelligence_summary(
         )
     else:
         try:
-            from souleyez.ai.ollama_provider import OllamaProvider
             from souleyez.ai.claude_provider import ClaudeProvider
+            from souleyez.ai.ollama_provider import OllamaProvider
 
             config = read_config()
             current_provider = config.get("ai", {}).get("provider", "ollama")
@@ -1616,12 +1617,12 @@ def render_new_tool_metrics(engagement_id: int, width: int):
 
     # Get data (cached to avoid lag)
     def _get_tool_metrics_data():
-        from souleyez.storage.hosts import HostManager
+        from souleyez.engine.background import list_jobs
         from souleyez.storage.findings import FindingsManager
+        from souleyez.storage.hosts import HostManager
         from souleyez.storage.osint import OsintManager
         from souleyez.storage.smb_shares import SMBSharesManager
         from souleyez.storage.web_paths import WebPathsManager
-        from souleyez.engine.background import list_jobs
 
         hm = HostManager()
         fm = FindingsManager()
@@ -2053,8 +2054,8 @@ def _quick_toggle_ai_provider():
 
 def _execute_ai_recommendation(engagement_id: int):
     """Execute AI recommendation from dashboard."""
-    from souleyez.ai.recommender import AttackRecommender
     from souleyez.ai.executor import InteractiveExecutor
+    from souleyez.ai.recommender import AttackRecommender
     from souleyez.ai.safety import ApprovalMode
     from souleyez.ui.progress_indicators import with_ai_quotes
 
@@ -2067,8 +2068,8 @@ def _execute_ai_recommendation(engagement_id: int):
 
     # Check available providers
     try:
-        from souleyez.ai.ollama_provider import OllamaProvider
         from souleyez.ai.claude_provider import ClaudeProvider
+        from souleyez.ai.ollama_provider import OllamaProvider
 
         ollama = OllamaProvider()
         claude = ClaudeProvider()
@@ -2452,10 +2453,11 @@ def render_recent_findings(engagement_id: int, width: int):
 
 def render_endpoints_and_redirects(engagement_id: int, width: int):
     """Render recent endpoints and HTTP redirects with suspicious path highlighting."""
-    from souleyez.storage.web_paths import WebPathsManager
+    from urllib.parse import urlparse
+
     from souleyez.storage.hosts import HostManager
     from souleyez.storage.osint import OsintManager
-    from urllib.parse import urlparse
+    from souleyez.storage.web_paths import WebPathsManager
 
     wpm = WebPathsManager()
     hm = HostManager()
@@ -3731,8 +3733,8 @@ def run_dashboard(follow_job_id: Optional[int] = None, refresh_interval: int = 1
 
 def _toggle_auto_chaining():
     """Toggle auto-chaining on/off with improved feedback."""
-    from souleyez.core.tool_chaining import ToolChaining
     from souleyez.auth import is_pro
+    from souleyez.core.tool_chaining import ToolChaining
     from souleyez.ui.interactive import _show_upgrade_prompt
 
     if not is_pro():
@@ -4162,7 +4164,7 @@ def _view_security_guide():
 
 def _show_help_menu():
     """Show help menu with available guides."""
-    from souleyez.ui.help_system import show_help, HelpContext
+    from souleyez.ui.help_system import HelpContext, show_help
 
     show_help(HelpContext.DASHBOARD)
 
@@ -4217,8 +4219,8 @@ def _show_toggle_menu(state: "DashboardState"):
 
 def _wait_for_input(timeout: int) -> Optional[str]:
     """Wait for keyboard input with timeout. Returns input or None."""
-    import sys
     import select
+    import sys
 
     try:
         # Check if input is available (Unix-like systems)
@@ -4277,7 +4279,7 @@ def _show_dashboard_menu(engagement_id: int) -> str:
         click.echo("  " + "─" * 76)
 
         # Check user tier for Pro features
-        from souleyez.auth import get_current_user, Tier
+        from souleyez.auth import Tier, get_current_user
 
         current_user = get_current_user()
         is_pro_user = current_user and current_user.tier == Tier.PRO
@@ -4293,8 +4295,6 @@ def _show_dashboard_menu(engagement_id: int) -> str:
         click.echo(
             "    "
             + click.style("[i]", fg="bright_magenta", bold=True)
-            + " or "
-            + click.style("[1]", fg="bright_magenta")
             + "  🎯 Intelligence Hub     "
             + pro_badge
             + " - Attack surface & correlation"
@@ -4303,9 +4303,7 @@ def _show_dashboard_menu(engagement_id: int) -> str:
         # Show all Pro features with appropriate badge
         click.echo(
             "    "
-            + click.style("[x]", fg="bright_magenta", bold=True)
-            + " or "
-            + click.style("[2]", fg="bright_magenta")
+            + click.style("[e]", fg="bright_magenta", bold=True)
             + "  🤖 AI Execute           "
             + pro_badge
             + " - AI-driven autonomous execution"
@@ -4313,8 +4311,6 @@ def _show_dashboard_menu(engagement_id: int) -> str:
         click.echo(
             "    "
             + click.style("[a]", fg="bright_magenta", bold=True)
-            + " or "
-            + click.style("[3]", fg="bright_magenta")
             + "  🔗 Automation           "
             + pro_badge
             + " - Chain rules & settings"
@@ -4322,132 +4318,100 @@ def _show_dashboard_menu(engagement_id: int) -> str:
         click.echo(
             "    "
             + click.style("[m]", fg="bright_magenta", bold=True)
-            + " or "
-            + click.style("[4]", fg="bright_magenta")
             + "  🔧 Metasploit           "
             + pro_badge
             + " - Generate exploits, launch msfconsole"
         )
         click.echo()
 
-        # Documentation section
-        click.echo(click.style("  📋 DOCUMENTATION", bold=True, fg="yellow"))
-        click.echo("  " + "─" * 76)
-        click.echo(
-            "    "
-            + click.style("[e]", fg="yellow", bold=True)
-            + " or "
-            + click.style("[5]", fg="yellow")
-            + "  📦 Evidence & Artifacts  - Screenshots, files, exports"
-        )
-        click.echo(
-            "    "
-            + click.style("[d]", fg="yellow", bold=True)
-            + " or "
-            + click.style("[6]", fg="yellow")
-            + "  ✅ Deliverables          - Progress tracking, checklists"
-        )
-
-        # Reports - Pro feature (show to all with appropriate badge)
-        pro_badge_yellow = (
-            click.style("💎", fg="yellow")
-            if is_pro_user
-            else click.style("🔒 PRO", fg="yellow")
-        )
-        click.echo(
-            "    "
-            + click.style("[r]", fg="yellow", bold=True)
-            + " or "
-            + click.style("[7]", fg="yellow")
-            + "  📄 Reports               "
-            + pro_badge_yellow
-            + " - Generate & export reports"
-        )
-        click.echo()
-
         # Data Management section
-        click.echo(
-            click.style("  📊 DATA MANAGEMENT (Deep Dive)", bold=True, fg="green")
-        )
+        click.echo(click.style("  📊 DATA MANAGEMENT", bold=True, fg="cyan"))
         click.echo("  " + "─" * 76)
         click.echo(
             "    "
             + click.style("[h]", fg="cyan", bold=True)
-            + " or "
-            + click.style("[8]", fg="cyan")
-            + "   🎯 Hosts             - Discovered hosts, tags, filtering"
+            + "  🎯 Hosts             - Discovered hosts, tags, filtering"
         )
         click.echo(
             "    "
             + click.style("[s]", fg="cyan", bold=True)
-            + " or "
-            + click.style("[9]", fg="cyan")
-            + "   🔌 Services          - Open ports, service enumeration"
+            + "  🔌 Services          - Open ports, service enumeration"
         )
         click.echo(
             "    "
             + click.style("[f]", fg="cyan", bold=True)
-            + " or "
-            + click.style("[10]", fg="cyan")
             + "  🔍 Findings          - All vulnerabilities (detailed view)"
         )
         click.echo(
             "    "
             + click.style("[c]", fg="cyan", bold=True)
-            + " or "
-            + click.style("[11]", fg="cyan")
             + "  🔑 Credentials       - Discovered users, passwords, hashes"
         )
         click.echo()
         click.echo(
             "    "
-            + click.style("[b]", fg="cyan", bold=True)
-            + " or "
-            + click.style("[12]", fg="cyan")
+            + click.style("[w]", fg="cyan", bold=True)
             + "  🌐 Web Paths         - Directory enumeration results"
         )
         click.echo(
             "    "
-            + click.style("[n]", fg="cyan", bold=True)
-            + " or "
-            + click.style("[13]", fg="cyan")
+            + click.style("[b]", fg="cyan", bold=True)
             + "  📁 SMB Shares        - SMB enumeration, accessible shares"
         )
         click.echo(
             "    "
             + click.style("[l]", fg="cyan", bold=True)
-            + " or "
-            + click.style("[14]", fg="cyan")
-            + "  💉 SQLMap Intelligence - Deduplicated injections, high-value tables"
+            + "  💉 SQLMap            - SQL injection discoveries"
         )
         click.echo(
             "    "
             + click.style("[p]", fg="cyan", bold=True)
-            + " or "
-            + click.style("[15]", fg="cyan")
-            + "  🔌 WordPress Data     - WPScan vulnerabilities"
+            + "  🔌 WordPress         - WPScan vulnerabilities"
+        )
+        click.echo(
+            "    "
+            + click.style("[x]", fg="cyan", bold=True)
+            + "  💣 Exploits          - SearchSploit exploit database"
         )
         click.echo(
             "    "
             + click.style("[o]", fg="cyan", bold=True)
-            + " or "
-            + click.style("[16]", fg="cyan")
-            + "  💣 Exploits           - SearchSploit exploit database"
+            + "  🔍 OSINT             - DNS, WHOIS, emails, infrastructure"
+        )
+        click.echo()
+
+        # Documentation section
+        click.echo(click.style("  📋 DOCUMENTATION", bold=True, fg="yellow"))
+        click.echo("  " + "─" * 76)
+        click.echo(
+            "    "
+            + click.style("[v]", fg="yellow", bold=True)
+            + "  📦 Evidence Vault        - Screenshots, files, exports"
         )
         click.echo(
             "    "
-            + click.style("[t]", fg="cyan", bold=True)
-            + " or "
-            + click.style("[17]", fg="cyan")
-            + "  🔍 OSINT/Discovery    - DNS, WHOIS, emails, infrastructure"
+            + click.style("[d]", fg="yellow", bold=True)
+            + "  ✅ Deliverables          - Progress tracking, checklists"
+        )
+
+        # Reports - Pro feature (show to all with appropriate badge)
+        pro_badge_yellow = (
+            click.style("💎", fg="yellow")
+            if is_pro_user
+            else click.style("🔒 PRO", fg="yellow")
+        )
+        click.echo(
+            "    "
+            + click.style("[g]", fg="yellow", bold=True)
+            + "  📄 Generate Reports      "
+            + pro_badge_yellow
+            + " - Generate & export reports"
         )
         click.echo()
 
         click.echo(
             "    "
             + click.style("[q]", fg="red", bold=True)
-            + " or "
-            + click.style("[18]", fg="red")
             + "  ← Return to Command Center"
         )
         click.echo()
@@ -4459,49 +4423,30 @@ def _show_dashboard_menu(engagement_id: int) -> str:
         try:
             choice = input().strip().lower()
 
-            # Map both letters and numbers
+            # Map letters to actions
             choice_map = {
-                # Intelligence & Actions (4 items: 1-4)
+                # Intelligence & Actions
                 "i": "intelligence_hub",
-                "1": "intelligence_hub",
-                "x": "ai_execute",
-                "2": "ai_execute",
+                "e": "ai_execute",
                 "a": "automation",
-                "3": "automation",
                 "m": "msf",
-                "4": "msf",
-                # Documentation (3 items: 5-7)
-                "e": "evidence",
-                "5": "evidence",
+                # Documentation
+                "v": "evidence",
                 "d": "deliverables",
-                "6": "deliverables",
-                "r": "reports",
-                "7": "reports",
-                # Data Management (10 items: 8-17)
+                "g": "reports",
+                # Data Management
                 "h": "hosts",
-                "8": "hosts",
                 "s": "services",
-                "9": "services",
                 "f": "findings",
-                "10": "findings",
                 "c": "credentials",
-                "11": "credentials",
-                "b": "web_paths",
-                "12": "web_paths",
-                "n": "smb_shares",
-                "13": "smb_shares",
+                "w": "web_paths",
+                "b": "smb_shares",
                 "l": "sqlmap",
-                "14": "sqlmap",
                 "p": "wordpress",
-                "15": "wordpress",
-                "o": "exploits_db",
-                "16": "exploits_db",
-                "t": "osint",
-                "17": "osint",
+                "x": "exploits_db",
+                "o": "osint",
                 # Exit
                 "q": "quit",
-                "18": "quit",
-                "0": "quit",
             }
 
             action = choice_map.get(choice, None)
@@ -5323,6 +5268,7 @@ def view_untested_usernames(engagement_id: int):
 def export_usernames_to_file(engagement_id: int, untested_creds: list):
     """Export untested usernames to a file."""
     import os
+
     from souleyez.storage.engagements import EngagementManager
 
     em = EngagementManager()

souleyez/ui/deliverables_view.py

@@ -2,8 +2,8 @@
 
 import click
 from rich.console import Console
+from rich.progress import BarColumn, Progress, TextColumn
 from rich.table import Table
-from rich.progress import Progress, BarColumn, TextColumn
 
 try:
     from rich.progress import TaskProgressColumn

souleyez/ui/design_system.py

@@ -2,11 +2,13 @@
 """
 souleyez.ui.design_system - Centralized design system for consistent UI
 """
+
+import os
 from typing import List
-from rich.table import Table
-from rich.console import Console
+
 from rich import box
-import os
+from rich.console import Console
+from rich.table import Table
 
 # Shared console instance
 _console = None

souleyez/ui/errors.py

@@ -5,8 +5,10 @@ souleyez.ui.errors - User-friendly error messages with fix suggestions
 This module provides consistent, helpful error messages that guide users
 toward solutions rather than just reporting problems.
 """
+
+from typing import List, Optional
+
 import click
-from typing import Optional, List
 
 
 class ErrorHelper:

souleyez/ui/evidence_linking_view.py

@@ -1,6 +1,7 @@
 """Evidence linking interface for deliverables."""
 
 import click
+
 from souleyez.storage.deliverable_evidence import EvidenceManager
 from souleyez.storage.deliverables import DeliverableManager
 from souleyez.ui.design_system import DesignSystem
@@ -183,8 +184,8 @@ def show_evidence_linking_view(deliverable_id: int):
 
 def _add_evidence_manual(deliverable_id: int, engagement_id: int):
     """Manually select and link evidence."""
-    from souleyez.storage.findings import FindingsManager
     from souleyez.storage.credentials import CredentialsManager
+    from souleyez.storage.findings import FindingsManager
 
     em = EvidenceManager()
     fm = FindingsManager()

souleyez/ui/evidence_vault.py

@@ -4,11 +4,14 @@ Evidence & Artifacts - Unified evidence collection view.
 Displays all artifacts organized by pentesting methodology phases.
 Consolidates Evidence Vault and Screenshots into a single view.
 """
-import click
-from typing import Dict, List, Optional, Set
+
 import shutil
+from typing import Dict, List, Optional, Set
+
+import click
 from rich.console import Console
 from rich.table import Table
+
 from souleyez.ui.design_system import DesignSystem
 from souleyez.ui.errors import engagement_not_found
 
@@ -147,8 +150,8 @@ def view_evidence_vault(engagement_id: int):
 
     Consolidates Evidence Vault and Screenshots into a unified view.
     """
-    from souleyez.storage.evidence import EvidenceManager
     from souleyez.storage.engagements import EngagementManager
+    from souleyez.storage.evidence import EvidenceManager
     from souleyez.storage.screenshots import ScreenshotManager
 
     em = EngagementManager()
@@ -508,6 +511,7 @@ def _export_selected_evidence(selected_items: List[Dict], engagement_id: int):
     import os
     import zipfile
     from datetime import datetime
+
     from souleyez.storage.engagement import EngagementManager
 
     click.echo(
@@ -953,11 +957,12 @@ def view_screenshots(engagement_id: int, screenshots: List[Dict], sm):
 
     Embedded screenshots view within Evidence & Artifacts.
     """
-    from rich.console import Console
-    from rich.table import Table
-    from pathlib import Path
     import subprocess
     import sys
+    from pathlib import Path
+
+    from rich.console import Console
+    from rich.table import Table
 
     while True:
         DesignSystem.clear_screen()

souleyez/ui/exploit_suggestions_view.py

@@ -11,17 +11,19 @@ Features:
 - Detailed info views with full context
 """
 
-import click
 import os
 import tempfile
+from typing import Dict, List, Optional, Tuple
+
+import click
 from rich.console import Console
-from rich.table import Table
 from rich.panel import Panel
-from typing import Dict, List, Optional, Tuple
-from souleyez.ui.design_system import DesignSystem
-from souleyez.storage.hosts import HostManager
+from rich.table import Table
+
+from souleyez.core.msf_integration import MSFConsoleManager, MSFResourceGenerator
 from souleyez.storage import exploit_attempts
-from souleyez.core.msf_integration import MSFResourceGenerator, MSFConsoleManager
+from souleyez.storage.hosts import HostManager
+from souleyez.ui.design_system import DesignSystem
 
 console = Console()
 
@@ -818,7 +820,9 @@ def _execute_msf_module(
         script += f"set PAYLOAD {payload}\n"
         script += "set LHOST 0.0.0.0  # CHANGE THIS to your IP\n"
         script += "set LPORT 4444\n"
-        console.print("  [yellow]⚠️  Remember to set LHOST to your IP address![/yellow]")
+        console.print(
+            "  [yellow]⚠️  Remember to set LHOST to your IP address![/yellow]"
+        )
 
     # Add check before exploit
     script += "\ncheck\n"

souleyez/ui/export_view.py

@@ -1,7 +1,9 @@
 """Export deliverables UI."""
 
-import click
 import os
+
+import click
+
 from souleyez.storage.deliverable_exporter import DeliverableExporter
 from souleyez.storage.engagements import EngagementManager
 from souleyez.ui.design_system import DesignSystem

souleyez/ui/gap_analysis_view.py

@@ -5,12 +5,15 @@ souleyez.ui.gap_analysis_view - Gap Analysis View
 Displays comparison between Wazuh (passive) and scan (active)
 vulnerability detection to identify gaps in coverage.
 """
+
+from typing import Dict, List
+
 import click
 from rich.console import Console
 from rich.table import Table
-from typing import List, Dict
-from souleyez.intelligence.gap_analyzer import GapAnalyzer, GapAnalysisResult
+
 from souleyez.integrations.wazuh import WazuhConfig, WazuhVulnSync
+from souleyez.intelligence.gap_analyzer import GapAnalysisResult, GapAnalyzer
 from souleyez.ui.design_system import DesignSystem
 
 console = Console()
@@ -200,8 +203,8 @@ def _render_summary_dashboard(
 
 def _render_severity_breakdown(breakdown: Dict, width: int) -> None:
     """Render severity breakdown."""
-    from rich.table import Table
     from rich import box
+    from rich.table import Table
 
     SEVERITY_ICONS = {"Critical": "🔴", "High": "🟠", "Medium": "🟡", "Low": "⚪"}
 

souleyez/ui/help_system.py

@@ -4,9 +4,11 @@ souleyez.ui.help_system - In-app help and tooltips system
 
 Provides context-sensitive help throughout the application.
 """
-import click
+
 from typing import Dict, List, Optional
 
+import click
+
 
 class HelpContext:
     """Enum-like class for help contexts."""
@@ -295,6 +297,7 @@ def show_help(
         engagement_type: Optional engagement preset type for preset-specific tips
     """
     import shutil
+
     from souleyez.ui.design_system import DesignSystem
 
     help_data = HELP_CONTENT.get(context, HELP_CONTENT[HelpContext.GENERAL])

souleyez/ui/intelligence_view.py

@@ -3,9 +3,11 @@
 Intelligence View UI.
 Visual interface for attack surface analysis and exploitation tracking.
 """
-import click
-from typing import Dict, List
+
 import shutil
+from typing import Dict, List
+
+import click
 
 
 def get_terminal_width() -> int:
@@ -328,9 +330,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)
@@ -725,6 +728,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)