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 import requests
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -29,7 +30,7 @@ class YouTubeTranscriptApi():
@staticmethod @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. Retrieves the transcripts for a list of videos.

View File

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