summaryrefslogtreecommitdiff
path: root/data_analyser/__main__.py
diff options
context:
space:
mode:
Diffstat (limited to 'data_analyser/__main__.py')
-rwxr-xr-xdata_analyser/__main__.py25
1 files changed, 1 insertions, 24 deletions
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)