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

souleyez/storage/migrations/_013_add_host_tags.py

@@ -1,26 +1,32 @@
 """
 Migration: Add tags column to hosts table
 
-This migration adds support for host tagging, allowing users to 
+This migration adds support for host tagging, allowing users to
 categorize and organize hosts with custom labels.
 """
 
+
 def upgrade(db):
     """Add tags column to hosts table."""
     # Check if column already exists
     cursor = db.execute("PRAGMA table_info(hosts)")
     columns = [row[1] for row in cursor.fetchall()]
 
-    if 'tags' not in columns:
-        db.execute("""
+    if "tags" not in columns:
+        db.execute(
+            """
             ALTER TABLE hosts
             ADD COLUMN tags TEXT DEFAULT NULL
-        """)
+        """
+        )
     # If column already exists, migration is idempotent - do nothing
-    
+
+
 def downgrade(db):
     """
     SQLite doesn't support DROP COLUMN before version 3.35.0.
     For now, downgrade is not supported.
     """
-    raise NotImplementedError("Downgrade not supported for SQLite ALTER TABLE ADD COLUMN")
+    raise NotImplementedError(
+        "Downgrade not supported for SQLite ALTER TABLE ADD COLUMN"
+    )

souleyez/storage/migrations/_014_exploit_attempts.py

@@ -6,9 +6,11 @@ their outcomes, and related metadata. This enables filtering by attempt status
 and provides a historical record of exploitation activities.
 """
 
+
 def upgrade(db):
     """Create exploit_attempts table."""
-    db.execute("""
+    db.execute(
+        """
         CREATE TABLE IF NOT EXISTS exploit_attempts (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL,
@@ -26,28 +28,38 @@ def upgrade(db):
             FOREIGN KEY (service_id) REFERENCES services(id) ON DELETE SET NULL,
             UNIQUE(engagement_id, host_id, service_id, exploit_identifier)
         )
-    """)
+    """
+    )
 
     # Create indexes for efficient querying
-    db.execute("""
+    db.execute(
+        """
         CREATE INDEX IF NOT EXISTS idx_exploit_attempts_engagement
         ON exploit_attempts(engagement_id)
-    """)
+    """
+    )
 
-    db.execute("""
+    db.execute(
+        """
         CREATE INDEX IF NOT EXISTS idx_exploit_attempts_host
         ON exploit_attempts(host_id)
-    """)
+    """
+    )
 
-    db.execute("""
+    db.execute(
+        """
         CREATE INDEX IF NOT EXISTS idx_exploit_attempts_status
         ON exploit_attempts(status)
-    """)
+    """
+    )
 
-    db.execute("""
+    db.execute(
+        """
         CREATE INDEX IF NOT EXISTS idx_exploit_attempts_identifier
         ON exploit_attempts(exploit_identifier)
-    """)
+    """
+    )
+
 
 def downgrade(db):
     """Drop exploit_attempts table and indexes."""

souleyez/storage/migrations/_015_add_mac_os_fields.py

@@ -5,27 +5,35 @@ This migration adds support for storing MAC addresses and OS detection
 confidence scores from Nmap scans.
 """
 
+
 def upgrade(db):
     """Add mac_address and os_accuracy columns to hosts table."""
     # Check if columns already exist
     cursor = db.execute("PRAGMA table_info(hosts)")
     columns = [row[1] for row in cursor.fetchall()]
 
-    if 'mac_address' not in columns:
-        db.execute("""
+    if "mac_address" not in columns:
+        db.execute(
+            """
             ALTER TABLE hosts
             ADD COLUMN mac_address TEXT DEFAULT NULL
-        """)
+        """
+        )
 
-    if 'os_accuracy' not in columns:
-        db.execute("""
+    if "os_accuracy" not in columns:
+        db.execute(
+            """
             ALTER TABLE hosts
             ADD COLUMN os_accuracy INTEGER DEFAULT NULL
-        """)
+        """
+        )
+
 
 def downgrade(db):
     """
     SQLite doesn't support DROP COLUMN before version 3.35.0.
     For now, downgrade is not supported.
     """
-    raise NotImplementedError("Downgrade not supported for SQLite ALTER TABLE ADD COLUMN")
+    raise NotImplementedError(
+        "Downgrade not supported for SQLite ALTER TABLE ADD COLUMN"
+    )

souleyez/storage/migrations/_016_add_domain_field.py

