souleyez 2.43.34__py3-none-any.whl → 3.0.7__py3-none-any.whl

souleyez/__init__.py

@@ -1 +1 @@
-__version__ = "2.43.34"
+__version__ = "3.0.7"

souleyez/ai/__init__.py

@@ -2,18 +2,18 @@
 souleyez.ai - AI-powered attack path recommendations and report generation
 """
 
-from .ollama_service import OllamaService, OLLAMA_AVAILABLE
-from .context_builder import ContextBuilder
-from .recommender import AttackRecommender
-from .llm_provider import LLMProvider, LLMProviderType
-from .ollama_provider import OllamaProvider
 from .claude_provider import (
-    ClaudeProvider,
     ANTHROPIC_AVAILABLE,
-    set_claude_api_key,
+    ClaudeProvider,
     clear_claude_api_key,
+    set_claude_api_key,
 )
+from .context_builder import ContextBuilder
 from .llm_factory import LLMFactory
+from .llm_provider import LLMProvider, LLMProviderType
+from .ollama_provider import OllamaProvider
+from .ollama_service import OLLAMA_AVAILABLE, OllamaService
+from .recommender import AttackRecommender
 from .report_context import ReportContextBuilder
 from .report_service import AIReportService
 

souleyez/ai/action_mapper.py

@@ -2,9 +2,10 @@
 """
 souleyez.ai.action_mapper - Map AI recommendations to executable commands
 """
-import re
-from typing import Optional, Dict, Any
+
 import logging
+import re
+from typing import Any, Dict, Optional
 
 logger = logging.getLogger(__name__)
 

souleyez/ai/chain_advisor.py

@@ -5,11 +5,12 @@ souleyez.ai.chain_advisor - AI-powered tool chain recommendations
 Uses LLM to analyze scan results and suggest additional tools to run,
 complementing the static rule-based chaining system.
 """
+
 import logging
 import re
 from dataclasses import dataclass, field
 from enum import Enum
-from typing import Dict, Any, List, Optional
+from typing import Any, Dict, List, Optional
 
 from .llm_factory import LLMFactory
 

souleyez/ai/claude_provider.py

@@ -6,7 +6,7 @@ Requires API key stored securely via CryptoManager.
 """
 
 import logging
-from typing import Optional, Dict, Any
+from typing import Any, Dict, Optional
 
 from .llm_provider import LLMProvider, LLMProviderType
 
@@ -240,8 +240,8 @@ def set_claude_api_key(api_key: str) -> bool:
     Returns:
         bool: True if stored successfully, False otherwise
     """
-    from souleyez.storage.crypto import get_crypto_manager
     from souleyez.config import read_config, write_config
+    from souleyez.storage.crypto import get_crypto_manager
 
     crypto = get_crypto_manager()
     if not crypto.is_unlocked():

souleyez/ai/context_builder.py

@@ -2,11 +2,13 @@
 """
 souleyez.ai.context_builder - Build context from engagement data for LLM
 """
+
 from typing import Optional
-from ..storage.engagements import EngagementManager
-from ..storage.hosts import HostManager
+
 from ..storage.credentials import CredentialsManager
+from ..storage.engagements import EngagementManager
 from ..storage.findings import FindingsManager
+from ..storage.hosts import HostManager
 
 
 class ContextBuilder:

souleyez/ai/executor.py

@@ -2,16 +2,19 @@
 """
 souleyez.ai.executor - Interactive AI-driven command execution
 """
-import subprocess
+
 import logging
+import subprocess
+from typing import Any, Dict, Optional
+
 import click
-from typing import Optional, Dict, Any
-from .recommender import AttackRecommender
+
+from ..storage.execution_log import ExecutionLogManager
 from .action_mapper import ActionMapper
-from .safety import SafetyFramework, ApprovalMode, RiskLevel
-from .result_parser import ResultParser
 from .feedback_handler import FeedbackHandler
