diff options
| -rw-r--r-- | components/videos.py | 2 | ||||
| -rw-r--r-- | data_analyser/utils.py | 9 |
2 files changed, 7 insertions, 4 deletions
diff --git a/components/videos.py b/components/videos.py index 31b5005..cbb7f66 100644 --- a/components/videos.py +++ b/components/videos.py @@ -12,7 +12,7 @@ class VideoTuple(NamedTuple): thumbnail: str summary: str analysed: bool = False - duration_string: str = "" + duration: int = -1 @classmethod def from_rss_entry(cls, entry: Any) -> Self: diff --git a/data_analyser/utils.py b/data_analyser/utils.py index 115c216..95dc2fe 100644 --- a/data_analyser/utils.py +++ b/data_analyser/utils.py @@ -1,13 +1,16 @@ from components.subscriptions.main import Subscription from components.subscriptions.typing import SubsDict from components.videos import VideoTuple -from components.ytdlp import obtain_vid_info +from components.extractor.obtain_vid_info import obtain_vid_duration from pymongo.collection import Collection def analyse_video(vid_tuple: VideoTuple) -> VideoTuple: - info = obtain_vid_info(vid_tuple.link) - return vid_tuple._replace(analysed=True, duration_string=info["duration_string"]) + try: + duration = obtain_vid_duration(vid_tuple.link) + except: + duration = -2 + return vid_tuple._replace(analysed=True, duration=duration) def analyse_subscription(sub: Subscription) -> bool: updated = False |