@@ -6,21 +6,27 @@ by tools like CrackMapExec/NetExec, which is needed for Active Directory
 attacks like GetNPUsers.
 """
 
+
 def upgrade(db):
     """Add domain column to hosts table."""
     # Check if column already exists
     cursor = db.execute("PRAGMA table_info(hosts)")
     columns = [row[1] for row in cursor.fetchall()]
 
-    if 'domain' not in columns:
-        db.execute("""
+    if "domain" not in columns:
+        db.execute(
+            """
             ALTER TABLE hosts
             ADD COLUMN domain TEXT DEFAULT NULL
-        """)
+        """
+        )
+
 
 def downgrade(db):
     """
     SQLite doesn't support DROP COLUMN before version 3.35.0.
     For now, downgrade is not supported.
     """
-    raise NotImplementedError("Downgrade not supported for SQLite ALTER TABLE ADD COLUMN")
+    raise NotImplementedError(
+        "Downgrade not supported for SQLite ALTER TABLE ADD COLUMN"
+    )

souleyez/storage/migrations/_017_msf_sessions.py

@@ -13,7 +13,8 @@ def upgrade(conn):
     cursor = conn.cursor()
 
     # Create msf_sessions table
-    cursor.execute("""
+    cursor.execute(
+        """
         CREATE TABLE IF NOT EXISTS msf_sessions (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL,
@@ -39,23 +40,30 @@ def upgrade(conn):
             FOREIGN KEY (host_id) REFERENCES hosts(id) ON DELETE CASCADE,
             UNIQUE(engagement_id, msf_session_id)
         )
-    """)
+    """
+    )
 
     # Create index for faster lookups
-    cursor.execute("""
+    cursor.execute(
+        """
         CREATE INDEX IF NOT EXISTS idx_msf_sessions_engagement
         ON msf_sessions(engagement_id)
-    """)
+    """
+    )
 
-    cursor.execute("""
+    cursor.execute(
+        """
         CREATE INDEX IF NOT EXISTS idx_msf_sessions_host
         ON msf_sessions(host_id)
-    """)
+    """
+    )
 
-    cursor.execute("""
+    cursor.execute(
+        """
         CREATE INDEX IF NOT EXISTS idx_msf_sessions_active
         ON msf_sessions(is_active)
-    """)
+    """
+    )
 
     conn.commit()
 

souleyez/storage/migrations/_018_add_osint_target.py

@@ -15,18 +15,22 @@ def upgrade(conn):
     # Check if column already exists
     cursor.execute("PRAGMA table_info(osint_data)")
     columns = [row[1] for row in cursor.fetchall()]
-    
+
     # Add target column only if it doesn't exist
-    if 'target' not in columns:
-        cursor.execute("""
+    if "target" not in columns:
+        cursor.execute(
+            """
             ALTER TABLE osint_data ADD COLUMN target TEXT
-        """)
+        """
+        )
 
     # Create index for faster lookups by target
-    cursor.execute("""
+    cursor.execute(
+        """
         CREATE INDEX IF NOT EXISTS idx_osint_target
         ON osint_data(target)
-    """)
+    """
+    )
 
     conn.commit()
 

souleyez/storage/migrations/_019_add_engagement_type.py

@@ -15,18 +15,22 @@ def upgrade(conn):
     # Check if column already exists
     cursor.execute("PRAGMA table_info(engagements)")
     columns = [row[1] for row in cursor.fetchall()]
-    
+
     # Add engagement_type column only if it doesn't exist
-    if 'engagement_type' not in columns:
-        cursor.execute("""
+    if "engagement_type" not in columns:
+        cursor.execute(
+            """
             ALTER TABLE engagements ADD COLUMN engagement_type TEXT DEFAULT 'custom'
-        """)
+        """
+        )
 
     # Create index for faster lookups by type
-    cursor.execute("""
+    cursor.execute(
+        """
         CREATE INDEX IF NOT EXISTS idx_engagements_type
         ON engagements(engagement_type)
-    """)
+    """
+    )
 
     conn.commit()
 

souleyez/storage/migrations/_020_add_rbac.py

@@ -7,6 +7,7 @@ Tables created:
 - audit_log: Security audit trail
 - engagement_permissions: Team access to engagements
 """
+
 import secrets
 import hashlib
 import os
@@ -16,7 +17,8 @@ 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,
@@ -34,10 +36,12 @@ 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,
@@ -48,10 +52,12 @@ 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,
@@ -64,10 +70,12 @@ 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,
@@ -79,7 +87,8 @@ 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:
@@ -89,17 +98,29 @@ def upgrade(conn):
 
     # Indexes for performance
     conn.execute("CREATE INDEX IF NOT EXISTS idx_sessions_user ON sessions(user_id)")