-from ..storage.execution_log import ExecutionLogManager
+from .recommender import AttackRecommender
+from .result_parser import ResultParser
+from .safety import ApprovalMode, RiskLevel, SafetyFramework
 
 logger = logging.getLogger(__name__)
 

souleyez/ai/feedback_handler.py

@@ -2,11 +2,13 @@
 """
 souleyez.ai.feedback_handler - Auto-update database after command execution
 """
+
 import logging
-from typing import Dict, Any, Optional
 from datetime import datetime
-from ..storage.hosts import HostManager
+from typing import Any, Dict, Optional
+
 from ..storage.credentials import CredentialsManager
+from ..storage.hosts import HostManager
 
 logger = logging.getLogger(__name__)
 

souleyez/ai/llm_provider.py

@@ -5,10 +5,10 @@ This module defines the abstract base class for LLM providers,
 enabling support for multiple backends (Ollama, Claude, etc.)
 """
 
+import logging
 from abc import ABC, abstractmethod
 from enum import Enum
-from typing import Optional, Dict, Any
-import logging
+from typing import Any, Dict, Optional
 
 logger = logging.getLogger(__name__)
 

souleyez/ai/ollama_provider.py

@@ -5,10 +5,10 @@ Wraps the existing OllamaService to implement the LLMProvider interface.
 """
 
 import logging
-from typing import Optional, Dict, Any
+from typing import Any, Dict, Optional
 
 from .llm_provider import LLMProvider, LLMProviderType
-from .ollama_service import OllamaService, OLLAMA_AVAILABLE
+from .ollama_service import OLLAMA_AVAILABLE, OllamaService
 
 logger = logging.getLogger(__name__)
 

souleyez/ai/ollama_service.py

@@ -7,7 +7,7 @@ without sending sensitive engagement data to the cloud.
 """
 
 import logging
-from typing import Optional, Dict, Any
+from typing import Any, Dict, Optional
 
 # Ollama is optional - not available in Ubuntu repos
 try:
@@ -252,40 +252,24 @@ class OllamaService:
             return None
 
         try:
-            import threading
+            import concurrent.futures
 
-            # Check if we're in main thread (signal only works in main thread)
-            in_main_thread = threading.current_thread() is threading.main_thread()
-
-            if in_main_thread:
-                # Use signal-based timeout (only works in main thread)
-                import signal
-
-                def timeout_handler(signum, frame):
-                    raise TimeoutError("AI generation timed out")
-
-                old_handler = signal.signal(signal.SIGALRM, timeout_handler)
-                signal.alarm(timeout)
+            # Use thread-based timeout instead of signals
+            # Signals can interrupt unrelated code (like click.prompt) causing crashes
+            def _do_generate():
+                return self.client.generate(model=model_name, prompt=prompt)
 
+            with concurrent.futures.ThreadPoolExecutor(max_workers=1) as executor:
+                future = executor.submit(_do_generate)
                 try:
-                    response = self.client.generate(model=model_name, prompt=prompt)
-                    signal.alarm(0)
-                    signal.signal(signal.SIGALRM, old_handler)
+                    response = future.result(timeout=timeout)
                     return response.get("response", "")
-                except TimeoutError:
-                    signal.alarm(0)
-                    signal.signal(signal.SIGALRM, old_handler)
+                except concurrent.futures.TimeoutError:
                     logger.error(f"Generation timed out after {timeout}s")
                     return None
                 except Exception as e:
-                    signal.alarm(0)
-                    signal.signal(signal.SIGALRM, old_handler)
                     logger.error(f"Generation error: {e}")
                     return None
-            else:
-                # In a thread - can't use signals, just call directly
-                response = self.client.generate(model=model_name, prompt=prompt)
-                return response.get("response", "")
         except Exception as e:
             logger.error(f"Generation failed: {e}")
             return None

souleyez/ai/path_scorer.py

@@ -2,8 +2,9 @@
 """
 souleyez.ai.path_scorer - Score and rank attack paths
 """
+
 import logging
-from typing import Dict, List, Any
+from typing import Any, Dict, List
 
 logger = logging.getLogger(__name__)
 

souleyez/ai/recommender.py

@@ -5,12 +5,14 @@ souleyez.ai.recommender - AI-powered attack path recommendations
 Uses LLM (via Ollama or Claude) to analyze engagement data and suggest
 the next most promising penetration testing step.
 """
