From 4c56fd786607400b52ffdfb51f23c9ed768f504d Mon Sep 17 00:00:00 2001
From: ben0815 <rubenmichler@gmx.de>
Date: Mon, 4 Mar 2024 23:46:02 +0100
Subject: [PATCH] add language option to yt.py

---
 helpers/yt.py | 28 ++++++++++++++++++++++++++--
 1 file changed, 26 insertions(+), 2 deletions(-)

diff --git a/helpers/yt.py b/helpers/yt.py
index a473d96..c984d00 100644
--- a/helpers/yt.py
+++ b/helpers/yt.py
@@ -1,3 +1,5 @@
+#!/usr/bin/env python3
+
 import re
 from googleapiclient.discovery import build
 from googleapiclient.errors import HttpError
@@ -7,7 +9,7 @@ import os
 import json
 import isodate
 import argparse
-
+import sys
 
 def get_video_id(url):
     # Extract video ID from URL
@@ -47,9 +49,29 @@ def main_function(url, options):
         duration_seconds = isodate.parse_duration(duration_iso).total_seconds()
         duration_minutes = round(duration_seconds / 60)
 
+        # Get video transcript language
+        try:
+            transcript_available = False
+            transcript_options = ''
+            transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
+            for transcript in transcript_list:
+                if options.language == transcript.language_code:
+                    transcript_available = True
+                else:
+                    transcript_options += transcript.language + ' (' + \
+                                          transcript.language_code + '); '
+
+            if not transcript_available:
+                # exit with existing languages, cause get_transcript will fail
+                sys.exit('"' + options.language + '" not available. ' + \
+                         'Following languages exists: ' + transcript_options)
+
+        except Exception as e:
+            print(e)
+
         # Get video transcript
         try:
-            transcript_list = YouTubeTranscriptApi.get_transcript(video_id)
+            transcript_list = YouTubeTranscriptApi.get_transcript(video_id, languages=[options.language])
             transcript_text = ' '.join([item['text']
                                        for item in transcript_list])
             transcript_text = transcript_text.replace('\n', ' ')
@@ -77,6 +99,8 @@ def main():
     parser = argparse.ArgumentParser(
         description='vm (video meta) extracts metadata about a video, such as the transcript and the video\'s duration. By Daniel Miessler.')
     parser.add_argument('url', nargs='?', help='YouTube video URL')
+    parser.add_argument('-l', '--language',
+                        help='Set transcript language (default en)', default='en')
     parser.add_argument('--duration', action='store_true',
                         help='Output only the duration')
     parser.add_argument('--transcript', action='store_true',