在实际的开发中,我们经常要设计一些工具类,对于时间来说,有时候需要将其处理成时间段。
例如,对于2024年08月01日
到2024年08月16日
的时间段,我们如何将其处理成时间段[2024-08-01, 2024-08-03], [2024-08-04, 2024-08-10], [2024-08-11, 2024-08-16]
。如下表所示:
周日 | 周一 | 周二 | 周三 | 周四 | 周五 | 周六 |
---|---|---|---|---|---|---|
1 | 2 | 3 | ||||
4 | 5 | 6 | 7 | 8 | 9 | 10 |
11 | 12 | 13 | 14 | 15 | 16 |
这里,我们采用迭代就可以实现,通过计算每个阶段的开始日期和结束日期,先根据当前阶段开始日期计算当周周日的日期,再判断总的结束日期是否在周日前,如果在,则将第一段时间的结束日期确定为总的结束日期。如果在超出了周日,则将周日定为该阶段的结束日期,以此类推。
这里我们用到几个函数
datetime.datetime.strptime
:将字符串转为日期对象
datetime.timedelta
: 计算当前日期前推或后推后的日期。
datetime.weekday
: 计算当前日期是星期几,其中0代表周末,1代表周一,2代表周二,以此类推。
def split_date_range(start_date_input: str, end_date_input: str) -> list:
# 将输入的日期字符串转换为日期对象
start_date = datetime.datetime.strptime(start_date_input, '%Y-%m-%d')
end_date =