diff options
Diffstat (limited to 'data_analyser')
| -rw-r--r-- | data_analyser/utils.py | 9 |
1 files changed, 6 insertions, 3 deletions
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 |