From eaeb27d20a4d212a3ec3260ca0ccbfff31496358 Mon Sep 17 00:00:00 2001 From: A Farzat Date: Mon, 6 Oct 2025 05:45:19 +0300 Subject: Move analyse functions to a separate file This way they can be imported by other modules (e.g. for testing). --- data_analyser/__main__.py | 25 +------------------------ 1 file changed, 1 insertion(+), 24 deletions(-) (limited to 'data_analyser/__main__.py') diff --git a/data_analyser/__main__.py b/data_analyser/__main__.py index 6f996d9..8b4a984 100755 --- a/data_analyser/__main__.py +++ b/data_analyser/__main__.py @@ -1,32 +1,9 @@ #!/usr/bin/env python from time import sleep +from .utils import analyse_collection from components.database import subscriptions -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 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"]) - -def analyse_subscription(sub: Subscription) -> bool: - updated = False - for i, vid in enumerate(sub.videos): - if not vid.analysed: - sub.videos[i] = analyse_video(vid) - updated = True - return updated - -def analyse_collection(subs_collection: Collection[SubsDict]) -> None: - for sub_dict in subs_collection.find(): - sub = Subscription(**sub_dict) - if analyse_subscription(sub): - sub.update_videos() while True: analyse_collection(subscriptions) -- cgit v1.2.3-70-g09d2