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

souleyez/storage/migrate_to_engagements.py

@@ -6,7 +6,7 @@ import sqlite3
 from pathlib import Path
 from .. import config
 
-DB_PATH = Path(config.get('database.path', '~/.souleyez/souleyez.db')).expanduser()
+DB_PATH = Path(config.get("database.path", "~/.souleyez/souleyez.db")).expanduser()
 
 
 def migrate():
@@ -21,7 +21,9 @@ def migrate():
     cursor = conn.cursor()
 
     # Check if workspaces table exists
-    cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='workspaces'")
+    cursor.execute(
+        "SELECT name FROM sqlite_master WHERE type='table' AND name='workspaces'"
+    )
     if not cursor.fetchone():
         print("Database already migrated or no workspaces table found.")
         conn.close()
@@ -39,7 +41,8 @@ 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,
@@ -53,19 +56,23 @@ 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,
@@ -83,21 +90,27 @@ 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")
 
         # Check if osint_data table exists and update it
-        cursor.execute("SELECT name FROM sqlite_master WHERE type='table' AND name='osint_data'")
+        cursor.execute(
+            "SELECT name FROM sqlite_master WHERE type='table' AND name='osint_data'"
+        )
         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,
@@ -106,12 +119,15 @@ 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")
 
@@ -119,11 +135,19 @@ def migrate():
         print("  - Recreating indexes...")
         cursor.execute("DROP INDEX IF EXISTS idx_hosts_workspace")
         cursor.execute("DROP INDEX IF EXISTS idx_findings_workspace")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_hosts_engagement ON hosts(engagement_id)")
+        cursor.execute(
+            "CREATE INDEX IF NOT EXISTS idx_hosts_engagement ON hosts(engagement_id)"
+        )
         cursor.execute("CREATE INDEX IF NOT EXISTS idx_hosts_ip ON hosts(ip_address)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_services_host ON services(host_id)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_findings_engagement ON findings(engagement_id)")
-        cursor.execute("CREATE INDEX IF NOT EXISTS idx_findings_severity ON findings(severity)")
+        cursor.execute(
+            "CREATE INDEX IF NOT EXISTS idx_services_host ON services(host_id)"
+        )
+        cursor.execute(
+            "CREATE INDEX IF NOT EXISTS idx_findings_engagement ON findings(engagement_id)"
+        )
+        cursor.execute(
+            "CREATE INDEX IF NOT EXISTS idx_findings_severity ON findings(severity)"
+        )
 
         # Update current_workspace file to current_engagement
         old_file = Path.home() / ".souleyez" / "current_workspace"

souleyez/storage/migrations/_001_add_credential_enhancements.py

@@ -11,31 +11,37 @@ import os
 def upgrade(conn: sqlite3.Connection):
     """Apply migration."""
     cursor = conn.cursor()
-    
+
     # Check if we should suppress output (silent mode)
-    silent = os.environ.get('SOULEYEZ_MIGRATION_SILENT', '0') == '1'
+    silent = os.environ.get("SOULEYEZ_MIGRATION_SILENT", "0") == "1"
 
     # 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:
@@ -43,16 +49,20 @@ 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/_002_add_status_tracking.py

@@ -6,22 +6,23 @@ Migration 002: Add status tracking fields to hosts and credentials tables
 
 def upgrade(conn):
     """Add notes and tracking fields."""
+
     # Helper to check if column exists
     def column_exists(table, column):
         cursor = conn.execute(f"PRAGMA table_info({table})")
         columns = [row[1] for row in cursor.fetchall()]
         return column in columns
-    
+
     # Add columns to credentials table if they don't exist
-    if not column_exists('credentials', 'notes'):
+    if not column_exists("credentials", "notes"):
         conn.execute("ALTER TABLE credentials ADD COLUMN notes TEXT")
-    if not column_exists('credentials', 'last_tested'):
+    if not column_exists("credentials", "last_tested"):
         conn.execute("ALTER TABLE credentials ADD COLUMN last_tested TIMESTAMP")
-    
+
     # Add columns to hosts table if they don't exist
-    if not column_exists('hosts', 'access_level'):
+    if not column_exists("hosts", "access_level"):
         conn.execute("ALTER TABLE hosts ADD COLUMN access_level TEXT DEFAULT 'none'")
-    if not column_exists('hosts', 'notes'):
+    if not column_exists("hosts", "notes"):
         conn.execute("ALTER TABLE hosts ADD COLUMN notes TEXT")
 
 
