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

souleyez/storage/migrate_to_engagements.py

@@ -2,8 +2,10 @@
 """
 Migration script to rename workspaces to engagements
 """
+
 import sqlite3
 from pathlib import Path
+
 from .. import config
 
 DB_PATH = Path(config.get("database.path", "~/.souleyez/souleyez.db")).expanduser()
@@ -41,8 +43,7 @@ def migrate():
 
         # Rename workspace_id to engagement_id in hosts table
         print("  - Updating hosts table...")
-        cursor.execute(
-            """
+        cursor.execute("""
             CREATE TABLE hosts_new (
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
                 engagement_id INTEGER NOT NULL,
@@ -56,23 +57,19 @@ def migrate():
                 created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
                 updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
             )
-        """
-        )
-        cursor.execute(
-            """
+        """)
+        cursor.execute("""
             INSERT INTO hosts_new
             SELECT id, workspace_id, ip_address, hostname, os_name, os_accuracy,
                    mac_address, status, tags, created_at, updated_at
             FROM hosts
-        """
-        )
+        """)
         cursor.execute("DROP TABLE hosts")
         cursor.execute("ALTER TABLE hosts_new RENAME TO hosts")
 
         # Rename workspace_id to engagement_id in findings table
         print("  - Updating findings table...")
-        cursor.execute(
-            """
+        cursor.execute("""
             CREATE TABLE findings_new (
                 id INTEGER PRIMARY KEY AUTOINCREMENT,
                 engagement_id INTEGER NOT NULL,
@@ -90,16 +87,13 @@ def migrate():
                 scan_id INTEGER,
                 created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
             )
-        """
-        )
-        cursor.execute(
-            """
+        """)
+        cursor.execute("""
             INSERT INTO findings_new
             SELECT id, workspace_id, host_id, service_id, finding_type, severity,
                    title, description, evidence, refs, port, path, tool, scan_id, created_at
             FROM findings
-        """
-        )
+        """)
         cursor.execute("DROP TABLE findings")
         cursor.execute("ALTER TABLE findings_new RENAME TO findings")
 
@@ -109,8 +103,7 @@ def migrate():
         )
         if cursor.fetchone():
             print("  - Updating osint_data table...")
-            cursor.execute(
-                """
+            cursor.execute("""
                 CREATE TABLE osint_data_new (
                     id INTEGER PRIMARY KEY AUTOINCREMENT,
                     engagement_id INTEGER NOT NULL,
@@ -119,15 +112,12 @@ def migrate():
                     source TEXT,
                     created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
                 )
-            """
-            )
-            cursor.execute(
-                """
+            """)
+            cursor.execute("""
                 INSERT INTO osint_data_new
                 SELECT id, workspace_id, data_type, value, source, created_at
                 FROM osint_data
-            """
-            )
+            """)
             cursor.execute("DROP TABLE osint_data")
             cursor.execute("ALTER TABLE osint_data_new RENAME TO osint_data")
 

souleyez/storage/migrations/_001_add_credential_enhancements.py

