如何比较双节棍中的日期？

import datetime

def compare_dates(date1, date2):
    # 将日期字符串转换为datetime对象
    dt1 = datetime.datetime.strptime(date1, "%Y-%m-%d")
    dt2 = datetime.datetime.strptime(date2, "%Y-%m-%d")

    # 比较年份
    if dt1.year < dt2.year:
        return f"{date1} 在 {date2} 之前"
    elif dt1.year > dt2.year:
        return f"{date1} 在 {date2} 之后"
    else:
        # 比较月份
        if dt1.month < dt2.month:
            return f"{date1} 在 {date2} 之前"
        elif dt1.month > dt2.month:
            return f"{date1} 在 {date2} 之后"
        else:
            # 比较天数
            if dt1.day < dt2.day:
                return f"{date1} 在 {date2} 之前"
            elif dt1.day > dt2.day:
                return f"{date1} 在 {date2} 之后"
            else:
                return f"{date1} 和 {date2} 相同"

# 示例用法
date1 = "2021-10-01"
date2 = "2021-12-25"
result = compare_dates(date1, date2)
print(result)