+
 import logging
 import re
-from typing import Optional, Dict, Any, List
-from .ollama_service import OllamaService
+from typing import Any, Dict, List, Optional
+
 from .context_builder import ContextBuilder
 from .llm_factory import LLMFactory
+from .ollama_service import OllamaService
 
 logger = logging.getLogger(__name__)
 
@@ -585,10 +587,10 @@ Be specific and actionable. Think like a professional pentester considering mult
 
     def _get_engagement_data(self, engagement_id: int) -> Dict[str, Any]:
         """Get engagement data for scoring."""
-        from ..storage.engagements import EngagementManager
-        from ..storage.hosts import HostManager
         from ..storage.credentials import CredentialsManager
+        from ..storage.engagements import EngagementManager
         from ..storage.findings import FindingsManager
+        from ..storage.hosts import HostManager
 
         em = EngagementManager()
         hm = HostManager()

souleyez/ai/report_context.py

@@ -5,7 +5,7 @@ Prepares engagement data in formats suitable for LLM prompt templates.
 """
 
 import logging
-from typing import Dict, Any, List, Optional
+from typing import Any, Dict, List, Optional
 
 logger = logging.getLogger(__name__)
 
@@ -19,10 +19,10 @@ class ReportContextBuilder:
     """
 
     def __init__(self):
+        from souleyez.storage.credentials import CredentialsManager
         from souleyez.storage.engagements import EngagementManager
         from souleyez.storage.findings import FindingsManager
         from souleyez.storage.hosts import HostManager
-        from souleyez.storage.credentials import CredentialsManager
 
         self.em = EngagementManager()
         self.fm = FindingsManager()

souleyez/ai/report_service.py

@@ -5,21 +5,21 @@ Provides methods for generating AI-enhanced report sections using
 configured LLM providers (Claude or Ollama).
 """
 
+import concurrent.futures
 import logging
 import re
-import concurrent.futures
-from typing import Optional, Dict, Any, List
+from typing import Any, Dict, List, Optional
 
-from .llm_provider import LLMProvider
 from .llm_factory import LLMFactory
+from .llm_provider import LLMProvider
 from .report_context import ReportContextBuilder
 from .report_prompts import (
-    REPORT_SYSTEM_PROMPT,
+    ATTACK_CHAIN_PROMPT,
     EXECUTIVE_SUMMARY_PROMPT,
     FINDING_ENHANCEMENT_PROMPT,
     REMEDIATION_PLAN_PROMPT,
+    REPORT_SYSTEM_PROMPT,
     RISK_RATING_PROMPT,
-    ATTACK_CHAIN_PROMPT,
 )
 
 logger = logging.getLogger(__name__)

souleyez/ai/result_parser.py

@@ -2,9 +2,10 @@
 """
 souleyez.ai.result_parser - Parse command execution results
 """
-import re
-from typing import Dict, Any, Optional
+
 import logging
+import re
+from typing import Any, Dict, Optional
 
 logger = logging.getLogger(__name__)
 

souleyez/ai/safety.py

@@ -2,9 +2,12 @@
 """
 souleyez.ai.safety - Safety framework for AI-driven command execution
 """
-import click
-from typing import Dict, Optional
+
 from enum import Enum
+from typing import Dict, Optional
+
+import click
+
 from souleyez.ui.design_system import DesignSystem
 
 

souleyez/auth/__init__.py

@@ -10,20 +10,20 @@ Usage:
         pass
 """
 
