在 Python 中对假期/工作日进行多步日期检查

Question

场景：我正在尝试构建一个日期检查功能。该函数的输入是：年、月和假期列表（只是天的列表）。该函数将检查 input_date 是否在假期列表中，如果是，则将一天与 input_date 相加。接下来它将检查是哪个工作日：如果是星期六（总和 2），如果是星期日（总和 1）。

问题：我的问题是，如果我只是创建几个 if 子句来检查这些条件，周末检查后的新日期也可能在假期列表中，所以该过程应该重复，直到找到不在假期列表中的工作日。

问题：有没有办法将这些检查包含在一个循环中？

到目前为止的代码：

def datecheck (input_year, input_month, holiday_list):      
    datetime.inputdate =  date(year, month, 1) 

        # this checks if inputdate is a holiday   
        if inputdate is in holiday_list: 
            inputdate == inputdate + datetime.timedelta(days=1)

        # next part guarantees inputdate is a weekday
        if inputdate.weekday() == 5: 
            inputdate == inputdate + datetime.timedelta(days=2)
        else if inputdate.weekday() == 6:
            inputdate == inputdate + datetime.timedelta(days=1)

        #check again for holiday?

        outputdate = inputdate
    return outputdate

Answer 1

while 循环可以解决问题：

    # this checks if inputdate is a holiday   
    while inputdate in holiday_list: 
        inputdate == inputdate + datetime.timedelta(days=1)

它将继续运行代码，直到表达式（如果 inputdate 在 holiday_list 中）计算结果为 false。

将其余逻辑添加到循环中的示例：

def datecheck (input_year, input_month, holiday_list):      
    inputdate =  date(input_year, input_month, 1) 

    valid = False

    while not valid:
        # this checks if inputdate is a holiday   
        if inputdate is in holiday_list: 
            inputdate == inputdate + datetime.timedelta(days=1)

        # next part guarantees inputdate is a weekday
        elif inputdate.weekday() == 5: 
            inputdate == inputdate + datetime.timedelta(days=2)

        elif inputdate.weekday() == 6:
            inputdate == inputdate + datetime.timedelta(days=1)

        else:
            valid = True

    return inputdate

我还清理了其他一些东西：

• else if 应该是 elif，而 is in 应该只是 in
• 一旦循环到位，就不需要outputdate，因为一旦循环完成，inputdate 将具有正确的值
• datetime.inputdate 应该只是 inputdate 我假设？ datetime 是标准库中模块的名称，但它没有 inputdate 属性
• 第一个 if 语句及以后的语句缩进太远
• 我还将正确的参数名称放入date()，但请注意，您需要在此代码上方使用from datetime import date 之类的名称，以便识别date。