最近做一个需求,是计算查询日期范围内产生的费用,需要将查询日期范围
跟实际产生费用日期范围
计算重叠天数,然后重叠天数*每天费用=查询日期范围内的费用
。
一个时间段跟另一个时间段,有如下三种情况:
1. 没有重叠
2. 部分重叠
3. 包含式重叠。即一个时间段被另一个时间段包含。
经过分析,将2个时间段按时间线排列,只需要算出最早end_date
跟最晚start_date
的交集就行了。
即overlap = (min(end1, end2) - max(start1, start2)).days + 1
下面是完整代码
def calc_overlap_days(s1, e1, s2, e2):
latest_start = max(s1, s2)
earliest_end = min(e1, e2)
overlap = (earliest_end - latest_start).days + 1
if overlap < 0:
overlap = 0
return overlap