diff options
Diffstat (limited to 'components/extractor/obtain_vid_info.py')
| -rw-r--r-- | components/extractor/obtain_vid_info.py | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/components/extractor/obtain_vid_info.py b/components/extractor/obtain_vid_info.py index bfbe861..45e2bdc 100644 --- a/components/extractor/obtain_vid_info.py +++ b/components/extractor/obtain_vid_info.py @@ -1,9 +1,25 @@ +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, html: str = '') -> int: +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') |