feat: add script to repair video captions from captions.json for broken videos

Author: martinkimoe

tools/add_captions_from_file.py

@@ -0,0 +1,63 @@
+"""
+This script adds captions to the database from a captions.json file
+for videos that have broken status 1.
+
+The necessity for this workaround arises because the APIs used during development
+to fetch YouTube video captions are blocked on the production server due to being flagged as
+a bot by YouTube.
+
+Use this script for repairing video entries when automatic caption fetching fails.
+
+"""
+
+from zeeguu.core.model import db
+from zeeguu.core.model.video import Video
+from zeeguu.core.model.caption import Caption
+from zeeguu.api.app import create_app
+from zeeguu.core.youtube_api.youtube_api import get_captions
+
+
+app = create_app()
+app.app_context().push()
+db_session = db.session
+
+# Get all videos with broken status 1
+videos_with_broken_status_1 = Video.query.filter_by(broken=1).all()
+print(f"Found {len(videos_with_broken_status_1)} videos with broken status 1")
+
+# For each video with broken status 1, check if it has captions in captions.json
+for video in videos_with_broken_status_1:
+    captions = get_captions(video.video_unique_key, "NOT_USED")
+    if captions:
+        # Add captions from captions.json
+        try:
+            captions_list = captions["captions"]
+            for caption in captions_list:
+                new_caption = Caption.create(
+                    session=db_session,
+                    video=video,
+                    time_start=caption["time_start"],
+                    time_end=caption["time_end"],
+                    text=caption["text"],
+                )
+                db_session.add(new_caption)
+            # Save all captions for video
+            print(
+                f"Saving {len(captions_list)} captions for video {video.video_unique_key}..."
+            )
+            db_session.commit()
+
+            # Set video broken status to 0 to indicate that the video is functional
+            print(
+                f"Setting video broken status to 0 for video {video.video_unique_key}..."
+            )
+            video.broken = 0  # SQLAlchemy detects this change on the managed object
+            db_session.commit()
+            print(f"Successfully added captions for video {video.video_unique_key}.")
+        except Exception as e:
+            print(f"Error adding captions for video {video.video_unique_key}: {e}")
+            db_session.rollback()
+            raise e
+
+    else:
+        print(f"Video {video.video_unique_key} does not have captions in captions.json")