get_transcripts signature updated to be python2 compatible
This commit is contained in:
parent
0142f9a61f
commit
bcff586ee5
|
@ -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.
|
||||
|
||||
|
|
|
@ -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])
|
||||
|
|
Loading…
Reference in New Issue