diff options
| -rw-r--r-- | components/extractor/obtain_vid_info.py | 13 | ||||
| -rw-r--r-- | requirements.txt | 1 |
2 files changed, 14 insertions, 0 deletions
diff --git a/components/extractor/obtain_vid_info.py b/components/extractor/obtain_vid_info.py new file mode 100644 index 0000000..bfbe861 --- /dev/null +++ b/components/extractor/obtain_vid_info.py @@ -0,0 +1,13 @@ +from urllib.request import urlopen + +from bs4 import BeautifulSoup +from isodate import parse_duration # type: ignore + +def obtain_vid_duration(url: str, html: str = '') -> int: + 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()) diff --git a/requirements.txt b/requirements.txt index 56de09d..53fab07 100644 --- a/requirements.txt +++ b/requirements.txt @@ -10,6 +10,7 @@ Flask==3.1.2 flask-cors==6.0.1 gunicorn==23.0.0 idna==3.10 +isodate==0.7.2 itsdangerous==2.2.0 Jinja2==3.1.6 MarkupSafe==3.0.2 |