From 2a2af44eacf0e962853dba0b6ecae19fe18f9ea3 Mon Sep 17 00:00:00 2001 From: A Farzat Date: Thu, 9 Oct 2025 18:10:00 +0300 Subject: Add the ability to use YouTube API This is to circumvent YouTube blocking. --- tests/obtain_vid_info.py | 6 +++--- tests/utils/vid_url_to_html.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) (limited to 'tests') diff --git a/tests/obtain_vid_info.py b/tests/obtain_vid_info.py index 701104d..2bdf330 100644 --- a/tests/obtain_vid_info.py +++ b/tests/obtain_vid_info.py @@ -6,12 +6,12 @@ from .utils.vid_url_to_html import get_vid_html_from_url class TestObtainVidInfo(TestCase): def test_obtain_vid_duration_from_shorts(self) -> None: url = "https://www.youtube.com/shorts/iD1Z7ccGyhk" - self.assertEqual(60, obtain_vid_duration(url, html=get_vid_html_from_url(url))) + self.assertEqual(60, obtain_vid_duration(url, '', html=get_vid_html_from_url(url))) def test_obtain_vid_duration_from_videos(self) -> None: url = "https://www.youtube.com/watch?v=WI4U1SVIO3I" - self.assertEqual(8*60+11, obtain_vid_duration(url, html=get_vid_html_from_url(url))) + self.assertEqual(8*60+11, obtain_vid_duration(url, '', html=get_vid_html_from_url(url))) def test_obtain_vid_duration_from_videos_with_params(self) -> None: url = "https://www.youtube.com/watch?v=k7RM-ot2NWY&list=PLZHQObOWTQDPD3MizzM2xVFitgF8hE_ab&index=2&pp=iAQB" - self.assertEqual(9*60+59, obtain_vid_duration(url, html=get_vid_html_from_url(url))) + self.assertEqual(9*60+59, obtain_vid_duration(url, '', html=get_vid_html_from_url(url))) diff --git a/tests/utils/vid_url_to_html.py b/tests/utils/vid_url_to_html.py index 60f6ccc..91fcb40 100644 --- a/tests/utils/vid_url_to_html.py +++ b/tests/utils/vid_url_to_html.py @@ -18,5 +18,5 @@ def get_vid_html_from_url(url: str) -> str: with open(f'tests/data/video@{extract_vid_id(url)}.html', 'r') as file: return file.read() -def obtain_vid_duration(url:str, html: str = '') -> int: +def obtain_vid_duration(url:str, vid_id:str, html:str='', api_key:str='') -> int: return get_random_vid_duration(url) -- cgit v1.2.3-70-g09d2