From c641e6f1948f683bf7b1fff08bce125cb64563df Mon Sep 17 00:00:00 2001 From: A Farzat Date: Wed, 8 Oct 2025 10:44:07 +0300 Subject: Add unit tests for obtain_vid_info --- components/extractor/check_url.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'components/extractor') diff --git a/components/extractor/check_url.py b/components/extractor/check_url.py index b574b22..13d7259 100644 --- a/components/extractor/check_url.py +++ b/components/extractor/check_url.py @@ -11,7 +11,10 @@ def is_video(url: str) -> bool: if not is_youtube(url): return False parsed_url = urlparse(url) - if parsed_url.path in ('/watch', '/shorts/', '/embed/'): + for path in ('/shorts/', '/embed/'): + if parsed_url.path.startswith(path): + return True + if parsed_url.path == '/watch' and 'v' in parse_qs(parsed_url.query): return True return parsed_url.netloc == 'youtu.be' -- cgit v1.2.3-70-g09d2