blob: 45e2bdc3702a9e8d0006a5317bb3ad0febae9476 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
from sys import stderr
from traceback import print_exc
from urllib.request import urlopen
from bs4 import BeautifulSoup
from isodate import parse_duration # type: ignore
from requests import get
def obtain_vid_duration(url: str, vid_id: str, html: str='', api_key: str='') -> int:
if api_key:
try:
data = get("https://www.googleapis.com/youtube/v3/videos", params={
'part': "contentDetails",
'id': vid_id[9:],
'key': api_key,
}).json()
duration_str = data['items'][0]['contentDetails']['duration']
print(vid_id[9:], duration_str)
return int(parse_duration(duration_str).total_seconds())
except:
print("Web scraping will be used due to an error with the following id:", vid_id, file=stderr)
print_exc()
html = html or urlopen(url).read().decode('utf-8')
soup = BeautifulSoup(html, 'html.parser')
duration_meta = soup.find('meta', itemprop='duration')
assert duration_meta
duration = parse_duration(duration_meta['content'])
return int(duration.total_seconds())
|
