import os
import time

import tool


@tool.singleton
class VideoManger:
    __video_cache_path = "datas/videos.json"
    # {"视频内容+视频时间":(评论数量,更新时间)}
    __video_data_cache_dict = {}

    def __init__(self):

        if os.path.exists(self.__video_cache_path):
            with open(self.__video_cache_path, encoding="utf-8", mode='r') as f:
                lines = f.readlines()
                if lines:
                    self.__video_data_cache_dict = eval(lines[0])

    def is_need_click(self, video_name, video_time, comment_count):
        """
        是否需要点击
        :param video_name: 视频名称
        :param video_time: 视频时间
        :param comment_count: 评论数量
        :return:
        """
        k = f"{video_name}{video_time}"
        if k not in self.__video_data_cache_dict:
            return True
        if comment_count - self.__video_data_cache_dict[k][0] > 0:
            return True
        return False

    def add_video_infos(self, video_infos):
        """
        添加视频信息
        :param video_infos:
        :return:
        """
        print("保存所有视频内容")
        self.__video_data_cache_dict = {f"{x[0]}{x[1]}": (x[2], time.time()) for x in video_infos}
        # 保存到文件
        with open(self.__video_cache_path, encoding="utf-8", mode='w') as f:
            f.write(f"{self.__video_data_cache_dict}")