# -*- coding: utf-8 -*-
"""
@File    : time_helper.py
@Time    : 2023/1/14 17:09
@Author  : geekbing
@LastEditTime : -
@LastEditors : -
@Description : -
"""


# 获取时间戳
import datetime
import time


def get_day(days: int = 0, **kwargs) -> str:
    """
    获取日期，默认无连接符格式
    :param days: 负数表示过去，整数表示未来
    :param kwargs: 可选h, m, s
        h: 第n小时
        m: 第n分钟
        s: 第n秒
    :return:

    >>> get_day()
    # 今天的日期

    >>> get_day(1)
    # 明天的日期

    >>> get_day(-1)
    # 昨天的日期

    >>> get_day(h=9,m=15,s=30)
    # 今天的9时15分30秒 20230116 9:15:30

    >>> get_day(sep='-', h=9,m=15,s=30)
    # 2021-08-07 9:15:30 # 今天的9时15分30秒, 日期分隔符是-
    """
    d = datetime.timedelta(days)
    n = datetime.datetime.now()
    sep: str = kwargs.get("sep", "")
    fmt = sep.join(["%Y", "%m", "%d"])
    if kwargs:
        h = kwargs.get("h", "00")
        m = kwargs.get("m", "00")
        s = kwargs.get("s", "00")
        fmt = f"{fmt} {h}:{m}:{s}"
    return (n + d).strftime(fmt)


def get_ts(interval: int = 0, t=None) -> str:
    """获取时间戳
    :param interval: 时间间隔，t为None时，时间间隔为秒，其它则为天
    :param t: 要转换成时间戳的时分秒
    :return:

    >>> get_ts(0)
    # 当前时间戳

    >>> get_ts(0, 9)
    # 今天9点的时间戳

    >>> get_ts(1, 9)
    # 明天9点的时间戳

    >>> get_ts(0, '9:15')
    # 今天9点15分的时间戳

    >>> get_ts(0, '9:15:10')
    # 今天9点15分10秒的时间戳
    """
    if isinstance(interval, int) and t is None:
        return str(int(time.time()) + interval)
    else:
        time_format = "%Y%m%d"
        t = str(t)
        if t.count(":") == 0 and len(t.split()) == 1:
            time_format = time_format + "%H"
        elif t.count(":") == 1 and len(t.split()) == 1:
            time_format = time_format + "%H:%M"
        elif t.count(":") == 2 and len(t.split()) == 1:
            time_format = time_format + "%H:%M:%S"
        time_arr = time.strptime(get_day(interval) + t, time_format)
        return str(int(time.mktime(time_arr)))


def get_ts_int(interval: int = 0, t=None) -> int:
    """获取时间戳
    :param interval: 时间间隔，t为None时，时间间隔为秒，其它则为天
    :param t: 要转换成时间戳的时分秒
    :return:
    """
    return int(get_ts(interval, t))


def get_day_fmt(fmt_type="sec", **kwargs):
    """获取日期

    fmt_type : str
        日期格式，默认'sec', 包含时分秒, 'day'只包含年月日
    kwargs : str, key可以是days, hours, seconds, 等
        时间差值，负数表示过去，正数表示未来
        昨天：days=-1，明天days=1

    >>> get_day_fmt()
    # 当前日期，包含时分秒
    2021-08-07 12:08:54

    >>> get_day_fmt('day')
    # 当前日期，只包含年月日
    2021-08-07

    # 昨天的日期
    >>> get_day_fmt('day', days=-1)
    2021-08-06
    """
    fmt = "%Y-%m-%d %H:%M:%S"
    if fmt_type == "day":
        fmt = "%Y-%m-%d"

    day = datetime.timedelta(**kwargs)
    now = datetime.datetime.now()
    return (now + day).strftime(fmt)


def get_day_h(days: int = 0, **kwargs):
    """
    >>> get_day_h(0,h=7)
    2021042207
    """
    d = datetime.timedelta(days)
    n = datetime.datetime.now()
    time_str = f"%Y%m%d"
    if kwargs:
        h = str(kwargs.get("h"))
        h = h.rjust(2, "0")
        time_str = f"{time_str}{h}"
    return (n + d).strftime(time_str)


def get_hour_ts(interval: int = 0) -> str:
    # 当前时间所在的小时，转换为时间戳
    # 2021-3-9 17:0:12, 取2021-3-9 17:0:0，然后转换为时间戳
    now_hour = int(
        datetime.datetime.now().replace(minute=0, second=0, microsecond=0).timestamp()
    )
    return str(now_hour + interval * 3600)


def get_hour() -> str:
    # 获取当前的小时，不足两位补0
    return str(datetime.datetime.now().hour).rjust(2, "0")


def get_hour_ts_int(interval: int = 0) -> int:
    return int(get_hour(interval))


def wait(i: int = 0):
    time.sleep(i)