@@ -4,8 +4,9 @@ Migration 001: Add credential enhancements
 - Adds indices for better performance
 - Adds created_at/updated_at tracking
 """
-import sqlite3
+
 import os
+import sqlite3
 
 
 def upgrade(conn: sqlite3.Connection):
@@ -18,30 +19,24 @@ def upgrade(conn: sqlite3.Connection):
     # Add indices for credentials table
     if not silent:
         print("  → Creating index on credentials(engagement_id)")
-    cursor.execute(
-        """
+    cursor.execute("""
         CREATE INDEX IF NOT EXISTS idx_credentials_engagement
         ON credentials(engagement_id)
-    """
-    )
+    """)
 
     if not silent:
         print("  → Creating index on credentials(host_id)")
-    cursor.execute(
-        """
+    cursor.execute("""
         CREATE INDEX IF NOT EXISTS idx_credentials_host
         ON credentials(host_id)
-    """
-    )
+    """)
 
     if not silent:
         print("  → Creating index on credentials(status)")
-    cursor.execute(
-        """
+    cursor.execute("""
         CREATE INDEX IF NOT EXISTS idx_credentials_status
         ON credentials(status)
-    """
-    )
+    """)
 
     # Add updated_at column if it doesn't exist
     try:
@@ -49,20 +44,16 @@ def upgrade(conn: sqlite3.Connection):
     except sqlite3.OperationalError:
         if not silent:
             print("  → Adding updated_at column to credentials table")
-        cursor.execute(
-            """
+        cursor.execute("""
             ALTER TABLE credentials
             ADD COLUMN updated_at TIMESTAMP
-        """
-        )
+        """)
         # Set default value for existing rows
-        cursor.execute(
-            """
+        cursor.execute("""
             UPDATE credentials
             SET updated_at = CURRENT_TIMESTAMP
             WHERE updated_at IS NULL
-        """
-        )
+        """)
 
     conn.commit()
     if not silent:

souleyez/storage/migrations/_003_add_execution_log.py

@@ -2,8 +2,9 @@
 """
 Migration 003: Add execution_log table for tracking AI-driven executions
 """
-import sqlite3
+
 import os
+import sqlite3
 
 
 def upgrade(conn):
@@ -14,8 +15,7 @@ def upgrade(conn):
     silent = os.environ.get("SOULEYEZ_MIGRATION_SILENT", "0") == "1"
 
     # Create execution_log table
-    cursor.execute(
-        """
+    cursor.execute("""
         CREATE TABLE IF NOT EXISTS execution_log (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL,
@@ -32,23 +32,18 @@ def upgrade(conn):
             feedback_applied TEXT,
             FOREIGN KEY (engagement_id) REFERENCES engagements(id)
         )
-    """
-    )
+    """)
 
     # Create index for faster queries
-    cursor.execute(
-        """
+    cursor.execute("""
         CREATE INDEX IF NOT EXISTS idx_execution_engagement 
         ON execution_log(engagement_id)
-    """
-    )
+    """)
 
-    cursor.execute(
-        """
+    cursor.execute("""
         CREATE INDEX IF NOT EXISTS idx_execution_timestamp 
         ON execution_log(executed_at DESC)
-    """
-    )
+    """)
 
     conn.commit()
     if not silent:

souleyez/storage/migrations/_005_screenshots.py

@@ -5,8 +5,7 @@ Migration 005: Add screenshots table for visual evidence management.
 
 def upgrade(db):
     """Add screenshots table."""
-    db.execute(
-        """
+    db.execute("""
         CREATE TABLE IF NOT EXISTS screenshots (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL,
@@ -23,8 +22,7 @@ def upgrade(db):
             FOREIGN KEY (host_id) REFERENCES hosts(id),
             FOREIGN KEY (finding_id) REFERENCES findings(id)
         )
-    """
-    )
+    """)
 
     db.execute(
         "CREATE INDEX IF NOT EXISTS idx_screenshots_engagement ON screenshots(engagement_id)"

souleyez/storage/migrations/_006_deliverables.py

@@ -5,8 +5,7 @@ Migration 006: Add deliverables tracking table.
 
 def upgrade(db):
     """Add deliverables tracking table."""
-    db.execute(
-        """
+    db.execute("""
         CREATE TABLE IF NOT EXISTS deliverables (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL,
@@ -24,8 +23,7 @@ def upgrade(db):
             created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
             FOREIGN KEY (engagement_id) REFERENCES engagements(id)
         )
-    """
-    )
+    """)
 
     db.execute(
         "CREATE INDEX IF NOT EXISTS idx_deliverables_engagement ON deliverables(engagement_id)"

souleyez/storage/migrations/_007_deliverable_templates.py

@@ -7,8 +7,7 @@ def upgrade(conn):
     """Add deliverable templates table."""
 
     # Templates table
-    conn.execute(
-        """
+    conn.execute("""
         CREATE TABLE IF NOT EXISTS deliverable_templates (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             name TEXT NOT NULL,
@@ -20,16 +19,13 @@ def upgrade(conn):
             created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
             is_builtin INTEGER DEFAULT 0
         )
-    """
-    )
+    """)
 
     # Index for faster lookups
-    conn.execute(
-        """
+    conn.execute("""
         CREATE INDEX IF NOT EXISTS idx_templates_framework
         ON deliverable_templates(framework, engagement_type)
-    """
-    )
+    """)
 
     print("✅ Migration 007: Deliverable templates table created")
 

souleyez/storage/migrations/_008_add_nuclei_table.py

@@ -5,8 +5,7 @@ Migration 008: Add nuclei findings table for vulnerability scan results.
 
 def upgrade(db):
     """Add nuclei_findings table."""
-    db.execute(
-        """
+    db.execute("""
         CREATE TABLE IF NOT EXISTS nuclei_findings (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             engagement_id INTEGER NOT NULL,
@@ -25,8 +24,7 @@ def upgrade(db):
             found_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
             FOREIGN KEY(engagement_id) REFERENCES engagements(id) ON DELETE CASCADE
         )
-    """
-    )
+    """)
 
     db.execute(
         "CREATE INDEX IF NOT EXISTS idx_nuclei_engagement ON nuclei_findings(engagement_id)"

souleyez/storage/migrations/_010_evidence_linking.py

@@ -6,8 +6,7 @@ Migration 010: Evidence linking system for deliverables
 def upgrade(conn):
     """Add evidence linking table."""
 
-    conn.execute(
-        """
+    conn.execute("""
         CREATE TABLE IF NOT EXISTS deliverable_evidence (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             deliverable_id INTEGER NOT NULL,
@@ -18,22 +17,17 @@ def upgrade(conn):
             notes TEXT,
             FOREIGN KEY (deliverable_id) REFERENCES deliverables(id) ON DELETE CASCADE
         )
-    """
-    )
+    """)
 
-    conn.execute(
-        """
+    conn.execute("""
         CREATE INDEX IF NOT EXISTS idx_deliverable_evidence
         ON deliverable_evidence(deliverable_id, evidence_type)
-    """
-    )
+    """)
 
-    conn.execute(
-        """
+    conn.execute("""
         CREATE INDEX IF NOT EXISTS idx_evidence_lookup
         ON deliverable_evidence(evidence_type, evidence_id)
-    """
-    )
+    """)
 
     print("✅ Migration 010: Evidence linking system created")
 

souleyez/storage/migrations/_012_team_collaboration.py

@@ -18,8 +18,7 @@ def upgrade(conn):
         pass  # Column may already exist
 
     # Activity log for tracking who did what
-    conn.execute(
-        """
+    conn.execute("""
         CREATE TABLE IF NOT EXISTS deliverable_activity (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             deliverable_id INTEGER NOT NULL,
@@ -31,34 +30,26 @@ def upgrade(conn):
             FOREIGN KEY (deliverable_id) REFERENCES deliverables(id) ON DELETE CASCADE,
             FOREIGN KEY (engagement_id) REFERENCES engagements(id) ON DELETE CASCADE
         )
-    """
-    )
+    """)
 
     # Index for faster activity queries
-    conn.execute(
-        """
+    conn.execute("""
         CREATE INDEX IF NOT EXISTS idx_activity_deliverable
         ON deliverable_activity(deliverable_id, created_at DESC)
-    """
-    )
+    """)
 
-    conn.execute(
-        """
+    conn.execute("""
         CREATE INDEX IF NOT EXISTS idx_activity_engagement
         ON deliverable_activity(engagement_id, created_at DESC)
-    """
-    )
+    """)
 
-    conn.execute(
-        """
+    conn.execute("""
         CREATE INDEX IF NOT EXISTS idx_activity_user
         ON deliverable_activity(user, created_at DESC)
-    """
-    )
+    """)
 
     # Comments on deliverables
-    conn.execute(
-        """
+    conn.execute("""
         CREATE TABLE IF NOT EXISTS deliverable_comments (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             deliverable_id INTEGER NOT NULL,
@@ -67,15 +58,12 @@ def upgrade(conn):
             created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
             FOREIGN KEY (deliverable_id) REFERENCES deliverables(id) ON DELETE CASCADE
         )
-    """
-    )
+    """)
 
-    conn.execute(
-        """
+    conn.execute("""
         CREATE INDEX IF NOT EXISTS idx_comments_deliverable
         ON deliverable_comments(deliverable_id, created_at DESC)
-    """
-    )
+    """)
 
     print("✅ Migration 012: Team collaboration tables created")
 

souleyez/storage/migrations/_013_add_host_tags.py

@@ -13,12 +13,10 @@ def upgrade(db):
     columns = [row[1] for row in cursor.fetchall()]
 
     if "tags" not in columns:
-        db.execute(
-            """
+        db.execute("""
             ALTER TABLE hosts
             ADD COLUMN tags TEXT DEFAULT NULL
-        """
-        )
+        """)
     # If column already exists, migration is idempotent - do nothing
 
 

souleyez/storage/migrations/_014_exploit_attempts.py

@@ -9,8 +9,7 @@ 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,
@@ -28,37 +27,28 @@ 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):

souleyez/storage/migrations/_015_add_mac_os_fields.py

@@ -13,20 +13,16 @@ def upgrade(db):
     columns = [row[1] for row in cursor.fetchall()]
 
     if "mac_address" not in columns:
-        db.execute(
-            """
+        db.execute("""
             ALTER TABLE hosts
             ADD COLUMN mac_address TEXT DEFAULT NULL
-        """
-        )
+        """)
 
     if "os_accuracy" not in columns:
-        db.execute(
-            """
+        db.execute("""
             ALTER TABLE hosts
             ADD COLUMN os_accuracy INTEGER DEFAULT NULL
-        """
-        )
+        """)
 
 
 def downgrade(db):

souleyez/storage/migrations/_016_add_domain_field.py

@@ -14,12 +14,10 @@ def upgrade(db):
     columns = [row[1] for row in cursor.fetchall()]
 
     if "domain" not in columns:
-        db.execute(
-            """
+        db.execute("""
             ALTER TABLE hosts
             ADD COLUMN domain TEXT DEFAULT NULL
-        """
-        )
+        """)
 
 
 def downgrade(db):

souleyez/storage/migrations/_017_msf_sessions.py

@@ -13,8 +13,7 @@ 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,
@@ -40,30 +39,23 @@ 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

@@ -18,19 +18,15 @@ def upgrade(conn):
 
     # Add target column only if it doesn't exist
     if "target" not in columns:
-        cursor.execute(
-            """
+        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

@@ -18,19 +18,15 @@ def upgrade(conn):
 
     # Add engagement_type column only if it doesn't exist
     if "engagement_type" not in columns:
-        cursor.execute(
-            """
+        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()