Python使用holiday-cn库实现节假日智能识别工具

# 1. 基础数据导入
holiday_data = {
    '2025': {
        'holidays': ['0101', '0128', '0129', '0130', '0131', '0203', '0204'],
        'workdays': ['0126', '0208', '0427']
    }
}

# 2. 数据验证和格式化
def validate_holiday_data(year_data):
    """验证节假日数据的有效性"""
    required_keys = ['holidays', 'workdays']
    for key in required_keys:
        if key not in year_data:
            raise ValueError(f"缺少必要字段: {key}")
    return year_data

# 3. 数据集成到系统
class HolidayManager:
    def __init__(self):
        self.holiday_data = {}
    
    def import_data(self, data):
        """导入节假日数据"""
        for year, year_data in data.items():
            validated_data = validate_holiday_data(year_data)
            self.holiday_data[year] = validated_data

from datetime import datetime
import holiday_cn

# 1. 字符串格式
result1 = holiday_cn.is_holiday('2025-01-01')

# 2. 时间戳格式
timestamp = 1735660800
result2 = holiday_cn.is_oliday(timestamp)

# 3. datetime对象
now = datetime.now()
result3 = holiday_cn.is_holiday(now)

# 3. 默认当前时间
result4 = holiday_cn.is_holiday()

import pandas as pd
from holiday_cn import ChinaHoliday

# 批量处理日期数据
def batch_check_holidays(date_list):
    """批量检查节假日状态"""
    results = []
    for date_str in date_list:
        is_holiday = ChinaHoliday.is_holiday(date_str)
        results.append({
            'date': date_str,
            'is_holiday': is_holiday,
            'type': 'holiday' if is_holiday else 'workday'
        })
    return pd.DataFrame(results)

# 使用示例
dates = ['2025-01-01', '2025-01-26', '2025-02-08']
df = batch_check_holidays(dates)
print(df)

class AttendanceSystem:
    def __init__(self):
        self.holiday_checker = ChinaHoliday()
    
    def calculate_work_days(self, start_date, end_date):
        """计算两个日期之间的实际工作日"""
        work_days = 0
        current_date = start_date
        
        while current_date <= end_date:
            if self.holiday_checker.is_workday(current_date):
                work_days += 1
            current_date += timedelta(days=1)
        
        return work_days

def plan_promotion_dates(year):
    """规划年度促销活动日期"""
    promotion_dates = []
    current_date = datetime(year, 1, 1)
    
    for i in range(365):
        if ChinaHoliday.is_holiday(current_date):
            # 节假日适合大型促销
            promotion_dates.append({
                'date': current_date,
                'type': 'major_promotion'
            })
        current_date += timedelta(days=1)
    
    return promotion_dates

class FinancialCalculator:
    def calculate_settlement_date(self, trade_date, settlement_days):
        """计算金融交易的结算日期"""
        current_date = trade_date
        days_counted = 0
        
        while days_counted < settlement_days:
            current_date += timedelta(days=1)
            if ChinaHoliday.is_workday(current_date):
                days_counted += 1
        
        return current_date

class OptimizedHolidayManager:
    def __init__(self):
        # 使用压缩数据格式
        self.compressed_data = {}
    
    def compress_date(self, date_str):
        """压缩日期格式，减少内存占用"""
        return date_str.replace('-', '')

from functools import lru_cache

class CachedHolidayChecker:
    def __init__(self):
        self.cache_hits = 0
        self.cache_misses = 0
    
    @lru_cache(maxsize=1000)
    def is_holiday_cached(self, date_str):
        """带缓存的节假日判断"""
        return ChinaHoliday.is_holiday(date_str)

import pandas as pd
from holiday_cn import ChinaHoliday

def add_holiday_info_to_dataframe(df, date_column):
    """为 DataFrame 添加节假日信息"""
    df['is_holiday'] = df[date_column].apply(
        lambda x: ChinaHoliday.is_holiday(x.strftime('%Y-%m-%d'))
    return df

# 使用示例
date_range = pd.date_range('2025-01-01', '2025-12-31')
df = pd.DataFrame({'date': date_range})
df_with_holiday = add_holiday_info_to_dataframe(df, 'date')

# Django models.py
from django.db import models
from holiday_cn import ChinaHoliday

class ProjectSchedule(models.Model):
    start_date = models.DateField()
    end_date = models.DateField()
    
    def get_actual_work_days(self):
        """获取实际工作日天数"""
        return self.calculate_work_days(self.start_date, self.end_date)

import numpy as np
from sklearn.preprocessing import LabelEncoder

class HolidayFeatureGenerator:
    def __init__(self):
        self.encoder = LabelEncoder()
    
    def generate_features(self, dates):
        """生成节假日相关特征"""
        features = []
        for date in dates:
            is_holiday = ChinaHoliday.is_holiday(date)
            features.append([
                int(is_holiday),
                int(ChinaHoliday.is_workday(date)),
                self.get_holiday_type(date)
            ])
        return np.array(features)