From 4372982119140dba5b132e09200b94c1b422a1dd Mon Sep 17 00:00:00 2001 From: A Farzat Date: Wed, 8 Oct 2025 12:39:50 +0300 Subject: Change duration to int in VideoTuple This allows data_analyser to use extractor functions instead of yt-dlp. --- data_analyser/utils.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'data_analyser') 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 -- cgit v1.2.3-70-g09d2