2. 请写一个模块，实现如下功能，并写一段代码调用该模块来测试功能是否正确。（30分）） 实现根据待显示的时间和当前时间的时间差展示不同的样式。如果时间差大于一年（此处以365天为例)，则显示哪年哪月；如果时间差小于或等于一年且大于一个月（此处以30天为例)，则显示哪年哪月哪日；如果时间差小于或等于一个月且大于一天，则显示几天前。依此类推， 显示几小时前、几分钟前和几秒前。也就是要求输出类似如下的结果：

以下是实现上述功能的 Python 代码：

import datetime

def time_diff_style(time_str):
    now = datetime.datetime.now()
    time = datetime.datetime.strptime(time_str,'%Y-%m-%d %H:%M:%S')
    diff = now - time
    if diff.days > 365:
        return time.strftime('%Y-%m')
    elif diff.days > 30:
        return time.strftime('%Y-%m-%d')
    elif diff.days > 0:
        return str(diff.days) + '天前'
    elif diff.seconds > 3600:
        return str(diff.seconds // 3600) + '小时前'
    elif diff.seconds > 60:
        return str(diff.seconds // 60) + '分钟前'
    else:
        return '刚刚'

# 测试代码
time_str = '2020-08-15 10:30:00'
print(time_diff_style(time_str))

假设当前时间是 2021 年 8 月 20 日，运行上述代码输出结果为 `1天前`，即表示距离输入的时间 `2020-08-15 10:30:00` 已经过去了 1 天。