summaryrefslogtreecommitdiff
path: root/api/utils.py
blob: 2c9e1e52d9cca1c8372838a6405bccfda4a51ae5 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from typing import Any, Dict, List
from components.subscriptions.main import Subscription
from components.subscriptions.typing import SubsDict
from components.videos import VideoTuple

def vid_dicts_from_tuple_list(tuple_list: List[VideoTuple]) -> List[Dict[str, Any]]:
    return [VideoTuple._make(vid)._asdict() for vid in tuple_list]

def sub_info_from_dict(sub_dict: SubsDict) -> Dict[str, Any]:
    return {
        **sub_dict,
        "videos": len(sub_dict["videos"]),
        "new_vids": len(Subscription(**sub_dict).get_new_vids()),
    }