如何计算周内两个日期之间的差额？

import datetime

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

    # 确定两个日期所在的周几
    weekday1 = date1.isoweekday()
    weekday2 = date2.isoweekday()

    # 计算两个日期之间的天数差额
    difference = (date2 - date1).days

    # 判断是否需要考虑周末
    if weekday1 > 5 and weekday2 > 5:
        # 两个日期都在周末
        difference -= 2
    elif weekday1 > 5 or weekday2 > 5:
        # 一个日期在周末，一个日期在工作日
        difference -= 1

    return difference

# 示例用法
date1 = "2022-01-01"
date2 = "2022-01-07"
difference = calculate_weekday_difference(date1, date2)
print("差额为:", difference)