Phase 2 Complete: SQL migration with SQLModel and Alembic

scripts/audit_database.py

@@ -0,0 +1,59 @@
+import os
+import sys
+from sqlmodel import Session, select, create_engine, inspect
+
+# Add root directory to sys.path
+sys.path.append(os.getcwd())
+
+from app.database import engine
+from app.models.auth import UserTable
+from app.models.watchlist import WatchlistItemTable, WatchlistSettingsTable
+from app.models.favorites import FavoriteTable
+from app.models.sonarr import SonarrMappingTable, SonarrConfigTable
+
+def audit_db():
+    print("--- Ohm Stream Downloader: SQL Database Audit ---")
+    
+    if not os.path.exists("ohm_streaming.db"):
+        print("ERROR: ohm_streaming.db not found!")
+        return
+
+    inspector = inspect(engine)
+    tables = inspector.get_table_names()
+    print(f"Tables found: {', '.join(tables)}")
+    
+    expected_tables = ["users", "watchlist_items", "watchlist_settings", "favorites", "sonarr_mappings", "sonarr_config", "alembic_version"]
+    missing = [t for t in expected_tables if t not in tables]
+    if missing:
+        print(f"WARNING: Missing tables: {', '.join(missing)}")
+    else:
+        print("SUCCESS: All core tables are present.")
+
+    with Session(engine) as session:
+        # Check users
+        users_count = len(session.exec(select(UserTable)).all())
+        print(f"Users: {users_count}")
+        
+        # Check watchlist
+        watchlist_count = len(session.exec(select(WatchlistItemTable)).all())
+        print(f"Watchlist Items: {watchlist_count}")
+        
+        # Check settings
+        settings_count = len(session.exec(select(WatchlistSettingsTable)).all())
+        print(f"Watchlist Settings entries: {settings_count}")
+        
+        # Check favorites
+        fav_count = len(session.exec(select(FavoriteTable)).all())
+        print(f"Favorites: {fav_count}")
+        
+        # Check Sonarr
+        sonarr_map_count = len(session.exec(select(SonarrMappingTable)).all())
+        print(f"Sonarr Mappings: {sonarr_map_count}")
+        
+        # Sample data check
+        if fav_count > 0:
+            sample_fav = session.exec(select(FavoriteTable).limit(1)).first()
+            print(f"Sample Favorite: {sample_fav.title} ({sample_fav.provider})")
+
+if __name__ == "__main__":
+    audit_db()