-    conn.execute("CREATE INDEX IF NOT EXISTS idx_sessions_token ON sessions(token_hash)")
-    conn.execute("CREATE INDEX IF NOT EXISTS idx_sessions_expires ON sessions(expires_at)")
-    conn.execute("CREATE INDEX IF NOT EXISTS idx_audit_user ON audit_log(user_id, timestamp DESC)")
-    conn.execute("CREATE INDEX IF NOT EXISTS idx_audit_action ON audit_log(action, timestamp DESC)")
-    conn.execute("CREATE INDEX IF NOT EXISTS idx_audit_timestamp ON audit_log(timestamp DESC)")
-    conn.execute("CREATE INDEX IF NOT EXISTS idx_engagement_perms ON engagement_permissions(user_id, engagement_id)")
+    conn.execute(
+        "CREATE INDEX IF NOT EXISTS idx_sessions_token ON sessions(token_hash)"
+    )
+    conn.execute(
+        "CREATE INDEX IF NOT EXISTS idx_sessions_expires ON sessions(expires_at)"
+    )
+    conn.execute(
+        "CREATE INDEX IF NOT EXISTS idx_audit_user ON audit_log(user_id, timestamp DESC)"
+    )
+    conn.execute(
+        "CREATE INDEX IF NOT EXISTS idx_audit_action ON audit_log(action, timestamp DESC)"
+    )
+    conn.execute(
+        "CREATE INDEX IF NOT EXISTS idx_audit_timestamp ON audit_log(timestamp DESC)"
+    )
+    conn.execute(
+        "CREATE INDEX IF NOT EXISTS idx_engagement_perms ON engagement_permissions(user_id, engagement_id)"
+    )
 
     conn.commit()
 
     # Check if we should print (not in silent mode)
-    if not os.environ.get('SOULEYEZ_MIGRATION_SILENT'):
+    if not os.environ.get("SOULEYEZ_MIGRATION_SILENT"):
         print("✅ Migration 020: RBAC tables created")
 
 

souleyez/storage/migrations/_021_wazuh_integration.py

@@ -14,7 +14,8 @@ 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,
@@ -30,11 +31,13 @@ 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,
@@ -52,13 +55,22 @@ def upgrade(conn):
             checked_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
             FOREIGN KEY (engagement_id) REFERENCES engagements(id) ON DELETE CASCADE
         )
-    """)
+    """
+    )
 
     # Indexes
-    cursor.execute("CREATE INDEX IF NOT EXISTS idx_wazuh_config_engagement ON wazuh_config(engagement_id)")
-    cursor.execute("CREATE INDEX IF NOT EXISTS idx_detection_results_job ON detection_results(job_id)")
-    cursor.execute("CREATE INDEX IF NOT EXISTS idx_detection_results_engagement ON detection_results(engagement_id)")
-    cursor.execute("CREATE INDEX IF NOT EXISTS idx_detection_results_status ON detection_results(detection_status)")
+    cursor.execute(
+        "CREATE INDEX IF NOT EXISTS idx_wazuh_config_engagement ON wazuh_config(engagement_id)"
+    )
+    cursor.execute(
+        "CREATE INDEX IF NOT EXISTS idx_detection_results_job ON detection_results(job_id)"
+    )
+    cursor.execute(
+        "CREATE INDEX IF NOT EXISTS idx_detection_results_engagement ON detection_results(engagement_id)"
+    )
+    cursor.execute(
+        "CREATE INDEX IF NOT EXISTS idx_detection_results_status ON detection_results(detection_status)"
+    )
 
     conn.commit()
 

souleyez/storage/migrations/_022_wazuh_indexer_columns.py

@@ -17,13 +17,15 @@ def upgrade(conn):
     cursor.execute("PRAGMA table_info(wazuh_config)")
     columns = [col[1] for col in cursor.fetchall()]
 
-    if 'indexer_url' not in columns:
+    if "indexer_url" not in columns:
         cursor.execute("ALTER TABLE wazuh_config ADD COLUMN indexer_url TEXT")
 
-    if 'indexer_user' not in columns:
-        cursor.execute("ALTER TABLE wazuh_config ADD COLUMN indexer_user TEXT DEFAULT 'admin'")
+    if "indexer_user" not in columns:
+        cursor.execute(
+            "ALTER TABLE wazuh_config ADD COLUMN indexer_user TEXT DEFAULT 'admin'"
+        )
 
-    if 'indexer_password' not in columns:
+    if "indexer_password" not in columns:
         cursor.execute("ALTER TABLE wazuh_config ADD COLUMN indexer_password TEXT")
 
     conn.commit()

souleyez/storage/migrations/_023_fix_detection_results_fk.py

@@ -15,7 +15,9 @@ def upgrade(conn):
     cursor = conn.cursor()
 
     # Check if the bad FK exists by examining the schema
