# 日期類型顏色映射
COLOR_SCHEME = {
    "節(jié)假日": "#FFB6C1",    # 淺紅色 - 法定節(jié)假日
    "工作日": "#F0F8FF",    # 淺藍(lán)色 - 正常工作日
    "周末休息": "#FFE4E1",  # 淺粉色 - 周末休息
    "調(diào)休上班": "#FFF8DC"   # 淺黃色 - 調(diào)休工作日
}

def is_holiday(self, year, month, day):
    """判斷是否為節(jié)假日"""
    try:
        date_obj = datetime(year, month, day).date()
        
        # 使用chinese_calendar判斷是否為節(jié)假日
        if chinese_calendar.is_holiday(date_obj):
            # 獲取節(jié)假日名稱
            holiday_name = chinese_calendar.get_holiday_detail(date_obj)
            if holiday_name:
                return True, holiday_name[0]
            else:
                return True, "節(jié)假日"
        
        return False, ""
    except Exception as e:
        # 異常處理：回退到備用邏輯
        return self.fallback_holiday_check(year, month, day)

def is_workday(self, year, month, day):
    """判斷是否為工作日（使用chinese_calendar）"""
    try:
        date_obj = datetime(year, month, day).date()
        return chinese_calendar.is_workday(date_obj)
    except:
        # 回退邏輯：非周末且非節(jié)假日
        date_obj = datetime(year, month, day)
        is_weekend = date_obj.weekday() >= 5
        is_holiday, _ = self.is_holiday(year, month, day)
        return not (is_weekend or is_holiday)

def create_day_widget(self, parent, year, month, day, row, col):
    """創(chuàng)建日期小部件"""
    date_obj = datetime(year, month, day)
    
    # 使用chinese_calendar判斷日期類型
    is_workday = self.is_workday(year, month, day)
    is_holiday, holiday_name = self.is_holiday(year, month, day)
    is_weekend = date_obj.weekday() >= 5
    
    # 智能顏色和標(biāo)識選擇
    if is_holiday:
        bg_color, type_text, type_color = "#FFB6C1", "假", "red"
    elif is_workday:
        if is_weekend:
            # 調(diào)休工作日（周末上班）
            bg_color, type_text, type_color = "#FFF8DC", "班", "orange"
        else:
            # 正常工作日
            bg_color, type_text, type_color = "#F0F8FF", "工", "blue"
    else:
        # 周末休息日
        bg_color, type_text, type_color = "#FFE4E1", "休", "purple"
    
    # 創(chuàng)建可視化組件...

def display_summary(self, year, month):
    """顯示匯總信息"""
    total_days = calendar.monthrange(year, month)[1]
    workdays = weekend_rest = holidays = weekend_work = 0
    
    for day in range(1, total_days + 1):
        date_obj = datetime(year, month, day)
        is_weekend = date_obj.weekday() >= 5
        is_workday = self.is_workday(year, month, day)
        is_holiday, _ = self.is_holiday(year, month, day)
        
        if is_holiday:
            holidays += 1
        elif is_workday:
            if is_weekend:
                weekend_work += 1  # 周末調(diào)休上班
            else:
                workdays += 1  # 正常工作日
        else:
            weekend_rest += 1  # 周末休息
    
    # 動(dòng)態(tài)顯示統(tǒng)計(jì)結(jié)果
    summary_text = (f"總天數(shù): {total_days}天 | "
                   f"工作日: {workdays}天 | "
                   f"周末休息: {weekend_rest}天 | "
                   f"節(jié)假日: {holidays}天")
    
    if weekend_work > 0:
        summary_text += f" | 調(diào)休上班: {weekend_work}天"

try:
    # 使用chinese_calendar主邏輯
    return chinese_calendar.is_holiday(date_obj)
except Exception as e:
    # 回退到備用邏輯
    return self.fallback_logic()

def load_holiday_data_thread(self):
    """異步加載節(jié)假日數(shù)據(jù)"""
    thread = threading.Thread(target=self.load_holiday_data)
    thread.daemon = True
    thread.start()