blob: 921015de4036ea77b4c1c1d66d80f370eab9219d (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
from traceback import print_exc
from pymongo.collection import Collection
from components.subscriptions.main import Subscription
from components.subscriptions.typing import SubsDict
from components.videos import VideoTuple
from components.extractor.obtain_vid_info import obtain_vid_duration
def analyse_video(vid_tuple: VideoTuple, api_key: str='') -> VideoTuple:
try:
duration = obtain_vid_duration(vid_tuple.link, vid_tuple.id, api_key=api_key)
except:
print_exc()
duration = -2
return vid_tuple._replace(analysed=True, duration=duration)
def analyse_subscription(sub: Subscription, api_key: str='') -> bool:
updated = False
for i, vid in enumerate(sub.videos):
if not vid.analysed:
sub.videos[i] = analyse_video(vid, api_key)
updated = True
return updated
def analyse_collection(subs_collection: Collection[SubsDict], api_key: str='') -> int:
num_updated = 0
for sub_dict in subs_collection.find():
sub = Subscription(**sub_dict)
sub._collection = subs_collection
if analyse_subscription(sub, api_key):
sub.update_videos()
num_updated += 1
return num_updated
|