-    cursor.execute("SELECT sql FROM sqlite_master WHERE type='table' AND name='detection_results'")
+    cursor.execute(
+        "SELECT sql FROM sqlite_master WHERE type='table' AND name='detection_results'"
+    )
     row = cursor.fetchone()
 
     if not row:
@@ -30,7 +32,8 @@ 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,
@@ -48,7 +51,8 @@ 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")
@@ -58,9 +62,15 @@ def upgrade(conn):
     cursor.execute("ALTER TABLE detection_results_new RENAME TO detection_results")
 
     # Recreate indexes
-    cursor.execute("CREATE INDEX IF NOT EXISTS idx_detection_results_job ON detection_results(job_id)")
-    cursor.execute("CREATE INDEX IF NOT EXISTS idx_detection_results_engagement ON detection_results(engagement_id)")
-    cursor.execute("CREATE INDEX IF NOT EXISTS idx_detection_results_status ON detection_results(detection_status)")
+    cursor.execute(
+        "CREATE INDEX IF NOT EXISTS idx_detection_results_job ON detection_results(job_id)"
+    )
+    cursor.execute(
+        "CREATE INDEX IF NOT EXISTS idx_detection_results_engagement ON detection_results(engagement_id)"
+    )
+    cursor.execute(
+        "CREATE INDEX IF NOT EXISTS idx_detection_results_status ON detection_results(detection_status)"
+    )
 
     conn.commit()
 

souleyez/storage/migrations/_024_wazuh_vulnerabilities.py

@@ -15,7 +15,8 @@ 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,
@@ -58,10 +59,12 @@ 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,
@@ -71,15 +74,28 @@ def upgrade(conn):
             last_sync_errors TEXT,
             FOREIGN KEY (engagement_id) REFERENCES engagements(id) ON DELETE CASCADE
         )
-    """)
+    """
+    )
 
     # Indexes for performance
-    cursor.execute("CREATE INDEX IF NOT EXISTS idx_wazuh_vulns_engagement ON wazuh_vulnerabilities(engagement_id)")
-    cursor.execute("CREATE INDEX IF NOT EXISTS idx_wazuh_vulns_host ON wazuh_vulnerabilities(host_id)")
-    cursor.execute("CREATE INDEX IF NOT EXISTS idx_wazuh_vulns_cve ON wazuh_vulnerabilities(cve_id)")
-    cursor.execute("CREATE INDEX IF NOT EXISTS idx_wazuh_vulns_severity ON wazuh_vulnerabilities(severity)")
-    cursor.execute("CREATE INDEX IF NOT EXISTS idx_wazuh_vulns_agent_ip ON wazuh_vulnerabilities(agent_ip)")
-    cursor.execute("CREATE INDEX IF NOT EXISTS idx_wazuh_vulns_status ON wazuh_vulnerabilities(status)")
+    cursor.execute(
+        "CREATE INDEX IF NOT EXISTS idx_wazuh_vulns_engagement ON wazuh_vulnerabilities(engagement_id)"
+    )
+    cursor.execute(
+        "CREATE INDEX IF NOT EXISTS idx_wazuh_vulns_host ON wazuh_vulnerabilities(host_id)"
+    )
+    cursor.execute(
+        "CREATE INDEX IF NOT EXISTS idx_wazuh_vulns_cve ON wazuh_vulnerabilities(cve_id)"
+    )
+    cursor.execute(
+        "CREATE INDEX IF NOT EXISTS idx_wazuh_vulns_severity ON wazuh_vulnerabilities(severity)"
+    )
+    cursor.execute(
+        "CREATE INDEX IF NOT EXISTS idx_wazuh_vulns_agent_ip ON wazuh_vulnerabilities(agent_ip)"
+    )
+    cursor.execute(
+        "CREATE INDEX IF NOT EXISTS idx_wazuh_vulns_status ON wazuh_vulnerabilities(status)"
+    )
 
     conn.commit()
 

souleyez/storage/migrations/_025_multi_siem_support.py

@@ -18,17 +18,15 @@ def upgrade(conn):
     columns = [col[1] for col in cursor.fetchall()]
 
     # Add siem_type column (default 'wazuh' for backwards compatibility)
-    if 'siem_type' not in columns:
+    if "siem_type" not in columns:
         cursor.execute(
             "ALTER TABLE wazuh_config ADD COLUMN siem_type TEXT DEFAULT 'wazuh'"
         )
 
     # Add config_json for storing SIEM-specific configuration
     # This allows flexible storage of different fields per SIEM type
-    if 'config_json' not in columns:
-        cursor.execute(
-            "ALTER TABLE wazuh_config ADD COLUMN config_json TEXT"
-        )
+    if "config_json" not in columns:
+        cursor.execute("ALTER TABLE wazuh_config ADD COLUMN config_json TEXT")
 
     conn.commit()