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

souleyez/storage/migrations/_020_add_rbac.py

@@ -8,17 +8,16 @@ Tables created:
 - engagement_permissions: Team access to engagements
 """
 
-import secrets
 import hashlib
 import os
+import secrets
 
 
 def upgrade(conn):
     """Add RBAC tables."""
 
     # Users table - core user accounts
-    conn.execute(
-        """
+    conn.execute("""
         CREATE TABLE IF NOT EXISTS users (
             id TEXT PRIMARY KEY,
             username TEXT UNIQUE NOT NULL,
@@ -36,12 +35,10 @@ def upgrade(conn):
             failed_login_attempts INTEGER DEFAULT 0,
             locked_until TIMESTAMP
         )
-    """
-    )
+    """)
 
     # Sessions table - active login sessions
-    conn.execute(
-        """
+    conn.execute("""
         CREATE TABLE IF NOT EXISTS sessions (
             id TEXT PRIMARY KEY,
             user_id TEXT NOT NULL,
@@ -52,12 +49,10 @@ def upgrade(conn):
             user_agent TEXT,
             FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE CASCADE
         )
-    """
-    )
+    """)
 
     # Audit log - immutable security trail
-    conn.execute(
-        """
+    conn.execute("""
         CREATE TABLE IF NOT EXISTS audit_log (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
@@ -70,12 +65,10 @@ def upgrade(conn):
             ip_address TEXT,
             success BOOLEAN DEFAULT TRUE
         )
-    """
-    )
+    """)
 
     # Engagement permissions - team access control
-    conn.execute(
-        """
+    conn.execute("""
         CREATE TABLE IF NOT EXISTS engagement_permissions (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL,
@@ -87,8 +80,7 @@ def upgrade(conn):
             FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE CASCADE,
             UNIQUE(engagement_id, user_id)
         )
-    """
-    )
+    """)
 
     # Add owner_id to engagements table
     try:

souleyez/storage/migrations/_021_wazuh_integration.py

@@ -14,8 +14,7 @@ def upgrade(conn):
     cursor = conn.cursor()
 
     # Wazuh connection configuration
-    cursor.execute(
-        """
+    cursor.execute("""
         CREATE TABLE IF NOT EXISTS wazuh_config (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL UNIQUE,
@@ -31,13 +30,11 @@ def upgrade(conn):
             updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
             FOREIGN KEY (engagement_id) REFERENCES engagements(id) ON DELETE CASCADE
         )
-    """
-    )
+    """)
 
     # Detection validation results per job
     # Note: job_id references jobs.json (file-based), not a SQLite table
-    cursor.execute(
-        """
+    cursor.execute("""
         CREATE TABLE IF NOT EXISTS detection_results (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             job_id INTEGER NOT NULL,
@@ -55,8 +52,7 @@ def upgrade(conn):
             checked_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
             FOREIGN KEY (engagement_id) REFERENCES engagements(id) ON DELETE CASCADE
         )
-    """
-    )
+    """)
 
     # Indexes
     cursor.execute(

souleyez/storage/migrations/_023_fix_detection_results_fk.py

@@ -32,8 +32,7 @@ def upgrade(conn):
         return
 
     # Recreate table without the bad FK
-    cursor.execute(
-        """
+    cursor.execute("""
         CREATE TABLE detection_results_new (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             job_id INTEGER NOT NULL,
@@ -51,8 +50,7 @@ def upgrade(conn):
             checked_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
             FOREIGN KEY (engagement_id) REFERENCES engagements(id) ON DELETE CASCADE
         )
-    """
-    )
+    """)
 
     # Copy existing data
     cursor.execute("INSERT INTO detection_results_new SELECT * FROM detection_results")

souleyez/storage/migrations/_024_wazuh_vulnerabilities.py

@@ -15,8 +15,7 @@ def upgrade(conn):
     cursor = conn.cursor()
 
     # Wazuh vulnerabilities table
-    cursor.execute(
-        """
+    cursor.execute("""
         CREATE TABLE IF NOT EXISTS wazuh_vulnerabilities (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL,
@@ -59,12 +58,10 @@ def upgrade(conn):
 
             UNIQUE(engagement_id, agent_id, cve_id, package_name)
         )
-    """
-    )
+    """)
 
     # Sync metadata table
-    cursor.execute(
-        """
+    cursor.execute("""
         CREATE TABLE IF NOT EXISTS wazuh_vuln_sync (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL UNIQUE,
@@ -74,8 +71,7 @@ def upgrade(conn):
             last_sync_errors TEXT,
             FOREIGN KEY (engagement_id) REFERENCES engagements(id) ON DELETE CASCADE
         )
-    """
-    )
+    """)
 
     # Indexes for performance
     cursor.execute(

souleyez/storage/migrations/_026_add_engagement_scope.py

@@ -17,8 +17,7 @@ def upgrade(conn):
     """Add scope validation tables and columns."""
 
     # Engagement scope definitions table
-    conn.execute(
-        """
+    conn.execute("""
         CREATE TABLE IF NOT EXISTS engagement_scope (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL,
@@ -31,12 +30,10 @@ def upgrade(conn):
             FOREIGN KEY (engagement_id) REFERENCES engagements(id) ON DELETE CASCADE,
             UNIQUE(engagement_id, scope_type, value)
         )
-    """
-    )
+    """)
 
     # Scope validation audit log
-    conn.execute(
-        """
+    conn.execute("""
         CREATE TABLE IF NOT EXISTS scope_validation_log (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL,
@@ -49,8 +46,7 @@ def upgrade(conn):
             created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
             FOREIGN KEY (engagement_id) REFERENCES engagements(id) ON DELETE CASCADE
         )
-    """
-    )
+    """)
 
     # Add scope_enforcement column to engagements
     try:

souleyez/storage/migrations/_027_multi_siem_persistence.py

@@ -26,8 +26,7 @@ def upgrade(conn):
         cursor.execute("ALTER TABLE wazuh_config ADD COLUMN config_json TEXT")
 
     # Create new table with correct constraint
-    cursor.execute(
-        """
+    cursor.execute("""
         CREATE TABLE IF NOT EXISTS siem_config_new (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL,
@@ -46,12 +45,10 @@ def upgrade(conn):
             FOREIGN KEY (engagement_id) REFERENCES engagements(id) ON DELETE CASCADE,
             UNIQUE(engagement_id, siem_type)
         )
-    """
-    )
+    """)
 
     # Copy existing data
-    cursor.execute(
-        """
+    cursor.execute("""
         INSERT OR IGNORE INTO siem_config_new (
             id, engagement_id, siem_type, api_url, api_user, api_password,
             indexer_url, indexer_user, indexer_password, verify_ssl, enabled,
@@ -62,8 +59,7 @@ def upgrade(conn):
             indexer_url, indexer_user, indexer_password, verify_ssl, enabled,
             config_json, created_at, updated_at
         FROM wazuh_config
-    """
-    )
+    """)
 
     # Drop old table and rename new one
     cursor.execute("DROP TABLE wazuh_config")
@@ -87,8 +83,7 @@ def downgrade(conn):
     """Revert to single SIEM per engagement (lossy - keeps only first config per engagement)."""
     cursor = conn.cursor()
 
-    cursor.execute(
-        """
+    cursor.execute("""
         CREATE TABLE IF NOT EXISTS wazuh_config_old (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL UNIQUE,
@@ -106,12 +101,10 @@ def downgrade(conn):
             updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
             FOREIGN KEY (engagement_id) REFERENCES engagements(id) ON DELETE CASCADE
         )
-    """
-    )
+    """)
 
     # Copy only first config per engagement
-    cursor.execute(
-        """
+    cursor.execute("""
         INSERT OR IGNORE INTO wazuh_config_old (
             engagement_id, api_url, api_user, api_password,
             indexer_url, indexer_user, indexer_password, verify_ssl, enabled,
@@ -123,8 +116,7 @@ def downgrade(conn):
             siem_type, config_json, created_at, updated_at
         FROM wazuh_config
         GROUP BY engagement_id
-    """
-    )
+    """)
 
     cursor.execute("DROP TABLE wazuh_config")
     cursor.execute("ALTER TABLE wazuh_config_old RENAME TO wazuh_config")

souleyez/storage/migrations/__init__.py

@@ -3,8 +3,7 @@
 
 # Import all migrations - they register themselves
 from . import (
-    # 001-010
-    _001_add_credential_enhancements,
+    _001_add_credential_enhancements,  # 001-010; 011-020; 021-030
     _002_add_status_tracking,
     _003_add_execution_log,
     _005_screenshots,
@@ -13,7 +12,6 @@ from . import (
     _008_add_nuclei_table,
     _009_add_cme_tables,
     _010_evidence_linking,
-    # 011-020
     _011_timeline_tracking,
     _012_team_collaboration,
     _013_add_host_tags,
@@ -24,7 +22,6 @@ from . import (
     _018_add_osint_target,
     _019_add_engagement_type,
     _020_add_rbac,
-    # 021-030
     _021_wazuh_integration,
     _022_wazuh_indexer_columns,
     _023_fix_detection_results_fk,

souleyez/storage/migrations/migration_manager.py

@@ -5,8 +5,9 @@ souleyez.storage.migrations.migration_manager - Database migration system
 Uses compiled registry pattern for Nuitka compatibility.
 All migrations are imported in __init__.py and registered in MIGRATIONS_REGISTRY.
 """
