blob: 91fcb4065320db1484dbfc068bc68714172ea505 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
from urllib.parse import urlparse, parse_qs
from components.extractor.check_url import is_video
from .get_random_vid_info import get_random_vid_duration
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()
def obtain_vid_duration(url:str, vid_id:str, html:str='', api_key:str='') -> int:
return get_random_vid_duration(url)
|
