db updates

2025-09-09 08:50:21 -04:00
parent 9fcf5d8303
commit dd215d2f38
9 changed files with 908 additions and 74 deletions
@@ -0,0 +1,99 @@
+#!/usr/bin/env python3
+"""
+Test script for bulk_update_movies.py
+This tests the bulk update functionality without modifying data
+"""
+
+import os
+import sys
+from pathlib import Path
+
+def test_bulk_update_dry_run():
+    """Test bulk update script in dry-run mode"""
+    print("🧪 Testing Bulk Update Script")
+    print("=" * 40)
+    
+    # Check if we can import the required modules
+    try:
+        from clients.radarr_db_client import RadarrDbClient
+        from core.database import NFOGuardDatabase
+        print("✅ Successfully imported required modules")
+    except Exception as e:
+        print(f"❌ Import error: {e}")
+        return False
+    
+    # Test database connections
+    print("\n🔌 Testing Database Connections...")
+    
+    # Test Radarr database
+    try:
+        radarr_db = RadarrDbClient.from_env()
+        if not radarr_db:
+            print("❌ Radarr database connection failed - check environment variables")
+            return False
+        print("✅ Radarr database connection successful")
+    except Exception as e:
+        print(f"❌ Radarr database error: {e}")
+        return False
+    
+    # Test NFOGuard database
+    try:
+        db_path = Path(os.environ.get("DB_PATH", "/app/data/media_dates.db"))
+        nfo_db = NFOGuardDatabase(db_path)
+        print(f"✅ NFOGuard database connection successful: {db_path}")
+    except Exception as e:
+        print(f"❌ NFOGuard database error: {e}")
+        return False
+    
+    # Test query execution (dry run)
+    print("\n📊 Testing Query Execution...")
+    
+    try:
+        query = """
+        SELECT 
+            mm."ImdbId" as imdb_id,
+            m."Id" as movie_id,
+            mm."Title" as title,
+            mm."Year" as year,
+            m."Path" as path
+        FROM "Movies" m
+        JOIN "MovieMetadata" mm ON m."MovieMetadataId" = mm."Id"
+        WHERE mm."ImdbId" IS NOT NULL AND mm."ImdbId" != ''
+        ORDER BY mm."Title"
+        LIMIT 5
+        """
+        
+        with radarr_db._get_connection() as conn:
+            if radarr_db.db_type == "postgresql":
+                import psycopg2.extras
+                cursor = conn.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
+            else:
+                cursor = conn.cursor()
+            
+            cursor.execute(query)
+            movies = cursor.fetchall()
+            
+        print(f"✅ Successfully queried Radarr database: {len(movies)} sample movies found")
+        
+        # Show sample data
+        for i, movie in enumerate(movies):
+            if radarr_db.db_type == "postgresql":
+                imdb_id, title, year = movie['imdb_id'], movie['title'], movie['year']
+            else:
+                imdb_id, title, year = movie[0], movie[2], movie[3]
+            print(f"  Sample {i+1}: {title} ({year}) - {imdb_id}")
+            
+    except Exception as e:
+        print(f"❌ Query execution failed: {e}")
+        return False
+    
+    print("\n✅ All tests passed! Bulk update script should work correctly.")
+    return True
+
+if __name__ == "__main__":
+    success = test_bulk_update_dry_run()
+    if not success:
+        print("\n❌ Bulk update test failed!")
+        sys.exit(1)
+    else:
+        print("\n🎉 Bulk update test successful!")