@@ -30,4 +31,6 @@ def downgrade(conn):
     SQLite doesn't support DROP COLUMN easily.
     Would need to recreate tables without these columns.
     """
-    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/_003_add_execution_log.py

@@ -9,12 +9,13 @@ import os
 def upgrade(conn):
     """Apply migration - add execution_log table."""
     cursor = conn.cursor()
-    
+
     # Check if we should suppress output (silent mode)
-    silent = os.environ.get('SOULEYEZ_MIGRATION_SILENT', '0') == '1'
+    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,
@@ -31,18 +32,23 @@ 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,7 +5,8 @@ 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,
@@ -22,11 +23,18 @@ 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)")
-    db.execute("CREATE INDEX IF NOT EXISTS idx_screenshots_host ON screenshots(host_id)")
-    db.execute("CREATE INDEX IF NOT EXISTS idx_screenshots_finding ON screenshots(finding_id)")
+    db.execute(
+        "CREATE INDEX IF NOT EXISTS idx_screenshots_engagement ON screenshots(engagement_id)"
+    )
+    db.execute(
+        "CREATE INDEX IF NOT EXISTS idx_screenshots_host ON screenshots(host_id)"
+    )
+    db.execute(
+        "CREATE INDEX IF NOT EXISTS idx_screenshots_finding ON screenshots(finding_id)"
+    )
 
 
 def downgrade(db):

souleyez/storage/migrations/_006_deliverables.py

@@ -5,7 +5,8 @@ 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,
@@ -23,11 +24,18 @@ 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)")
-    db.execute("CREATE INDEX IF NOT EXISTS idx_deliverables_status ON deliverables(status)")
-    db.execute("CREATE INDEX IF NOT EXISTS idx_deliverables_category ON deliverables(category)")
+    db.execute(
+        "CREATE INDEX IF NOT EXISTS idx_deliverables_engagement ON deliverables(engagement_id)"
+    )
+    db.execute(
+        "CREATE INDEX IF NOT EXISTS idx_deliverables_status ON deliverables(status)"
+    )
+    db.execute(
+        "CREATE INDEX IF NOT EXISTS idx_deliverables_category ON deliverables(category)"
+    )
 
 
 def downgrade(db):

souleyez/storage/migrations/_007_deliverable_templates.py

@@ -2,11 +2,13 @@
 Migration 007: Add deliverable templates system
 """
 
+
 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,
@@ -18,14 +20,17 @@ 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,7 +5,8 @@ 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,
@@ -24,10 +25,15 @@ 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)")
-    db.execute("CREATE INDEX IF NOT EXISTS idx_nuclei_severity ON nuclei_findings(severity)")
+    db.execute(
+        "CREATE INDEX IF NOT EXISTS idx_nuclei_engagement ON nuclei_findings(engagement_id)"
+    )
+    db.execute(
+        "CREATE INDEX IF NOT EXISTS idx_nuclei_severity ON nuclei_findings(severity)"
+    )
     db.execute("CREATE INDEX IF NOT EXISTS idx_nuclei_cve ON nuclei_findings(cve_id)")
 
 

souleyez/storage/migrations/_010_evidence_linking.py

@@ -2,10 +2,12 @@
 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,
@@ -16,18 +18,23 @@ 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/_011_timeline_tracking.py

@@ -2,56 +2,63 @@
 Migration 011: Timeline tracking for deliverables and engagements
 """
 
+
 def upgrade(conn):
     """Add timeline tracking columns."""
-    
+
     # Add timeline columns to deliverables
     try:
         conn.execute("ALTER TABLE deliverables ADD COLUMN started_at TIMESTAMP")
     except Exception:
         pass  # Column might already exist
-    
+
     try:
         conn.execute("ALTER TABLE deliverables ADD COLUMN completed_at TIMESTAMP")
     except Exception:
         pass
-    
+
     try:
-        conn.execute("ALTER TABLE deliverables ADD COLUMN estimated_hours FLOAT DEFAULT 0")
+        conn.execute(
+            "ALTER TABLE deliverables ADD COLUMN estimated_hours FLOAT DEFAULT 0"
+        )
     except Exception:
         pass
-    
+
     try:
         conn.execute("ALTER TABLE deliverables ADD COLUMN actual_hours FLOAT DEFAULT 0")
     except Exception:
         pass
-    
+
     try:
         conn.execute("ALTER TABLE deliverables ADD COLUMN blocker TEXT")
     except Exception:
         pass
-    
+
     try:
         conn.execute("ALTER TABLE deliverables ADD COLUMN assigned_to TEXT")
     except Exception:
         pass
-    
+
     # Add engagement type to engagements
     try:
-        conn.execute("ALTER TABLE engagements ADD COLUMN engagement_type TEXT DEFAULT 'network'")
+        conn.execute(
+            "ALTER TABLE engagements ADD COLUMN engagement_type TEXT DEFAULT 'network'"
+        )
     except Exception:
         pass
-    
+
     try:
-        conn.execute("ALTER TABLE engagements ADD COLUMN estimated_hours FLOAT DEFAULT 0")
+        conn.execute(
+            "ALTER TABLE engagements ADD COLUMN estimated_hours FLOAT DEFAULT 0"
+        )
     except Exception:
         pass
-    
+
     try:
         conn.execute("ALTER TABLE engagements ADD COLUMN actual_hours FLOAT DEFAULT 0")
     except Exception:
         pass
-    
+
     print("✅ Migration 011: Timeline tracking columns added")
 
 

souleyez/storage/migrations/_012_team_collaboration.py

@@ -2,22 +2,24 @@
 Migration 012: Add team collaboration features
 """
 
+
 def upgrade(conn):
     """Add team collaboration tables."""
-    
+
     # Add columns to deliverables table for team collaboration
     try:
         conn.execute("ALTER TABLE deliverables ADD COLUMN assigned_to TEXT")
     except Exception:
         pass  # Column may already exist
-    
+
     try:
         conn.execute("ALTER TABLE deliverables ADD COLUMN blocker TEXT")
     except Exception:
         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,
@@ -29,26 +31,34 @@ 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,
@@ -57,13 +67,16 @@ 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")