From bcff586ee58e209f93cca2a75e2e5a65038af2eb Mon Sep 17 00:00:00 2001 From: Jonas Depoix Date: Fri, 20 Apr 2018 15:53:11 +0200 Subject: [PATCH] get_transcripts signature updated to be python2 compatible --- src/transcript_api.py | 3 ++- youtube_transcript_api.py | 10 +++++++--- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/src/transcript_api.py b/src/transcript_api.py index 2f3e098..837792b 100644 --- a/src/transcript_api.py +++ b/src/transcript_api.py @@ -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. diff --git a/youtube_transcript_api.py b/youtube_transcript_api.py index 8862935..266101f 100755 --- a/youtube_transcript_api.py +++ b/youtube_transcript_api.py @@ -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])