import datetime
from chinese_calendar import (is_holiday, is_workday)


def get_data_datetime(show_date):
    show_date_obj = datetime.datetime.strptime(show_date, '%Y-%m-%d')
    target_start_datetime_obj = show_date_obj + datetime.timedelta(hours=6)
    target_end_datetime_obj = show_date_obj + datetime.timedelta(hours=29, minutes=59, seconds=59)
    data_start_datetime_obj = show_date_obj + datetime.timedelta(days=-7) + datetime.timedelta(hours=6)
    data_end_datetime_obj = show_date_obj + datetime.timedelta(hours=5, minutes=59, seconds=59)
    history_start = datetime.datetime.strftime(data_start_datetime_obj, '%Y-%m-%d %H:%M:%S')
    history_end = datetime.datetime.strftime(data_end_datetime_obj, '%Y-%m-%d %H:%M:%S')
    target_start = datetime.datetime.strftime(target_start_datetime_obj, '%Y-%m-%d %H:%M:%S')
    target_end = datetime.datetime.strftime(target_end_datetime_obj, '%Y-%m-%d %H:%M:%S')
    return history_start, history_end, target_start, target_end


def get_date_type(show_date, template_date):
    show_date_obj = datetime.datetime.strptime(show_date, '%Y-%m-%d')
    template_date_obj = datetime.datetime.strptime(template_date, '%Y-%m-%d')
    result = dict()
    # 处理参考数据日期
    history_date_list = [(show_date_obj - datetime.timedelta(days=i)) for i in range(7, 0, -1)]
    history_date = '、'.join(
        [f'{datetime.date.strftime(d, "%Y-%m-%d")}（{"节假日" if is_holiday(d) else "工作日"}）' for d in
         history_date_list])
    # 处理目标日期
    show_date = f'{datetime.date.strftime(show_date_obj, "%Y-%m-%d")}（{"节假日" if is_holiday(show_date_obj) else "工作日"}）'
    show_date_is_holiday = is_holiday(show_date_obj)
    # 处理模板日期
    template_date = f'{datetime.date.strftime(template_date_obj, "%Y-%m-%d")}（{"节假日" if is_holiday(template_date_obj) else "工作日"}）'
    template_date_is_holiday = is_holiday(template_date_obj)
    return history_date, show_date, show_date_is_holiday, template_date, template_date_is_holiday


if __name__ == '__main__':
    get_date_type("2026-06-13", "2026-06-01")