查找下一个条件日期

Question

让我们有两个变量：

1. 开始日期
2. days_of_months::list（例如：[1,2,10,31]）

输出：一个日期>=start_date which ("%d") = 列表成员之一。 输出日期必须尽可能接近 start_date。

示例

开始日期：2020-11-17

days_of_month：[5,10,20]

输出：2020-11-20

开始日期：2020-11-17

days_of_month：[5,10]

输出：2020-12-05

开始日期：2020-12-17

days_of_month：[5,10]

输出：2021-01-05

开始日期：2021-02-17

days_of_month：[5,10,31]

输出：2021-03-05

开始日期：2021-04-17

days_of_month：[31]

输出：2021-05-31（2021年4月还没有31天）

建议答案：

from calendar import monthrange
from datetime import datetime
from datetime import timedelta

def find_condition_date(start_date,days_month_list):
    year = int(start_date.strftime("%Y"))
    month = int(start_date.strftime("%m"))
    day = int(start_date.strftime("%d"))
    
    current_month_days = monthrange(year, month)[1]
    
    if(day in days_month_list):
        return start_date
    else:
        for day_of_month_selected in days_month_list:
            if(day_of_month_selected>day and day_of_month_selected<=current_month_days):
                add_days = day_of_month_selected - day
                start_date = start_date + timedelta(days=add_days)
                return start_date
            
        selected_index = 0
        total_days_selected = len(days_month_list)
        if(month==12):
            year+=1
            month = 1
        else:
            month +=1
        current_month_days = monthrange(year,month)[1]
        
        while(True):
            if(selected_index<total_days_selected):
                first_day_of_month_selected = int(days_month_list[selected_index])
                if(current_month_days>=first_day_of_month_selected):
                    return datetime(year,month,first_day_of_month_selected).date()
                else:
                    selected_index = 0
                    if(month==12):
                        year+=1
                        month = 1
                    else:
                        month +=1
                    current_month_days = monthrange(year,month)[1]
            else:
                selected_index = 0
                if(month==12):
                    year+=1
                    month = 1
                else:
                    month +=1
                current_month_days = monthrange(year,month)[1]

#Example 1          
start_date = datetime(2020,11,17)
days_month_list = [5,10,20]
print(find_condition_date(start_date,days_month_list))

#Example 2          
start_date = datetime(2020,11,17)
days_month_list = [5,10]
print(find_condition_date(start_date,days_month_list))

#Example 3          
start_date = datetime(2020,12,17)
days_month_list = [5,10]
print(find_condition_date(start_date,days_month_list))

#Example 4
start_date = datetime(2021,2,17)
days_month_list = [5,10,31]
print(find_condition_date(start_date,days_month_list))

#Example 5
start_date = datetime(2021,4,17)
days_month_list = [31]
print(find_condition_date(start_date,days_month_list))

上面的代码似乎按预期工作。

Answer 1

首先，从start_date中提取日期

然后对月份天数进行排序，遍历排序后的数组，找到第一个大于开始日期的值

Answer 2

遍历可用日期，直到找到下一个最接近的有效日期（根据需要递增月/年）：

from datetime import date


def valid_date(y, m, d):
    try:
        date(y, m, d)
        return True
    except ValueError:
        return False


def find_next_date(start_date, days_of_month):
    current_year, current_month, current_day = map(int, start_date.split('-'))
    while True:
        for day in days_of_month:
            if day >= current_day and valid_date(current_year, current_month, day):
                return f'{current_year}-{current_month:02}-{day:02}'

        current_month += 1
        current_day = 1
        if current_month > 12:
            current_month = 1
            current_year += 1


assert find_next_date('2020-11-17', [5, 10, 20]) == '2020-11-20'
assert find_next_date('2020-11-17', [5, 10]) == '2020-12-05'
assert find_next_date('2020-12-17', [5, 10]) == '2021-01-05'
assert find_next_date('2021-02-17', [5, 10, 31]) == '2021-03-05'
assert find_next_date('2021-04-17', [31]) == '2021-05-31'