feat: add migration and verification scripts to initialize missing user analytics documents

migrate_analytics.py

@@ -0,0 +1,119 @@
+"""
+Migration Script: Initialize analytics collection for all existing users.
+Creates a default analytics document (matching AnalyticsModel) for every user
+that doesn't already have one.
+"""
+
+import firebase_admin
+from firebase_admin import credentials, firestore
+from google.cloud.firestore_v1 import SERVER_TIMESTAMP
+
+def main():
+    # Initialize Firebase Admin SDK
+    cred = credentials.Certificate("firebase-service-account.json")
+    firebase_admin.initialize_app(cred)
+    db = firestore.client()
+
+    print("=" * 60)
+    print("  Analytics Collection Migration")
+    print("=" * 60)
+
+    # Step 1: Read all users
+    print("\n[1/3] Reading users collection...")
+    users_ref = db.collection("users")
+    user_docs = users_ref.stream()
+
+    user_ids = []
+    for doc in user_docs:
+        user_ids.append(doc.id)
+        print(f"  Found user: {doc.id}")
+
+    print(f"\n  Total users found: {len(user_ids)}")
+
+    if not user_ids:
+        print("\n  No users found in the database. Nothing to migrate.")
+        return
+
+    # Step 2: Check which users already have analytics documents
+    print("\n[2/3] Checking existing analytics documents...")
+    analytics_ref = db.collection("analytics")
+    existing_analytics = set()
+    for doc in analytics_ref.stream():
+        existing_analytics.add(doc.id)
+
+    users_needing_analytics = [uid for uid in user_ids if uid not in existing_analytics]
+    already_have = [uid for uid in user_ids if uid in existing_analytics]
+
+    if already_have:
+        print(f"  Users already with analytics: {len(already_have)}")
+        for uid in already_have:
+            print(f"    - {uid} (skipping)")
+
+    print(f"  Users needing analytics: {len(users_needing_analytics)}")
+
+    if not users_needing_analytics:
+        print("\n  All users already have analytics documents. Nothing to do.")
+        return
+
+    # Step 3: Create default analytics documents using batched writes
+    print("\n[3/3] Creating default analytics documents...")
+    
+    # Default schema matching AnalyticsModel / initializeUserAnalytics
+    default_analytics = {
+        "totalVideos": 0,
+        "totalMinutes": 0,
+        "totalSavedHours": 0.0,
+        "favoriteCategory": "None",
+        "currentStreak": 0,
+        "thisWeekVideos": 0,
+        "thisMonthSavedHours": 0.0,
+        "categoryCount": {},
+        "lastUpdated": SERVER_TIMESTAMP,
+    }
+
+    # Firestore batch limit is 500 operations per batch
+    batch = db.batch()
+    batch_count = 0
+    total_created = 0
+
+    for uid in users_needing_analytics:
+        doc_ref = analytics_ref.document(uid)
+        batch.set(doc_ref, default_analytics)
+        batch_count += 1
+        total_created += 1
+        print(f"  Queued: analytics/{uid}")
+
+        # Commit every 500 operations (Firestore batch limit)
+        if batch_count >= 500:
+            batch.commit()
+            print(f"  Committed batch ({batch_count} documents)")
+            batch = db.batch()
+            batch_count = 0
+
+    # Commit remaining
+    if batch_count > 0:
+        batch.commit()
+        print(f"  Committed final batch ({batch_count} documents)")
+
+    # Step 4: Verify
+    print("\n" + "=" * 60)
+    print("  Verification")
+    print("=" * 60)
+    
+    verify_count = 0
+    for doc in db.collection("analytics").stream():
+        verify_count += 1
+        data = doc.to_dict()
+        print(f"  ✓ analytics/{doc.id}")
+        print(f"    totalVideos={data.get('totalVideos')}, "
+              f"totalMinutes={data.get('totalMinutes')}, "
+              f"favoriteCategory={data.get('favoriteCategory')}, "
+              f"lastUpdated={data.get('lastUpdated')}")
+
+    print(f"\n  Analytics collection now has {verify_count} document(s)")
+    print(f"  New documents created: {total_created}")
+    print("\n  Migration complete!")
+
+
+if __name__ == "__main__":
+    main()

verify_analytics.py

@@ -0,0 +1,16 @@
+"""Quick verification of analytics collection."""
+import firebase_admin
+from firebase_admin import credentials, firestore
+
+cred = credentials.Certificate("firebase-service-account.json")
+firebase_admin.initialize_app(cred)
+db = firestore.client()
+
+docs = list(db.collection("analytics").stream())
+print(f"Analytics collection: {len(docs)} documents")
+for d in docs:
+    data = d.to_dict()
+    print(f"  OK analytics/{d.id}")
+    print(f"     totalVideos={data.get('totalVideos')}, totalMinutes={data.get('totalMinutes')}, "
+          f"favoriteCategory={data.get('favoriteCategory')}, lastUpdated={data.get('lastUpdated')}")
+print("\nDone.")