get_transcripts signature updated to be python2 compatible

This commit is contained in:
Jonas Depoix 2018-04-20 15:53:11 +02:00
parent 0142f9a61f
commit bcff586ee5
2 changed files with 9 additions and 4 deletions

View File

@ -6,6 +6,7 @@ import logging
import requests
logger = logging.getLogger(__name__)
@ -29,7 +30,7 @@ class YouTubeTranscriptApi():
@staticmethod
def get_transcripts(*video_ids, continue_after_error=False):
def get_transcripts(video_ids, continue_after_error=False):
"""
Retrieves the transcripts for a list of videos.

View File

@ -1,4 +1,4 @@
#!./.venv/bin/python
#!/usr/bin/python
import sys
@ -6,11 +6,15 @@ import json
from pprint import pprint
import logging
from src.transcript_api import YouTubeTranscriptApi
if __name__ == '__main__':
logging.basicConfig()
if sys.argv[1] == '--json':
print(json.dumps(YouTubeTranscriptApi.get_transcripts(*sys.argv[2:], continue_after_error=True)[0]))
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])
pprint(YouTubeTranscriptApi.get_transcripts(sys.argv[1:], continue_after_error=True)[0])