+
 import sqlite3
-from typing import List, Dict
+from typing import Dict, List
 
 # Import registry from package
 from . import MIGRATIONS_REGISTRY, get_all_versions
@@ -20,16 +21,14 @@ class MigrationManager:
     def _ensure_migrations_table(self):
         """Create migrations tracking table if it doesn't exist."""
         conn = sqlite3.connect(self.db_path)
-        conn.execute(
-            """
+        conn.execute("""
             CREATE TABLE IF NOT EXISTS schema_migrations (
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
                 version TEXT UNIQUE NOT NULL,
                 name TEXT NOT NULL,
                 applied_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
             )
-        """
-        )
+        """)
         conn.commit()
         conn.close()
 
@@ -199,13 +198,11 @@ class MigrationManager:
         if applied:
             print("✅ Applied:")
             conn = sqlite3.connect(self.db_path)
-            cursor = conn.execute(
-                """
+            cursor = conn.execute("""
                 SELECT version, name, applied_at
                 FROM schema_migrations
                 ORDER BY version
-            """
-            )
+            """)
             for row in cursor.fetchall():
                 print(f"  [{row[0]}] {row[1]} (applied: {row[2]})")
             conn.close()

souleyez/storage/msf_sessions.py

@@ -5,8 +5,8 @@ Functions for managing Metasploit Framework session records in the database.
 """
 
 import logging
-from typing import Optional, List, Dict, Any
 from datetime import datetime
+from typing import Any, Dict, List, Optional
 
 logger = logging.getLogger(__name__)
 

souleyez/storage/osint.py

@@ -2,7 +2,9 @@
 """
 souleyez.storage.osint - OSINT data management
 """
-from typing import List, Dict, Any, Optional
+
+from typing import Any, Dict, List, Optional
+
 from .database import get_db
 
 

souleyez/storage/recommendation_engine.py

@@ -2,11 +2,12 @@
 AI-powered recommendation engine for deliverables.
 """
 
