python脚本-工作日、休息日查询YYYY-MM-DD(万年历版)——>改SDK查询

注释都有,看代码

# coding=utf-8
import datetime
from chinese_calendar import is_workday


class WorkDay:
    def __init__(self,):
        pass

    # 传入日期,查询法定节假日
    def get_isWorkDay(self,data_time):
        try:
            date_obj = datetime.datetime.strptime(data_time, '%Y-%m-%d')

            # 判断是否为工作日
            if is_workday(date_obj):
                data = {"is_workday":1,"day_msg":f'{data_time}: 是工作日'}
            else:
                data = {"is_workday":0,"day_msg":f'{data_time}: 是休息日'}

            return {"success": True, "data": data, "msg": "查询法定节假日完成"}
        except Exception as e:
            return {"success": False, "data": {}, "msg": str(e)}


# 调用示例
if __name__ == '__main__':
    wh = WorkDay()
    return_msg = wh.get_isWorkDay(data_time="2025-1-1")
    if return_msg.get("success") is False:
        # logger.error(return_msg) # 函数用此return,错误上报
    else:
        # logger.debug({"success": True, "data": return_msg.get("data",{}), "msg": return_msg.get("msg","")}) 


你可能感兴趣的:(python,python)