Python生成中国节假日工作表，快速给数据库内生成工作日历

<script src="https://cdn.jsdelivr.net/npm/chinese-days/dist/index.min.js"></script>
<script>
  // 检查某个日期是否为工作日
  console.log(isWorkday('2023-01-01')); // false
  // isHoliday 检查某个日期是否为节假日
  console.log(isHoliday('2023-01-01')); // true
  //  检查某个日期是否为调休日（in lieu day）
  //    检查 2024-05-02 返回 `true` 则表示是一个调休日。
  console.log(isInLieu('2024-05-02')); // true
  //    检查 2024-05-01 返回 `false` 则表示不是一个调休日。
  console.log(isInLieu('2024-05-01')); // false
</script>

class Holiday(Enum):
    def __new__(cls, english, chinese, days):
        obj = object.__new__(cls)
        obj._value_ = english

        obj.chinese = chinese
        obj.days = days
        return obj

    new_years_day = "New Year's Day", "元旦", 1
    spring_festival = "Spring Festival", "春节", 3
    tomb_sweeping_day = "Tomb-sweeping Day", "清明", 1
    labour_day = "Labour Day", "劳动节", 1
    dragon_boat_festival = "Dragon Boat Festival", "端午", 1
    national_day = "National Day", "国庆节", 3
    mid_autumn_festival = "Mid-autumn Festival", "中秋", 1

    # special holidays
    anti_fascist_70th_day = "Anti-Fascist 70th Day", "中国人民抗日战争暨世界反法西斯战争胜利70周年纪念日", 1

import datetime

# 判断 2018年4月30号 是不是节假日
from chinese_calendar import is_holiday, is_workday
april_last = datetime.date(2018, 4, 30)
assert is_workday(april_last) is False
assert is_holiday(april_last) is True

# 或者在判断的同时，获取节日名
import chinese_calendar as calendar  # 也可以这样 import
on_holiday, holiday_name = calendar.get_holiday_detail(april_last)
assert on_holiday is True
assert holiday_name == calendar.Holiday.labour_day.value

# 还能判断法定节假日是不是调休
import chinese_calendar
assert chinese_calendar.is_in_lieu(datetime.date(2006, 2, 1)) is False
assert chinese_calendar.is_in_lieu(datetime.date(2006, 2, 2)) is True

# year为所属的年，如：2024
begin = datetime.date(year, 1, 1)

def get_whole_year(year=TARGET_YEAR):
    """
    获取一年内所有的日期
    :param year: 获取的年
    :return: 日期数组
    """
    begin = datetime.date(year, 1, 1)  # 设置开始日期为给定年的1月1日
    now = begin
    end = datetime.date(year, 12, 31)  # 设置结束日期为给定年的12月31日
    delta = datetime.timedelta(days=1)  # 定义日期增量为1天
    days = []  # 初始化日期数组
    while now <= end:  # 循环直到当前日期达到结束日期
        days.append(now.strftime("%Y-%m-%d"))  # 将当前日期以"YYYY-MM-DD"格式添加到数组中
        now += delta  # 增加日期增量
    return days  # 返回日期数组

class DATATYPE(Enum):
    """
    日期类型枚举类
    """
    WORKDAY = ("0", "普通工作日")
    WEEKEND = ("3", "普通周末")
    HOLIDAY = ("1", "节日假期")
    WORKING_HOLIDAY = ("2", "节日补班")

    def __init__(self, code: str, description: str):
        self.code = code
        self.description = description

    @property
    def code(self) -> str:
        return self._code

    @code.setter
    def code(self, value: str):
        self._code = value

    @property
    def description(self) -> str:
        return self._description

    @description.setter
    def description(self, value: str):
        self._description = value

if calendar.is_holiday(date):
    print("{}是节假日".format(judge_date))
    on_holiday, holiday_name = calendar.get_holiday_detail(date)
    # 判断是否为节日（否：周末；是：节日）
    if holiday_name is not None:
        return DATATYPE.HOLIDAY.code + "-" + str(holiday_name)
    else:
        return DATATYPE.WEEKEND.code

def judge_date_type(judge_date):
    """
    判断日期的类型
    :param judge_date:
    :return: 判断的类型；如果是假期有关，附带假期备注
    """
    date = datetime.datetime.strptime(judge_date, '%Y-%m-%d').date()
    if calendar.is_holiday(date):
        print("{}是节假日".format(judge_date))
        on_holiday, holiday_name = calendar.get_holiday_detail(date)
        # 判断是否为节日（否：周末；是：节日）
        if holiday_name is not None:
            return DATATYPE.HOLIDAY.code + "-" + str(holiday_name)
        else:
            return DATATYPE.WEEKEND.code
    elif calendar.is_workday(date):
        on_holiday, holiday_name = calendar.get_holiday_detail(date)
        if holiday_name is not None:
            # 节日名称为非空，说明是补班，否则就是普通工作日
            print("{}是补班日".format(judge_date))
            return DATATYPE.WORKING_HOLIDAY.code + "-" + str(holiday_name)
        else:
            return DATATYPE.WORKDAY.code
    else:
        # 理论上不存在没有匹配的情况
        print("{}没有匹配" .format(judge_date))
        assert False