blob: 355e8306ae75343011763baf88fe66feafb9a51d (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
from urllib.parse import urlparse, parse_qs
from components.extractor.check_url import is_video
def extract_vid_id(url: str) -> str:
if not is_video(url):
raise Exception(url + " is not a YouTube video URL")
parsed_url = urlparse(url)
if parsed_url.netloc == 'youtu.be':
return parsed_url.path[1:]
if parsed_url.path == '/watch':
query_params = parse_qs(parsed_url.query)
return query_params['v'][0]
return parsed_url.path.split('/')[-1]
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()
|
