python计算两个日期的间隔天数

# -*- coding: utf-8 -*-

import datetime

def is_leap(year):
if (year % 4 == 0 and year % 100) or year % 400 == 0:
return 1
else:
return 0

def get_dates(year, month, day):
month_day = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
if is_leap(year):
month_day[1] = 29
days = 0
for y in range(1, year):
days += 365 + is_leap(y)
for m in range(month):
days += month_day[m]
days += day
return days

def fun(year1, month1, day1, year2, month2, day2):
d1 = get_dates(year1, month1, day1)
d2 = get_dates(year2, month2, day2)
return d1 - d2

if __name__ == '__main__':
year1, month1, day1 = 2017, 10, 17
year2, month2, day2 = 1989, 11, 23
d1 = datetime.date(year1, month1, day1)
d2 = datetime.date(year2, month2, day2)
print((d1 - d2).days)
print(fun(year1, month1, day1, year2, month2, day2))