datetime获取三个月之后的时间

import datetime


def get_year():
    '''
    因学习python两周,代码略low,欢迎大佬指正
    获取年份
    :return:
    '''
    year = str(datetime.datetime.now()).split('-')[0]
    return year


def panduan_year():
    '''
    判断是平年还是闰年
    :return:
    '''
    year = int(get_year())
    return year % 400 == 0 or year % 4 == 0 and year % 100 != 0


def get_days():
    '''
    因datetime.timedelta无法传递月或者年，只能传递下方的参数
    days=0, seconds=0, microseconds=0,
    milliseconds=0, minutes=0, hours=0, weeks=0):
    此方法为获取三个月之后的时间
    :return:
    '''
    current_month = datetime.datetime.now().month
    current_month_today = datetime.datetime.now().day
    if panduan_year():
        days_list = [31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
        all_days = sum([days_list[i] for i in range(current_month - 1, current_month + 3)])
        count_days = datetime.timedelta(days=(all_days - current_month_today))
        return datetime.datetime.now() + count_days
    else:
        days_list = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
        all_days = sum([days_list[i] for i in range(current_month - 1, current_month + 3)])
        count_days = datetime.timedelta(days=(all_days - current_month_today))
        return datetime.datetime.now() + count_days


print(get_days())