KeyArgo
/
youtube-transcript-api
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update CLI to make use of new DELIMITER and combine method

This commit is contained in:
Chris Howell 2020-07-09 00:06:36 -07:00
parent 5861bdb104
commit 74d36a821e
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
youtube_transcript_api/_cli.py
View File

@ -25,21 +25,24 @@ class YouTubeTranscriptCli():
transcripts = [] transcripts = []
exceptions = [] exceptions = []
Formatter = formats.get_formatter(parsed_args.format)
for video_id in parsed_args.video_ids: for video_id in parsed_args.video_ids:
try: try:
transcripts.append(self._fetch_transcript(parsed_args, proxies, cookies, video_id)) transcript = self._fetch_transcript(
parsed_args, proxies, cookies, video_id)
transcripts.append(Formatter.format(transcript))
except Exception as exception: except Exception as exception:
exceptions.append(exception) exceptions.append(exception)
Formatter = formats.get_formatter(parsed_args.format)
results = Formatter.format(transcripts)
return ''.join( return ''.join(
[str(exception) for exception in exceptions] + results [str(exception) for exception in exceptions]
+ ([Formatter.combine(transcripts)] if transcripts else [])
) )
def _fetch_transcript(self, parsed_args, proxies, cookies, video_id): def _fetch_transcript(self, parsed_args, proxies, cookies, video_id):
transcript_list = YouTubeTranscriptApi.list_transcripts(video_id, proxies=proxies, cookies=cookies) transcript_list = YouTubeTranscriptApi.list_transcripts(
video_id, proxies=proxies, cookies=cookies)
if parsed_args.list_transcripts: if parsed_args.list_transcripts:
return str(transcript_list) return str(transcript_list)