diff --git a/youtube_transcript_api.py b/youtube_transcript_api.py index b4e206e..8862935 100755 --- a/youtube_transcript_api.py +++ b/youtube_transcript_api.py @@ -4,7 +4,13 @@ import sys import json +from pprint import pprint + from src.transcript_api import YouTubeTranscriptApi + if __name__ == '__main__': - print(json.dumps(YouTubeTranscriptApi.get(*sys.argv[1:]))) + if sys.argv[1] == '--json': + print(json.dumps(YouTubeTranscriptApi.get_transcripts(*sys.argv[2:], continue_after_error=True)[0])) + else: + pprint(YouTubeTranscriptApi.get_transcripts(*sys.argv[1:], continue_after_error=True)[0])