+from .audit import AuditAction, AuditLogger, audit_log, get_audit_logger
 from .permissions import (
-    Role,
-    Tier,
+    PRO_TIER_PERMISSIONS,
+    ROLE_PERMISSIONS,
     Permission,
     PermissionChecker,
+    Role,
+    Tier,
     requires_permission,
     requires_pro,
     requires_role,
-    PRO_TIER_PERMISSIONS,
-    ROLE_PERMISSIONS,
 )
-from .user_manager import User, UserManager
 from .session_manager import Session, SessionManager
-from .audit import AuditLogger, AuditAction, audit_log, get_audit_logger
+from .user_manager import User, UserManager
 
 # Module-level session manager instance (initialized on first use)
 _session_manager: SessionManager = None

souleyez/auth/audit.py

@@ -5,11 +5,11 @@ All sensitive actions should be logged through this module.
 Logs are immutable and include user context automatically.
 """
 
-import sqlite3
 import json
+import sqlite3
 from datetime import datetime, timedelta
-from typing import Optional, Dict, Any, List
 from enum import Enum
+from typing import Any, Dict, List, Optional
 
 from souleyez.storage.database import get_db
 

souleyez/auth/engagement_access.py

@@ -8,12 +8,12 @@ Permission levels:
 """
 
 import sqlite3
-from enum import Enum
-from typing import Optional, List, Dict, Any
 from dataclasses import dataclass
 from datetime import datetime
+from enum import Enum
+from typing import Any, Dict, List, Optional
 
-from souleyez.auth import get_current_user, Role
+from souleyez.auth import Role, get_current_user
 
 
 class EngagementPermission(Enum):
@@ -131,13 +131,11 @@ class EngagementAccessManager:
 
         if user_role == Role.ADMIN:
             # Admins see everything
