from datetime import datetime
import numpy as np
from fastapi.responses import JSONResponse

import re
import os
import logging

# 获取日志记录器
logger = logging.getLogger(__name__)

from app.tools.effective_date import effective_date


# files是通过os.listdir拿到的目标文件夹里的所有文件
def get_time(files, time_type):
    try:
        logger.info("开始获取日报需分析的时间段")

        # 拿到文件夹下所有文件名
        # folder_path = r'E:\work_data\work\三工单日报\三工单\20250306\源数据'
        #
        # files = os.listdir(folder_path)

        # 获取生成日报的时间
        pattern_time = r"\d{1,2}月\d{1,2}日"
        # 另一个时间
        pattern_time2 = r"(\d{4})(\.doc|\.xls|-[^\d]|[\)）])"

        time_list = []

        for filename in files:
            print(filename)
            time_temp1 = re.search(pattern_time, filename)
            time_temp2 = re.search(pattern_time2, filename)

            if time_temp1:
                time_list.append(time_temp1.group())
            if time_temp2:
                temp_month = re.sub(r"^0", "", time_temp2.group(1)[:2])
                temp_day = re.sub(r"^0", "", time_temp2.group(1)[2:])
                time_list.append(temp_month + "月" + temp_day + "日")

        print(time_list)

        if (
            len(time_list) > 3
            and time_list[0] == time_list[1] == time_list[2] == time_list[3]
        ):
            # 把x月x日按照‘月’进行拆分
            date_list = time_list[0].split("月")

            # 获取到今天的年月日信息
            year = str(datetime.now().year)
            month = date_list[0]
            day = date_list[1].replace("日", "")

            # 调用自己写的方法获取昨天的年月日
            # 昨天
            year_before, month_before, day_before = effective_date(year, month, day)
            # 前天
            year_before2, month_before2, day_before2 = effective_date(
                str(year_before), str(month_before), str(day_before)
            )

            # 先设置时间的默认值，然后根据统计时间类型进行区分
            start_time = None
            end_time = None
            before_start_time = None

            if time_type == 0:
                # ！！！旧版，拿到生成日报的开始时间和结束时间
                # 当天17点开始时间/前一天17点结束时间
                start_time = datetime(year_before, month_before, day_before, 17, 0, 0)
                # 当天结束时间
                end_time = datetime(int(year), int(month), int(day), 17, 0, 0)
                # 前一天开始时间
                before_start_time = datetime(
                    year_before2, month_before2, day_before2, 17, 0, 0
                )

            elif time_type == 1:
                # --------------------------------20250429修改-------------------------------------------------
                # 从0点开始计算到23:59:59
                start_time = datetime(int(year), int(month), int(day), 0, 0, 0)
                # 当天结束时间
                end_time = datetime(int(year), int(month), int(day), 23, 59, 59)
                # 前一天开始时间
                before_start_time = datetime(
                    year_before, month_before, day_before, 0, 0, 0
                )
                # --------------------------------20250429修改-------------------------------------------------

            return (
                start_time,
                end_time,
                before_start_time,
                year,
                month,
                day,
                day_before,
                month_before,
            )
        else:
            raise Exception("请确认各文件是否为同一天的")

    except Exception as e:
        logger.exception(f"获取日报时间失败：{e}")
        print(f"获取日报时间失败：{e}")


# if __name__ == '__main__':
# get_time()