# 时间服务
from heurams.context import config_var
import time


def get_daystamp() -> int:
    """获取当前日戳(以天为单位的整数时间戳)"""
    time_override = config_var.get().get("daystamp_override", -1)
    if time_override != -1:
        return int(time_override)

    return int((time.time() + config_var.get().get("timezone_offset")) // (24 * 3600))


def get_timestamp() -> float:
    """获取 UNIX 时间戳"""
    # 搞这个类的原因是要支持可复现操作
    time_override = config_var.get().get("timestamp_override", -1)
    if time_override != -1:
        return float(time_override)

    return time.time()