from decorator.singleton import singleton
from utils.date_utils import DateUtils
# 引入数据库操作相关
from db.repository.config_rep import ConfigRepository


@singleton
class Config(object):
    """项目配置管理类"""

    def __init__(self) -> None:
        self.c_rep = ConfigRepository()
        # 数据爬取任务相关配置参数
        self.request_config = None
        self.login_info = None

    def check_request_config(self) -> bool:
        """检查当前数据库中的请求配置是否更新"""

        new_request_config = self.c_rep.read_request_config()
        if (self.request_config == None) or (new_request_config.version > self.request_config.version):
            self.request_config = new_request_config
            return True
        else:
            return False

    def get_login_info(self):
        """获取登录参数"""

        request_cookie_valid_duration = 6
        if self.request_config:
            request_cookie_valid_duration = self.request_config.request_cookie_valid_duration

        if self.login_info == None:
            self.login_info = self.c_rep.get_login_info()

        cookie = self.login_info.cookie
        timestamp = self.login_info.cookie_timestamp
        account = self.login_info.user_name
        password = self.login_info.password

        diff_hours = DateUtils.time_distance(timestamp, type="hour")
        if diff_hours > request_cookie_valid_duration:
            cookie = None
        return account, password, cookie

    def update_cookie(self, cookies: str):
        """更新登录cookie"""

        
        self.login_info.cookie = cookies
        self.login_info.cookie_timestamp = DateUtils.now()
        self.c_rep.update_login_info(self.login_info)