-from typing import Dict, List, Optional
 from datetime import datetime, timedelta
+from typing import Dict, List, Optional
+
 from .database import get_db
-from .deliverables import DeliverableManager
 from .deliverable_evidence import EvidenceManager
+from .deliverables import DeliverableManager
 from .timeline_tracker import TimelineTracker
 
 

souleyez/storage/screenshots.py

@@ -6,7 +6,8 @@ import os
 import shutil
 import time
 from pathlib import Path
-from typing import List, Dict, Optional
+from typing import Dict, List, Optional
+
 from .database import get_db
 
 SCREENSHOTS_DIR = Path("~/.souleyez/screenshots").expanduser()

souleyez/storage/smb_shares.py

@@ -2,7 +2,9 @@
 """
 souleyez.storage.smb_shares - SMB shares storage and retrieval
 """
-from typing import Dict, List, Any, Optional
+
+from typing import Any, Dict, List, Optional
+
 from .database import get_db
 
 

souleyez/storage/sqlmap_data.py

@@ -4,12 +4,14 @@ souleyez.storage.sqlmap_data
 
 Storage manager for SQLMap SQL injection discoveries (databases, tables, columns, dumped data)
 """
-import sqlite3
+
 import json
-from typing import Optional, List, Dict, Any
-from souleyez.storage.database import Database
-from souleyez.storage.crypto import get_crypto_manager
+import sqlite3
+from typing import Any, Dict, List, Optional
+
 from souleyez.log_config import get_logger
+from souleyez.storage.crypto import get_crypto_manager
+from souleyez.storage.database import Database
 
 logger = get_logger(__name__)
 

souleyez/storage/team_collaboration.py

@@ -2,10 +2,11 @@
 Team collaboration features for deliverables.
 """
 
-from typing import Dict, List, Optional
+import os
 from datetime import datetime
+from typing import Dict, List, Optional
+
 from .database import get_db
-import os
 
 
 class TeamCollaboration:

souleyez/storage/timeline_tracker.py

@@ -2,8 +2,9 @@
 Timeline tracking and velocity calculations for deliverables.
 """
 
-from typing import Dict, List, Optional
 from datetime import datetime, timedelta
+from typing import Dict, List, Optional
+
 from .database import get_db
 
 

souleyez/storage/wazuh_vulns.py

@@ -5,9 +5,11 @@ souleyez.storage.wazuh_vulns - Wazuh vulnerabilities database operations
 Stores vulnerabilities discovered by Wazuh agents for gap analysis
 between passive (agent-based) and active (scan-based) detection.
 """
+
 import json
 from datetime import datetime
-from typing import List, Dict, Any, Optional
+from typing import Any, Dict, List, Optional
+
 from .database import get_db
 
 

souleyez/storage/web_paths.py

@@ -2,7 +2,9 @@
 """
 souleyez.storage.web_paths - Web paths/directories management
 """
-from typing import List, Dict, Any, Optional
+
+from typing import Any, Dict, List, Optional
+
 from .database import get_db
 
 

souleyez/testing/credential_tester.py

@@ -2,9 +2,11 @@
 """
 souleyez.testing.credential_tester - Test credentials against services
 """
+
 import socket
 import subprocess
 from typing import Dict, List
+
 import click
 
 

souleyez/ui/__init__.py

@@ -2,7 +2,8 @@
 """
 souleyez.ui - User interface components
 """
-from .interactive import run_interactive_menu
+
 from .dashboard import run_dashboard
+from .interactive import run_interactive_menu
 
 __all__ = ["run_interactive_menu", "run_dashboard"]

souleyez/ui/ai_quotes.py

@@ -5,8 +5,8 @@ souleyez.ui.ai_quotes - Entertaining quotes for AI generation progress
 Displays rotating quotes during long AI operations to keep users entertained
 and informed that the system is still working.
 """
-import random
 
+import random
 
 # Hacker humor and pentesting jokes
 HACKER_HUMOR = [