Migration script

Author: MrDesjardins

CLAUDE.md

@@ -286,7 +286,9 @@ CREATE TABLE queue (
     channel TEXT,                  -- Channel/uploader name
     thumbnail_url TEXT,            -- Video thumbnail URL (best quality)
     position INTEGER NOT NULL,     -- Order in queue
-    created_at TEXT NOT NULL
+    created_at TEXT NOT NULL,
+    type TEXT DEFAULT 'youtube',   -- Type of item (youtube or summary)
+    week_year TEXT                 -- Week identifier for summary items
 );
 ```
 
@@ -583,7 +585,7 @@ This script will:
 2. Pull latest changes from main branch
 3. Update Python dependencies via `uv sync`
 4. Check and install missing system dependencies
-5. Run database migrations (migrate_database.py)
+5. Run database migrations (migrate_database.py, migrate_add_metadata.py, migrate_add_queue_columns.py)
 6. Update database schema (`services.database.init_database`)
 7. Restart the systemd service if it was running
 
@@ -607,13 +609,20 @@ The project includes multiple migration scripts for schema updates:
 - Creates backup before migration
 - Required for MediaSession API integration (car displays, lock screens)
 
+**migrate_add_queue_columns.py** - Queue enhancement migration:
+- Adds `type` column to queue table (default: 'youtube')
+- Adds `week_year` column to queue table (for weekly summary items)
+- Creates backup before migration
+- Required for queuing weekly summaries alongside YouTube videos
+
 **Manual migration**:
 ```sh
 uv run python migrate_database.py
 uv run python migrate_add_metadata.py
+uv run python migrate_add_queue_columns.py
 ```
 
-Both migrations run automatically during `./update.sh`.
+All migrations run automatically during `./update.sh`.
 
 ### Trilium Notes Title Migration
 

migrate_add_queue_columns.py

@@ -0,0 +1,89 @@
+"""
+Migration script to add type and week_year columns to queue table.
+
+This migration adds support for queuing weekly summaries alongside YouTube videos.
+"""
+
+import os
+import sqlite3
+import sys
+from pathlib import Path
+
+
+def migrate_queue_table() -> None:
+    """Add type and week_year columns to queue table if they don't exist."""
+    db_path_str = os.getenv("DATABASE_PATH", "./audio_history.db")
+    db_path = Path(db_path_str).expanduser().resolve()
+
+    print(f"Migrating database: {db_path}")
+
+    # Check if database exists
+    if not db_path.exists():
+        print(f"Error: Database not found at {db_path}")
+        sys.exit(1)
+
+    # Create backup
+    backup_path = db_path.with_suffix(f"{db_path.suffix}.backup-queue-columns")
+    print(f"Creating backup: {backup_path}")
+    import shutil
+
+    shutil.copy2(db_path, backup_path)
+
+    # Connect to database
+    conn = sqlite3.connect(db_path)
+    conn.row_factory = sqlite3.Row
+    cursor = conn.cursor()
+
+    try:
+        # Check if columns already exist
+        cursor.execute("PRAGMA table_info(queue)")
+        columns = [row[1] for row in cursor.fetchall()]
+
+        changes_made = False
+
+        # Add type column if it doesn't exist
+        if "type" not in columns:
+            print("Adding 'type' column to queue table...")
+            cursor.execute("""
+                ALTER TABLE queue ADD COLUMN type TEXT DEFAULT 'youtube'
+            """)
+            # Update existing rows to have 'youtube' type
+            cursor.execute("""
+                UPDATE queue SET type = 'youtube' WHERE type IS NULL
+            """)
+            changes_made = True
+            print("✓ Added 'type' column")
+        else:
+            print("✓ 'type' column already exists")
+
+        # Add week_year column if it doesn't exist
+        if "week_year" not in columns:
+            print("Adding 'week_year' column to queue table...")
+            cursor.execute("""
+                ALTER TABLE queue ADD COLUMN week_year TEXT
+            """)
+            changes_made = True
+            print("✓ Added 'week_year' column")
+        else:
+            print("✓ 'week_year' column already exists")
+
+        if changes_made:
+            conn.commit()
+            print("\n✅ Migration completed successfully!")
+            print(f"Backup saved at: {backup_path}")
+        else:
+            print("\n✅ No migration needed - columns already exist")
+            # Remove unnecessary backup
+            backup_path.unlink()
+
+    except Exception as e:
+        print(f"\n❌ Migration failed: {e}")
+        conn.rollback()
+        print(f"Database has been rolled back. Backup available at: {backup_path}")
+        sys.exit(1)
+    finally:
+        conn.close()
+
+
+if __name__ == "__main__":
+    migrate_queue_table()

update.sh

@@ -60,6 +60,7 @@ echo "----------------------------------------"
 # Run migrations (handles schema changes for existing databases)
 uv run python migrate_database.py
 uv run python migrate_add_metadata.py
+uv run python migrate_add_queue_columns.py
 
 # Then initialize/update schema (creates tables if they don't exist)
 uv run python -c "from services.database import init_database; init_database(); print('Database schema updated successfully')"