python - 计算从今天起5天的日期，在接下来的5天中的每一天都增加一天，即周末

Question

我正在使用 Robot Framework 进行测试，需要创建我自己的 Python 关键字。

将当前日期作为第 0 天(明天作为第 1 天)，我试图计算从今天起 5 天后的日期。如果接下来 5 天中的任何一天是星期六，我需要在我的计算中增加额外的一天。如果其中任何一天是星期日，则相同。

作为一个 Python 初学者，我有点力不从心，所以非常感谢任何帮助

Answer 1

基本上您需要添加 5 个工作日...应该这样做:

import datetime
def addBusinessDays(from_date, add_days):
    business_days_to_add = add_days
    current_date = from_date
    while business_days_to_add > 0:
        current_date += datetime.timedelta(days=1)
        weekday = current_date.weekday()
        if weekday >= 5: # sunday = 6
            continue
        business_days_to_add -= 1
    return current_date

#demo:
print '5 business days from today:'
print addBusinessDays(datetime.date.today(), 5)

更新:

解释如下:

1. 我们得到开始日期(我们需要添加工作日的日期)
2. 我们使用循环将日期每次添加 1 天(我们使用 datetime.timedelta(days=1) 将日期添加 1 天)
3. 添加每一天后，我们检查更新日期是否为工作日。如果是工作日，我们计算它，否则我们不计算它并继续