-            rows = conn.execute(
-                """
+            rows = conn.execute("""
                 SELECT e.*, 'admin' as permission_level
                 FROM engagements e
                 ORDER BY e.created_at DESC
-            """
-            ).fetchall()
+            """).fetchall()
         else:
             # Non-admins see owned + shared engagements
             rows = conn.execute(

souleyez/auth/permissions.py

@@ -6,8 +6,8 @@ Tiers: FREE, PRO (for licensing)
 """
 
 from enum import Enum, auto
-from typing import Set, Optional
 from functools import wraps
+from typing import Optional, Set
 
 
 class Role(Enum):

souleyez/auth/session_manager.py

@@ -7,18 +7,17 @@ Handles:
 - Current user context
 """
 
-import sqlite3
-import secrets
 import hashlib
 import json
+import secrets
+import sqlite3
+from dataclasses import dataclass
 from datetime import datetime, timedelta
 from pathlib import Path
 from typing import Optional
-from dataclasses import dataclass
 
-from .user_manager import User, UserManager
 from .permissions import Role, Tier
-
+from .user_manager import User, UserManager
 
 # Session configuration
 SESSION_TOKEN_BYTES = 32
@@ -250,6 +249,7 @@ class SessionManager:
             vault_failures: Number of recent vault unlock failures
         """
         import time
+
         import click
 
         if vault_failures >= 2:

souleyez/auth/user_manager.py

@@ -8,17 +8,16 @@ Handles:
 - Tier management for licensing
 """
 
-import sqlite3
 import hashlib
-import secrets
 import re
-from datetime import datetime
-from typing import Optional, List
+import secrets
+import sqlite3
 from dataclasses import dataclass
+from datetime import datetime
+from typing import List, Optional
 
 from .permissions import Role, Tier
 
-
 # OWASP 2023 recommendation for PBKDF2-HMAC-SHA256
 HASH_ITERATIONS = 480_000
 SALT_LENGTH = 32

souleyez/commands/audit.py

@@ -8,17 +8,18 @@ Commands:
 - souleyez audit export         - Export audit logs
 """
 
-import click
+import json
 from datetime import datetime
+
+import click
 from rich.console import Console
-from rich.table import Table
 from rich.panel import Panel
-import json
+from rich.table import Table
 
-from souleyez.security import require_login
 from souleyez.auth import Role
-from souleyez.auth.permissions import requires_role
 from souleyez.auth.audit import get_audit_logger
+from souleyez.auth.permissions import requires_role
+from souleyez.security import require_login
 
 console = Console()
 

souleyez/commands/auth.py

@@ -7,19 +7,20 @@ Commands:
 - souleyez whoami    - Show current user info
 """
 
-import click
 import getpass
+
+import click
 from rich.console import Console
 from rich.panel import Panel
 from rich.table import Table
 
 from souleyez.auth import (
-    init_auth,
-    get_session_manager,
+    Tier,
+    UserManager,
     get_current_user,
+    get_session_manager,
+    init_auth,
     is_logged_in,
-    UserManager,
-    Tier,
 )
 from souleyez.storage.database import get_db
 

souleyez/commands/deliverables.py

@@ -4,17 +4,16 @@ CLI commands for deliverable tracking.
 
 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
 except ImportError:
     TaskProgressColumn = None  # Not available in older rich versions
+from souleyez.security import require_password
 from souleyez.storage.deliverables import DeliverableManager
 from souleyez.storage.engagements import EngagementManager
-from souleyez.security import require_password
-
 
 console = Console()
 

souleyez/commands/engagement.py

@@ -12,14 +12,14 @@ import click
 from rich.console import Console
 from rich.table import Table
 
-from souleyez.security import require_login
-from souleyez.auth import get_current_user, Role, UserManager
+from souleyez.auth import Role, UserManager, get_current_user
 from souleyez.auth.engagement_access import (
     EngagementAccessManager,
     EngagementPermission,
 )
-from souleyez.storage.engagements import EngagementManager
+from souleyez.security import require_login
 from souleyez.storage.database import get_db
+from souleyez.storage.engagements import EngagementManager
 
 console = Console()
 

souleyez/commands/license.py

@@ -7,9 +7,10 @@ souleyez license deactivate       - Remove license
 souleyez license machine-id       - Show machine ID for hardware-bound licenses
 """
 
-import click
 from datetime import datetime
 
+import click
+
 
 @click.group()
 def license():
@@ -59,7 +60,7 @@ def activate(license_key: str):
 
         # Update user tier if auth system is in use
         try:
-            from souleyez.auth import get_current_user, UserManager
+            from souleyez.auth import UserManager, get_current_user
             from souleyez.auth.permissions import Tier
             from souleyez.storage.database import get_db
 

souleyez/commands/screenshots.py

@@ -2,14 +2,15 @@
 CLI commands for screenshot management.
 """
 
-import click
 from pathlib import Path
+
+import click
 from rich.console import Console
 from rich.table import Table
-from souleyez.storage.screenshots import ScreenshotManager
-from souleyez.storage.engagements import EngagementManager
-from souleyez.security import require_password
 
+from souleyez.security import require_password
+from souleyez.storage.engagements import EngagementManager
+from souleyez.storage.screenshots import ScreenshotManager
 
 console = Console()
 

souleyez/commands/user.py

@@ -9,23 +9,24 @@ Commands (admin only):
 - souleyez user passwd [username]   - Change password
 """
 
-import click
 import getpass
+
+import click
 from rich.console import Console
-from rich.table import Table
 from rich.panel import Panel
+from rich.table import Table
 
 from souleyez.auth import (
-    init_auth,
-    get_current_user,
-    is_logged_in,
-    UserManager,
     Role,
     Tier,
+    UserManager,
+    get_current_user,
+    init_auth,
+    is_logged_in,
 )
-from souleyez.storage.database import get_db
-from souleyez.security import require_login, require_admin
 from souleyez.licensing.validator import get_active_license
+from souleyez.security import require_admin, require_login
+from souleyez.storage.database import get_db
 
 console = Console()
 
@@ -410,6 +411,7 @@ def _log_audit(action: str, user_id: str, username: str, details: str = None):
     """Log an audit event."""
     import sqlite3
     from datetime import datetime
+
     from souleyez.storage.database import get_db
 
     try: