"cli" improved

This commit is contained in:
Jonas Depoix 2018-04-20 14:20:24 +02:00
parent 91fe71c86c
commit 0142f9a61f
1 changed files with 7 additions and 1 deletions

View File

@ -4,7 +4,13 @@ import sys
import json import json
from pprint import pprint
from src.transcript_api import YouTubeTranscriptApi from src.transcript_api import YouTubeTranscriptApi
if __name__ == '__main__': 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])