Python代码阅读（第79篇）：日期差计算

Python 代码阅读合集介绍：为什么不推荐Python初学者直接看项目源码

本篇阅读的代码在日期类型上实现了日期差的计算（按天作为单位以及按月作为单位）。

本篇阅读的代码片段来自于30-seconds-of-python。

days_diff

def days_diff(start, end):
  return (end - start).days

# EXAMPLES
from datetime import date

print(days_diff(date(2020, 10, 25), date(2020, 10, 28))) # 3

days_diff函数接收一个开始日期和一个结束日期，返回两个日期相差的天数。

函数直接使用timedelta = date1 - date2运算方式计算日期差，然后通过timedelta的days属性获取相差的天数。

months_diff

from math import ceil

def months_diff(start, end):
  return ceil((end - start).days / 30)

# EXAMPLES
from datetime import date

print(months_diff(date(2020, 10, 28), date(2020, 11, 25))) # 1

months_diff函数在days_diff基础上除以30，并通过math.ceil(x)取大于等于x的最大整数。也就是说不满一个月的，按照一个月计算。

from math import ceil

def days_diff(start, end):
  return (end - start).days

def months_diff(start, end):
  return ceil((end - start).days / 30)