From 0142f9a61f596fc936bce5cff363a6509ab915ed Mon Sep 17 00:00:00 2001 From: Jonas Depoix Date: Fri, 20 Apr 2018 14:20:24 +0200 Subject: [PATCH] "cli" improved --- youtube_transcript_api.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) 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])