Python 计算端午节是不是小长假

"""
计算端午是不是小长假
小长假放假的逻辑是：
节日只放一天，如果碰上周五、周六、周日、周一，就会三天连在一起放，组成小长假。

"""

from datetime import datetime


def is_holiday(_d):
    """
    判断给定日期是否为小长假。

    参数:
    _d: 字符串类型，格式为'YYYY-MM-DD'的日期字符串。

    返回值:
    布尔类型，如果日期是小长假则返回True，否则返回False。
    """
    _d = datetime.strptime(d, '%Y-%m-%d')  # 将日期字符串转换为datetime对象
    weekday = _d.weekday() + 1  # 获取日期的星期几，星期一为1，星期日为7
    holiday_range = [5, 6, 7, 1]  # 定义小长假的星期几范围（周五到下周一）

    # 判断日期是否为小长假
    return True if weekday in holiday_range or weekday + 1 in holiday_range or weekday - 1 in holiday_range else False


if __name__ == '__main__':
    d = '2024-06-10'
    if is_holiday(d):
        print(f'{d}是小长假。')
    else:
        print(f'{d}不是小长假。')