用于朋友互致问候列表的python代码

# 导入datetime模块获取当前时间
from datetime import datetime

# 定义一个朋友列表，包含朋友的姓名和生日
friends = [
    {"name": "张三", "birthday": "1990-01-01"},
    {"name": "李四", "birthday": "1992-05-10"},
    {"name": "王五", "birthday": "1995-11-20"}
]

# 定义一个函数，根据生日计算年龄
def calculate_age(birthday):
    today = datetime.today()
    birth_date = datetime.strptime(birthday, "%Y-%m-%d")
    age = today.year - birth_date.year
    if today.month < birth_date.month or (today.month == birth_date.month and today.day < birth_date.day):
        age -= 1
    return age

# 遍历朋友列表，生成问候语并打印
for friend in friends:
    age = calculate_age(friend["birthday"])
    print(f"你好，{friend['name']}！祝你{age}岁生日快乐！")